Archived
195 lines
6.4 KiB
Python
195 lines
6.4 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"
|
|
|
|
allow_update = True
|
|
|
|
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):
|
|
id = str(id)
|
|
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 parse_lesson(time, lesson):
|
|
text = ''
|
|
start, end = time.split("-")
|
|
|
|
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:
|
|
print("что-то не так")
|
|
print(lesson["name"])
|
|
|
|
return text
|
|
|
|
def create_schedule_tasks(manual=False):
|
|
global allow_update
|
|
if not manual:
|
|
bot.send_admin_message('Произошёл auto-update')
|
|
if not allow_update:
|
|
return
|
|
|
|
allow_update = True
|
|
|
|
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 = i.split("-")[0]
|
|
text = parse_lesson(i, lesson)
|
|
|
|
if text == '':
|
|
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)
|
|
|
|
return
|
|
|
|
def groups_in_json():
|
|
schdl = {}
|
|
with open(path_schedule, 'r', encoding='utf-8') as json_file:
|
|
schdl = json.load(json_file)
|
|
|
|
return schdl.keys()
|
|
|
|
def return_infos(id):
|
|
id = str(id)
|
|
data = {}
|
|
with open(path_users, 'r', encoding='utf-8') as json_file:
|
|
data = json.load(json_file)
|
|
data = data[id]
|
|
|
|
return data
|
|
|
|
def pause_bot():
|
|
schedule.clear()
|
|
global allow_update
|
|
allow_update = False
|
|
|
|
return
|
|
|
|
def get_schedule(id, day):
|
|
id = str(id)
|
|
if day == 'sun':
|
|
day = 'mon'
|
|
days = ['mon', 'tue', 'wed', 'thu', 'fri', 'sat']
|
|
russian_days = ['понедельник', 'вторник', 'среду', 'четверг', 'пятницу', 'субботу']
|
|
|
|
text = '<u>Расписание на {}</u>:\n\n'.format(russian_days[days.index(day)])
|
|
|
|
schdl_today = {}
|
|
with open(path_schedule, 'r', encoding='utf-8') as schedule_file:
|
|
group = ''
|
|
with open(path_users, 'r', encoding='utf-8') as user_file:
|
|
group = json.load(user_file)[id]['group']
|
|
|
|
if (group == 'other'):
|
|
return 'У тебя не выбрана группа для рассылки сообщений'
|
|
|
|
try:
|
|
schdl_today = json.load(schedule_file)[group][day]
|
|
except:
|
|
return 'Расписания твоей группы на {} нет'.format(russian_days[days.index(day)])
|
|
|
|
for i, lesson in schdl_today.items():
|
|
text += '•' + parse_lesson(i, lesson) + '\n\n'
|
|
|
|
return text
|