Archived
some upd
This commit is contained in:
+69
-109
@@ -5,69 +5,13 @@ from dotenv import load_dotenv
|
||||
from google.oauth2 import service_account
|
||||
from sshtunnel import SSHTunnelForwarder
|
||||
from psycopg2 import pool
|
||||
from datetime import datetime
|
||||
|
||||
load_dotenv()
|
||||
|
||||
bot = telebot.TeleBot(os.environ.get("TG_TEST_TOKEN"))
|
||||
|
||||
"""
|
||||
CREATE TABLE IF NOT EXISTS schedule(
|
||||
id BIGSERIAL PRIMARY KEY,
|
||||
day_of_week VARCHAR(255),
|
||||
begin_time VARCHAR(20),
|
||||
end_time VARCHAR(20),
|
||||
course VARCHAR(255),
|
||||
lector VARCHAR(255),
|
||||
is_lecture BOOLEAN DEFAULT FALSE,
|
||||
room VARCHAR(20),
|
||||
group VARCHAR(10),
|
||||
parity VARCHAR(10),
|
||||
);
|
||||
"""
|
||||
|
||||
"""
|
||||
День недели Время начала Название Чётность пары Лекция? Преподаватель Кабинет 2-ой преподаватель Прочие кабинеты Остальная информация
|
||||
Понедельник 8:45 Английский язык - FALSE Бим Мария Моисеевна 735 Шубина Юлия Витальевна 790
|
||||
Понедельник 10:30 Практикум на ЭВМ - FALSE Тюляева Вера Викторовна МЗ-4
|
||||
Понедельник 12:15 Действительный и комплексный анализ - TRUE Крицков Леонид Владимирович П-12
|
||||
Понедельник 15:00 Физическая культура - FALSE
|
||||
Понедельник 16:20 FALSE
|
||||
Вторник 8:45 Теория вероятностей и математическая статистика - TRUE Королёв Виктор Юрьевич П-13
|
||||
Вторник 10:30 Обыкновенные дифференциальные уравнения - FALSE Дмитриева Ирина Владимировна 607
|
||||
Вторник 12:15 Действительный и комплексный анализ - FALSE Сычугов Дмитрий Юрьевич 613
|
||||
Вторник 14:35 Электродинамика - FALSE Аксенов Владимир Николаевич 659
|
||||
Вторник 16:20 - FALSE
|
||||
Среда 8:45 Практикум на ЭВМ - FALSE Тюляева Вера Викторовна 505
|
||||
Среда 10:30 Теория вероятностей и математическая статистика - FALSE Ирина Геннадьевна Шевцова 645
|
||||
Среда 12:15 Обыкновенные дифференциальные уравнения - TRUE Денисов Александр Михайлович П-13
|
||||
Среда 15:00 FALSE
|
||||
Среда 16:20 FALSE
|
||||
Четверг 9:00 FALSE
|
||||
Четверг 10:30 FALSE
|
||||
Четверг 12:15 FALSE
|
||||
Четверг 15:00 FALSE
|
||||
Четверг 16:20 FALSE
|
||||
Пятница 9:00 FALSE
|
||||
Пятница 10:30 FALSE
|
||||
Пятница 12:15 FALSE
|
||||
Пятница 15:00 FALSE
|
||||
Пятница 16:20 FALSE
|
||||
Суббота 9:00 FALSE
|
||||
Суббота 10:30 FALSE
|
||||
Суббота 12:15 FALSE
|
||||
Суббота 15:00 FALSE
|
||||
Суббота 16:20 FALSE
|
||||
FALSE
|
||||
FALSE
|
||||
FALSE
|
||||
FALSE
|
||||
FALSE
|
||||
FALSE
|
||||
FALSE
|
||||
FALSE
|
||||
FALSE
|
||||
"""
|
||||
|
||||
SERVICE_ACCOUNT_FILE = "app/credentials.json"
|
||||
SERVICE_ACCOUNT_FILE = "credentials.json"
|
||||
|
||||
credentials = service_account.Credentials.from_service_account_file(
|
||||
SERVICE_ACCOUNT_FILE,
|
||||
@@ -81,10 +25,6 @@ gc = gspread.authorize(credentials)
|
||||
|
||||
table = gc.open_by_key(os.environ.get("TABLE_ID"))
|
||||
|
||||
ws = table.worksheets()
|
||||
|
||||
bot = telebot.TeleBot(os.environ.get("TG_TEST_TOKEN"))
|
||||
|
||||
if os.environ.get("ENV") == "dev":
|
||||
print("DEV: Connecting to local database")
|
||||
server = SSHTunnelForwarder(
|
||||
@@ -113,52 +53,72 @@ db = pool.SimpleConnectionPool(
|
||||
)
|
||||
|
||||
|
||||
for i in ws:
|
||||
print(i.title)
|
||||
def f():
|
||||
ws = table.worksheets()
|
||||
|
||||
if not (i.title == "Пример" or i.title == "main") and i.title.isdigit():
|
||||
df = pd.DataFrame(i.get_all_records())
|
||||
if not df.empty:
|
||||
with db.getconn() as conn:
|
||||
with conn.cursor() as cursor:
|
||||
cursor.execute(
|
||||
'DELETE FROM schedule WHERE "group" = %s', (str(i.title),)
|
||||
)
|
||||
conn.commit()
|
||||
for index, row in df.iterrows():
|
||||
day_of_week_map = {
|
||||
"Понедельник": "mon",
|
||||
"Вторник": "tue",
|
||||
"Среда": "wed",
|
||||
"Четверг": "thu",
|
||||
"Пятница": "fri",
|
||||
"Суббота": "sat",
|
||||
"Воскресенье": "sun",
|
||||
}
|
||||
row["День недели"] = day_of_week_map.get(
|
||||
row["День недели"], row["День недели"]
|
||||
)
|
||||
if row["Название"] == "":
|
||||
continue
|
||||
end_time = (
|
||||
pd.to_datetime(row["Время начала"], format="%H:%M")
|
||||
+ pd.Timedelta(hours=1, minutes=35)
|
||||
).strftime("%H:%M")
|
||||
for i in ws:
|
||||
# print(i.title)
|
||||
|
||||
if not (i.title == "Пример" or i.title == "main") and i.title.isdigit():
|
||||
df = pd.DataFrame(i.get_all_records())
|
||||
if not df.empty:
|
||||
with db.getconn() as conn:
|
||||
with conn.cursor() as cursor:
|
||||
cursor.execute(
|
||||
"""
|
||||
INSERT INTO schedule (day_of_week, begin_time, end_time, course, lector, is_lecture, room, "group", parity)
|
||||
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s)
|
||||
""",
|
||||
(
|
||||
row["День недели"],
|
||||
row["Время начала"],
|
||||
end_time,
|
||||
row["Название"],
|
||||
f'{row["Преподаватель"]}{"/" + str(row["2-ой преподаватель"]) if row["2-ой преподаватель"] else ""}',
|
||||
row["Лекция?"],
|
||||
f'{row["Кабинет"]}{"/" + str(row["Прочие кабинеты"]) if row["Прочие кабинеты"] else ""}',
|
||||
i.title,
|
||||
row["Чётность пары"],
|
||||
),
|
||||
'DELETE FROM schedule WHERE "group" = %s', (str(i.title),)
|
||||
)
|
||||
conn.commit()
|
||||
conn.commit()
|
||||
for index, row in df.iterrows():
|
||||
day_of_week_map = {
|
||||
"Понедельник": "mon",
|
||||
"Вторник": "tue",
|
||||
"Среда": "wed",
|
||||
"Четверг": "thu",
|
||||
"Пятница": "fri",
|
||||
"Суббота": "sat",
|
||||
"Воскресенье": "sun",
|
||||
}
|
||||
row["День недели"] = day_of_week_map.get(
|
||||
row["День недели"], row["День недели"]
|
||||
)
|
||||
if row["Название"] == "":
|
||||
continue
|
||||
end_time = (
|
||||
pd.to_datetime(row["Время начала"], format="%H:%M")
|
||||
+ pd.Timedelta(hours=1, minutes=35)
|
||||
).strftime("%H:%M")
|
||||
cursor.execute(
|
||||
"""
|
||||
INSERT INTO schedule (day_of_week, begin_time, end_time, course, lector, is_lecture, room, "group", parity)
|
||||
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s)
|
||||
""",
|
||||
(
|
||||
row["День недели"],
|
||||
row["Время начала"],
|
||||
end_time,
|
||||
row["Название"],
|
||||
f'{row["Преподаватель"]}{"/" + str(row["2-ой преподаватель"]) if row["2-ой преподаватель"] else ""}',
|
||||
row["Лекция?"],
|
||||
f'{row["Кабинет"]}{"/" + str(row["Прочие кабинеты"]) if row["Прочие кабинеты"] else ""}',
|
||||
i.title,
|
||||
row["Чётность пары"],
|
||||
),
|
||||
)
|
||||
conn.commit()
|
||||
|
||||
text = f"Last tg-bot update was at: {datetime.now().astimezone().strftime('%Y-%m-%d %H:%M %Z')}"
|
||||
table.get_worksheet(0).update_acell("A25", text)
|
||||
|
||||
return
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
f()
|
||||
except Exception as e:
|
||||
print(e)
|
||||
|
||||
bot.send_message(int(os.environ.get("ADMIN_ID")), "Check logs! upload error.")
|
||||
bot.send_message(int(os.environ.get("ADMIN_ID")), e)
|
||||
|
||||
exit(0)
|
||||
|
||||
Reference in New Issue
Block a user