This repository has been archived by the owner on Oct 15, 2021. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 2
/
sender.py
110 lines (97 loc) · 4.5 KB
/
sender.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
import os
import random
import time
import pymysql
import requests
import vk_api
from bs4 import BeautifulSoup
from pymysql.cursors import DictCursor
print("Sender launched")
mysql_l = os.environ['MYSQL_LOGIN']
mysql_p = os.environ["MYSQL_PASS"]
access_token = os.environ["ACCESS_TOKEN"]
vk = vk_api.VkApi(token=access_token)
def write_msg(user_id, random_id, message):
vk.method('messages.send', {'user_id': user_id, 'random_id': random_id, 'message': message})
def parsepage(table):
muudatused = []
for i in range(len(table)):
my_table = table[i]
rows = my_table.find_all('tr')
for row in rows:
muudatus = []
cells = row.find_all('td')
for cell in cells:
if cell.text not in ["\xa0", "Kuupäev", "Rühm", "Tund", "Õpetaja", "Ruum"]:
data = cell.text
muudatus.append(data)
# здесь есть полноценный список muudatus
if muudatus != []:
muudatused.append(muudatus)
else:
continue
return muudatused
def openfromfile(usergroup):
connection = pymysql.connect(
host='eu-cdbr-west-02.cleardb.net',
user=mysql_l,
password=mysql_p,
db='heroku_0ccfbccd1823b55',
cursorclass=DictCursor)
with connection.cursor() as cursor:
cursor.execute("""SELECT * FROM users WHERE sendStatus=1""")
row = cursor.fetchall()
# print(row)
for i in row:
# print(i)
usergroup[i['vkid']] = i['thkruhm']
cursor.close()
connection.close()
return usergroup
def makemuudatused(i, forshow):
if len(i) == 6:
forshow.append(f"🗓 {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]} ⏰ Урок: {i[3]} \n👨🏫 Преподаватель: {i[4]}\nКабинет: {i[5]}\n")
elif len(i) > 2 and i[3].lower() in "jääb ära":
forshow.append(f"🗓 {i[0]} Дата: {i[1]}\n🦆 {i[2]}\n❌ Не состоится\n")
elif len(i) > 4 and i[4].lower() in "jääb ära":
forshow.append(f"🗓 {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n❌ Не состоится\n")
elif len(i) > 4 and i[4].lower() in "söögivahetund":
forshow.append(f"🗓 {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]}\n ⏰ Урок: {i[3]}\n🆒 Обеденный перерыв\n")
elif len(i) > 5 and i[5].lower() in "iseseisev töö kodus":
forshow.append(f"🗓 {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n🏠 Самостоятельная работа дома\n")
elif len(i) > 5 and i[5].lower() in "iseseisev töö":
forshow.append(f"🗓 {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n📋 Самостоятельная работа\n")
elif len(i) > 5 and (i[5].lower() == "" or i[5].lower() == " "):
forshow.append(f"🗓 {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n👨🏫 Преподаватель: {i[4]}\n")
else:
forshow.append(f"🗓 В {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n")
return forshow
def getmuudatused(setgroup, user, justtable):
forshow = []
muudatused = parsepage(justtable)
for i in muudatused:
if setgroup.lower() in i[2].lower():
makemuudatused(i, forshow)
if len(forshow) > 0:
userfname = (vk.method('users.get', {'user_ids': user, 'fields': 'first_name'})[0])["first_name"]
kogutunniplaan = f"Доброе утро, {userfname}! Для группы 🦆 {setgroup} на данный момент следующие изменения в расписании:\n"
for w in forshow:
kogutunniplaan += f"{w}\n"
write_msg(user, (random.getrandbits(31) * random.choice([-1, 1])), kogutunniplaan)
elif len(forshow) == 0:
pass
def sendeveryday(justtable):
usergroup = {}
usergroup = openfromfile(usergroup)
print("Запускаю рассылку:")
print(time.strftime("%H:%M:%S"))
for i in usergroup.keys():
getmuudatused(usergroup[i], i, justtable)
def start_sender():
r = requests.get('http://www.tthk.ee/tunniplaani-muudatused/')
html_content = r.text
soup = BeautifulSoup(html_content, 'html.parser')
table = soup.findChildren('table')
sendeveryday(table)
if __name__ == '__main__' and time.strftime("%w", time.localtime()) in ['1', '2, '3', '4', '5']:
start_sender()