Archived
139 lines
5.5 KiB
Python
139 lines
5.5 KiB
Python
#!/usr/bin/python3.3
|
|
import json, schedule
|
|
import schedules as bot
|
|
from datetime import datetime
|
|
|
|
path_users = "json/users.json" # Нужный путь до json файлов
|
|
path_schedule = "json/schedule.json"
|
|
|
|
def save_user(infos):
|
|
for i in range(len(infos)):
|
|
if type(infos[i]) == type(None):
|
|
infos[i] = ''
|
|
infos[i] = str(infos[i])
|
|
infos = {
|
|
"id" : infos[0],
|
|
"first_name" : infos[1],
|
|
"last_name" : infos[2],
|
|
"username" : infos[3],
|
|
"group" : infos[4]
|
|
}
|
|
|
|
data = {}
|
|
with open(path_users, 'r', encoding='utf-8') as json_file:
|
|
data = json.load(json_file)
|
|
|
|
data[infos["id"]] = {
|
|
"username": infos["username"],
|
|
"first_name": infos["first_name"],
|
|
"last_name": infos["last_name"],
|
|
"group": infos["group"],
|
|
"timeout": "10",
|
|
"allow_message": "yes",
|
|
"thread": "General"
|
|
}
|
|
|
|
with open(path_users, 'w', encoding='utf-8') as json_file:
|
|
json.dump(data, json_file, ensure_ascii=False, indent=4)
|
|
create_schedule_tasks()
|
|
return
|
|
|
|
def change_user_param(id, key, value):
|
|
data = {}
|
|
with open(path_users, 'r', encoding='utf-8') as json_file:
|
|
data = json.load(json_file)
|
|
|
|
data[id][key] = value
|
|
|
|
with open(path_users, 'w', encoding='utf-8') as json_file:
|
|
json.dump(data, json_file, ensure_ascii=False, indent=4)
|
|
create_schedule_tasks()
|
|
return
|
|
|
|
def create_schedule_tasks():
|
|
schedule.clear()
|
|
users = {}
|
|
schdl = {}
|
|
with open(path_users, 'r', encoding='utf-8') as json_file:
|
|
users = json.load(json_file)
|
|
with open(path_schedule, 'r', encoding='utf-8') as json_file:
|
|
schdl = json.load(json_file)
|
|
for id, params in users.items():
|
|
if params['allow_message'] != 'yes':
|
|
continue
|
|
group = params['group']
|
|
if group == 'other':
|
|
continue
|
|
timeout = params['timeout']
|
|
for day in schdl[group]:
|
|
for i, lesson in schdl[group][day].items():
|
|
if lesson["name"] == "": # Неправильный формат schedule
|
|
continue
|
|
|
|
start, end = i.split("-")
|
|
text = ''
|
|
if len(lesson["infos"].split('|')) == 2:
|
|
teacher = lesson["infos"].split('|')[0]
|
|
room = lesson["infos"].split('|')[1]
|
|
text = '{}-{} | {}\
|
|
\n<b>{}</b>\
|
|
\n<i>{}</i>'.format(start, end, room, lesson["name"], teacher)
|
|
elif len(lesson["infos"].split('|')) == 4:
|
|
infos = lesson["infos"].split('|');
|
|
text = '{}-{}\
|
|
\n<b>{}</b>\
|
|
\n({}) <i>{}</i>\
|
|
\n({}) <i>{}</i>'.format(start, end, lesson["name"],
|
|
infos[1],
|
|
infos[0],
|
|
infos[3],
|
|
infos[2])
|
|
elif len(lesson["infos"].split('|')) == 1:
|
|
text = '{}-{}\
|
|
\n<b>{}</b>'.format(start, end, lesson["name"])
|
|
else: #неправильный формат infos в json файле
|
|
print("что-то не так")
|
|
print(lesson["name"])
|
|
continue
|
|
|
|
thread_id = params["thread"]
|
|
delta = '00:' + params["timeout"]
|
|
format = '%H:%M'
|
|
start_task = datetime.strptime(start, format) - datetime.strptime(delta, format)
|
|
tmp = ''
|
|
for i in str(start_task).split(':'):
|
|
tmp += (('0' + i ) if len(i) < 2 else i) + ":"
|
|
start_task = tmp[:-1]
|
|
if day == 'mon':
|
|
schedule.every().monday.at(start_task).do(bot.send_message, id, text, thread_id)
|
|
elif day == 'tue':
|
|
schedule.every().tuesday.at(start_task).do(bot.send_message, id, text, thread_id)
|
|
elif day == 'wed':
|
|
schedule.every().wednesday.at(start_task).do(bot.send_message, id, text, thread_id)
|
|
elif day == 'thu':
|
|
schedule.every().thursday.at(start_task).do(bot.send_message, id, text, thread_id)
|
|
elif day == 'fri':
|
|
schedule.every().friday.at(start_task).do(bot.send_message, id, text, thread_id)
|
|
elif day == 'sat':
|
|
schedule.every().saturday.at(start_task).do(bot.send_message, id, text, thread_id)
|
|
|
|
bot.send_message(bot.admin_id, '🛑 Произошёл update')
|
|
return
|
|
|
|
|
|
def check_group_in_json(number):
|
|
schdl = {}
|
|
with open(path_schedule, 'r', encoding='utf-8') as json_file:
|
|
schdl = json.load(json_file)
|
|
return str(number) in schdl.keys()
|
|
|
|
def return_infos(id):
|
|
data = {}
|
|
with open(path_users, 'r', encoding='utf-8') as json_file:
|
|
data = json.load(json_file)
|
|
data = data[id]
|
|
text = '<i>Выбранная группа</i>: <b>{}</b>\
|
|
\n<i>Время напоминания до урока</i>: <b>{}</b>\
|
|
\n<i>Разрешены ли напоминания</i>: <b>{}</b>'.format(data['group'], data['timeout'], ('да' if data['allow_message'] == 'yes' else 'нет'))
|
|
return text
|