def scheduled_job():
# update groups table
soup.get_groups()
# update xls files
for root, directories, filenames in os.walk('./schedule/'):
for filename in filenames:
u = os.path.join(root, filename).replace("\\", "/")
response = requests.get('http://www.sgu.ru{}/lesson'.format(
u[1:-4]))
with open('{}'.format(u), mode='rb') as xlsfile:
fileContent = xlsfile.read()
if fileContent != response.content:
with open('{}'.format(u), mode='wb') as f:
f.write(response.content)
# update users table
db = dbconn.sqldb(config.database)
users_urls = db.get_urls_from_users()
group_urls = db.select_url_from_groups()
for url in users_urls:
if url in group_urls:
db.update_schedule_by_url(url)
def callback_date(call):
dayweeks = {0: 'понедельник', 1: 'вторник', 2: 'среда', 3: 'четверг', 4: 'пятница', 5: 'суббота', 6: 'воскресенье'}
dayweeks_eng_number = {'monday': 0, 'tuesday': 1, 'wednesday': 2, 'thursday': 3, 'friday': 4, 'saturday': 5}
dayweeks_eng_rus = {'monday': 'понедельник', 'tuesday': 'вторник',
'wednesday': 'среду', 'thursday': 'четверг', 'friday': 'пятницу', 'saturday': 'субботу'}
db = dbconn.sqldb(config.database)
if call.data == 'today':
bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id, text=
'<b>Сегодня {0}, {1}.\n\nТекущая неделя – {2}.\n\n</b>Ваше расписание:\n\n'.format(
date.today().strftime('%d-%m-%Y'),
dayweeks[date.today().weekday()],
'числитель' if is_week_odd(date.today()) else 'знаменатель')
+ db.get_schedule(call.message.chat.id, date.today().weekday())[0][0],
parse_mode='HTML')
elif call.data == 'yesterday':
yesterday = date.today() + timedelta(days=-1)
bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id, text=
'<b>Вчерашний день: {1}, {0}.\n\nНеделя – {2}.\n\n</b>Расписание:\n\n'.format(
yesterday.strftime('%d-%m-%Y'),
dayweeks[yesterday.weekday()],
'числитель' if is_week_odd(yesterday) else 'знаменатель')
+ db.get_schedule(call.message.chat.id, yesterday.weekday())[0][0],
parse_mode='HTML')
elif call.data == 'tomorrow':
tomorrow = date.today() + timedelta(days=1)
bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id, text=
'<b>Завтра: {1}, {0}.\n\nНеделя – {2}.\n\n</b>Расписание:\n\n'.format(
tomorrow.strftime('%d-%m-%Y'),
dayweeks[tomorrow.weekday()],
'числитель' if is_week_odd(tomorrow) else 'знаменатель')
+ db.get_schedule(call.message.chat.id, tomorrow.weekday())[0][0],
parse_mode='HTML')
elif call.data == 'dayweek':
dayweeks_number_eng = {0: 'monday', 1: 'tuesday', 2: 'wednesday', 3: 'thursday', 4: 'friday', 5: 'saturday'}
keyboard = types.InlineKeyboardMarkup(row_width=3)
k = []
for i in dayweeks.keys():
if i != 6:
k.append(types.InlineKeyboardButton(text="{}".format(dayweeks[i].capitalize()),
callback_data=dayweeks_number_eng[i]))
keyboard.add(*k)
bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id, text=
'Выберите день недели:',
parse_mode='HTML')
bot.edit_message_reply_markup(chat_id=call.message.chat.id, message_id=call.message.message_id,
reply_markup=keyboard)
elif call.data in ('monday', 'tuesday', 'wednesday', 'thursday', 'friday', 'saturday'):
try:
bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id, text=
'Расписание на {}:\n\n'.format(dayweeks_eng_rus[call.data])
+ db.get_schedule(call.message.chat.id, dayweeks_eng_number[call.data])[0][0],
parse_mode='HTML')
except Exception as excep:
print('Error in edit_message_text: ')
print(str(excep))
def callback_date(call):
keyboard = types.InlineKeyboardMarkup(row_width=3)
groups = []
for url, group_rus in dbconn.sqldb(config.database).select_url_groups_by_department_and_course(
call.data.split('+')[0], call.data.split('+')[1]):
groups.append(types.InlineKeyboardButton(text="{}".format(group_rus), callback_data="{}".format(url)))
keyboard.add(*groups)
bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id, text="Выберите группу:")
bot.edit_message_reply_markup(chat_id=call.message.chat.id, message_id=call.message.message_id,
reply_markup=keyboard)
def sendmsgtoallusers():
markup = types.ReplyKeyboardMarkup()
markup.row('📚 Занятия', '📅 Сессия')
markup.row('📝 Сменить группу')
for user in dbconn.sqldb(config.database).get_all_ids():
try:
bot.send_message(int(user), sys.argv[1], reply_markup=markup)
print('Sent')
except:
print("Didn't send")
time.sleep(0.1)
print('Done! All messages were sent')
def send_msg(message):
db = dbconn.sqldb(config.database)
if not db.check_user(message.chat.id):
bot.send_message(message.chat.id,
'Я еще не знаю номер вашей группы.\nНажмите на кнопку "📝 Сменить группу", чтобы задать его.')
else:
keyboard = types.InlineKeyboardMarkup(row_width=2)
today = types.InlineKeyboardButton(text="Сегодня", callback_data="today")
yesterday = types.InlineKeyboardButton(text="Вчера", callback_data="yesterday")
tomorrow = types.InlineKeyboardButton(text="Завтра", callback_data="tomorrow")
anotherdayweek = types.InlineKeyboardButton(text="День недели", callback_data="dayweek")
keyboard.add(today, yesterday, tomorrow, anotherdayweek)
bot.send_message(message.chat.id, "Выберите день, расписание которого надо узнать:", reply_markup=keyboard)
dashbot.track(message)
def send_session(message):
db = dbconn.sqldb(config.database)
if not db.check_user(message.chat.id):
bot.send_message(message.chat.id,
'Я еще не знаю номер вашей группы.\nНажмите на кнопку "📝 Сменить группу", чтобы задать его.')
else:
try:
url = db.select_url_by_id(message.chat.id)[0]
session = soup.get_session(url)
db.update_session(message.chat.id, session)
except:
session = db.select_session(message.chat.id)[0]
bot.send_message(message.chat.id, '<b>Сегодня: ' + date.today().strftime('%d-%m-%Y') +
'</b>\n\nРасписание сессии:\n\n' + session, parse_mode='HTML')
dashbot.track(message)
def callback_inline(call):
# call.data is a department
db = dbconn.sqldb(config.database)
keyboard = types.InlineKeyboardMarkup(row_width=3)
k = []
for course_rus, course_eng in db.select_course_by_department(call.data):
k.append(types.InlineKeyboardButton(text="{}".format(course_rus),
callback_data="{}".format(call.data + '+' + course_eng)))
keyboard.add(*k)
bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id, text="Выберите курс:")
bot.edit_message_reply_markup(chat_id=call.message.chat.id, message_id=call.message.message_id,
reply_markup=keyboard)
def callback_date(call):
db = dbconn.sqldb(config.database)
dep_and_group = db.select_by_url(call.data)
if db.check_user(call.message.chat.id):
db.update_user_schedule(call.message.chat.id,
call.data)
else:
db.insert_user_schedule(call.message.chat.id,
call.from_user.first_name + (
"" if call.from_user.last_name is None else " " + call.from_user.last_name) + (
"" if call.from_user.username is None else " @" + call.from_user.username),
call.data)
try:
bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id,
text="Готово! Ваш факультет - {0}. Ваша группа - {1}."
.format(dep_and_group[0], dep_and_group[1]))
except Exception as excep:
print('Error in edit_message_text: ')
print(str(excep))
def get_groups(): # get all groups from parsed site
db = dbconn.sqldb(config.database)
db.delete_from_groups()
completedata = []
for department in departments.keys():
print(departments[department])
url = "http://www.sgu.ru/schedule/{}/".format(department)
f = request.urlopen(url)
soup = BeautifulSoup(f.read(), 'html.parser')
for link in (soup.find_all(
'fieldset', 'do form_education form-wrapper')[0].find_all(
'fieldset', 'course form-wrapper')):
for k in link.find_all('a'):
course = link.legend.string[
0:-1] if link.legend is not None else 'ДРУГОЕ'
completedata.append(
(k.get('href'), k.string, course, courses[course],
departments[department], department))
db.insert_into_groups(completedata)
keyboard = types.InlineKeyboardMarkup(row_width=3)
k = []
for course_rus, course_eng in db.select_course_by_department(call.data):
k.append(types.InlineKeyboardButton(text="{}".format(course_rus),
callback_data="{}".format(call.data + '+' + course_eng)))
keyboard.add(*k)
bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id, text="Выберите курс:")
bot.edit_message_reply_markup(chat_id=call.message.chat.id, message_id=call.message.message_id,
reply_markup=keyboard)
@bot.callback_query_handler(func=lambda msg: msg.data in
[i + '+' + course for course in
dbconn.sqldb(config.database).select_distinct_course() for i in
soup.departments.keys()]) # choose group
def callback_date(call):
keyboard = types.InlineKeyboardMarkup(row_width=3)
groups = []
for url, group_rus in dbconn.sqldb(config.database).select_url_groups_by_department_and_course(
call.data.split('+')[0], call.data.split('+')[1]):
groups.append(types.InlineKeyboardButton(text="{}".format(group_rus), callback_data="{}".format(url)))
keyboard.add(*groups)
bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id, text="Выберите группу:")
bot.edit_message_reply_markup(chat_id=call.message.chat.id, message_id=call.message.message_id,
reply_markup=keyboard)
@bot.callback_query_handler(func=lambda msg: msg.data in ('today', 'yesterday', 'tomorrow',