bot.py

# Connecting vk_api
# Connecting time tools
import datetime
# Connecting tools of deploy
import os
# Connecting parsing tools
import re

# Connecting pyMySQL
import pymysql
import requests
import vk_api
from bs4 import BeautifulSoup
from pymysql.cursors import DictCursor
from vk_api.keyboard import VkKeyboard, VkKeyboardColor
from vk_api.longpoll import VkLongPoll, VkEventType
from vk_api.utils import get_random_id


class Server:

    def __init__(self, api_token):
        self.vk = vk_api.VkApi(token=api_token)
        self.longpoll = VkLongPoll(self.vk)  # API, that makes possible get messages.
        self.bot = Bot(self.vk)
        self.writeyourgroup = []
        self.writesearchgroup = []
        self.writeweekday = []
        self.writedate = []

    def start(self):
        print("Bot successfully deployed and started.")  # Console message when bot deployed.
        for event in self.longpoll.listen():
            if event.type == VkEventType.MESSAGE_NEW:
                if event.to_me:
                    if event.text.lower() == 'начать':  # Start command
                        self.bot.sendKeyboard(keyboard=Keyboard.keyboard, vkid=event.user_id,
                                              msg='Выберите вариант из списка ниже.')
                    elif event.text.lower() == 'covid-19':  # Returns COVID-19 data
                        self.bot.sendMsg(vkid=event.user_id, msg=COVID.getData())
                    elif event.text.lower() == 'по датам':  # Selection keyboard of the next 5 days
                        self.writedate.append(event.user_id)
                        self.bot.sendKeyboard(keyboard=Keyboard.fiveDaysKeyboard, vkid=event.user_id,
                                              msg='Выберите дату из списка ниже:')
                    elif event.text.lower() == 'по дню недели':  # Selection keyboard of days of the week
                        self.writeweekday.append(event.user_id)
                        self.bot.sendKeyboard(keyboard=Keyboard.weekDaysKeyboard, vkid=event.user_id,
                                              msg='Выберите день недели из списка ниже:')
                    elif event.text.lower() == 'в какой я группе?':  # Return current user's group
                        if SQL.getUserGroup(vkid=event.user_id) is not None:
                            self.bot.sendMsg(vkid=event.user_id,
                                             msg=f'Вы указали, что Ваша группа:'
                                                 f' {SQL.getUserGroup(vkid=event.user_id)}.\n'
                                                 'Для того, чтобы изменить свою группу нажмите \"Изменить группу\".')
                        else:
                            self.bot.sendMsg(vkid=event.user_id,
                                             msg=f'Вы не указали группу. Нажмите кнопку '
                                                 f'\"Изменить группу\" и введите её.')
                    elif event.text.lower() == 'изменить группу':  # User can change group
                        self.bot.sendMsg(vkid=event.user_id, msg="В какой группе вы находитесь?\n"
                                                                 "Для групп, которые делятся на подгруппы указывается "
                                                                 "только группа: MEHpv19 вместо MEHpv19-2.\n"
                                                                 "Укажите код вашей группы:")
                        self.writeyourgroup.append(event.user_id)
                    elif event.text.lower()[
                         -3:] in TimeCatcher.getGroupList() and event.user_id in self.writeyourgroup:
                        # Receives group of the user
                        SQL.setUserGroup(vkid=event.user_id, group=event.text)
                        self.bot.sendMsg(vkid=event.user_id,
                                         msg=f'Вы указали, что Ваша группа: {SQL.getUserGroup(vkid=event.user_id)}.')
                        self.writeyourgroup.remove(event.user_id)
                    elif event.text.lower() == 'моя группа':
                        if SQL.getUserGroup(vkid=event.user_id) is not None:
                            self.bot.sendMsg(vkid=event.user_id, msg=Changes.makeChanges(SQL.getUserGroup(vkid=event.user_id)))
                        else:
                            self.bot.sendMsg(vkid=event.user_id,
                                             msg=f'Вы не указали группу. Нажмите кнопку \"Изменить группу\" и введите её.')
                    elif event.text.lower() == 'по группам':  # Changes by group
                        self.bot.sendMsg(vkid=event.user_id,
                                         msg="Введите код группы, для которой нужно найти изменения:")
                        self.writesearchgroup.append(event.user_id)
                    elif event.text.lower()[-3:] in TimeCatcher.getGroupList() and event.user_id in self.writesearchgroup:
                        self.bot.sendMsg(vkid=event.user_id, msg=Changes.makeChanges(event.text))
                        self.writesearchgroup.remove(event.user_id)
                    elif event.text in TimeCatcher.keyboardNumDays and event.user_id in self.writeweekday:
                        self.bot.sendMsg(vkid=event.user_id, msg=Changes.makeChanges(event.text))
                        self.writeweekday.remove(event.user_id)
                    elif event.text[-4:] == str(datetime.date.today().year) and event.user_id in self.writedate:
                        self.bot.sendMsg(vkid=event.user_id, msg=Changes.makeChanges(event.text))
                        self.writedate.remove(event.user_id)
                    elif event.text.lower() == 'рассылка':
                        if SQL.getUserGroup(vkid=event.user_id) is not None:
                            self.bot.sendMsg(vkid=event.user_id, msg=SQL.sendStatus(vkid=event.user_id))
                        else:
                            self.bot.sendMsg(vkid=event.user_id, msg='Укажите сначала вашу группу.')
                    else:
                        self.bot.sendMsg(vkid=event.user_id, msg="Данной команды не существует.")
            elif event.type == VkEventType.USER_TYPING:
                print(f"Пользователь {event.user_id} пишет.")  # Console msg when user typing something


class TimeCatcher:

    def __init__(self):
        self.keyboardNumDays = ['E', 'T', 'K', 'N', 'R', 'L', 'P']  # Days of the week in Estonian language
        self.dayOfWeek = {'E': 'Понедельник',
                          'T': 'Вторник',
                          'K': 'Среда',
                          'N': 'Четверг',
                          'R': 'Пятница',
                          'L': 'Суббота',
                          'P': "Воскресенье"}  # Days of the week in Russian langauage from Estonian
        self.datelist = []
        for i in range(5):  # Taking days of the week and dates for the next 5 days
            x = datetime.date.today() + datetime.timedelta(hours=2) + datetime.timedelta(days=i)
            self.datelist.append([self.keyboardNumDays[x.weekday()], x.day, x.month, x.year])
        for i in self.datelist:  # Dates in Estonia stated with zeros in single-digit numbers
            if i[1] < 10:
                i[1] = str(i[1])
                i[1] = '0' + i[1]
            if i[2] < 10:
                i[2] = str(i[2])
                i[2] = '0' + i[2]

    def todayWeekDay():  # Getting today's day of the week
        return (datetime.date.today() + datetime.timedelta(hours=2)).weekday()

    def getGroupList():  # Group list for 2017-2020 year
        groupList = []
        yearnow = datetime.date.today().year
        for i in range(int(str(yearnow)[:-2]) - 3, int(str(yearnow)[:-2]) + 1, 1):
            groupList.append('v' + str(i))
            groupList.append('e' + str(i))
        return groupList


class Keyboard:

    def __init__(self):
        # Default keyboard from start
        self.keyboard = VkKeyboard(one_time=False, inline=False)
        self.keyboard.add_button('Моя группа', color=VkKeyboardColor.PRIMARY)
        self.keyboard.add_button('В какой я группе?', color=VkKeyboardColor.POSITIVE)
        self.keyboard.add_line()
        self.keyboard.add_button('По датам', color=VkKeyboardColor.DEFAULT)
        self.keyboard.add_button('По дню недели', color=VkKeyboardColor.DEFAULT)
        self.keyboard.add_button('По группам', color=VkKeyboardColor.DEFAULT)
        self.keyboard.add_line()  # Переход на вторую строку
        self.keyboard.add_button('COVID-19', color=VkKeyboardColor.NEGATIVE)
        self.keyboard.add_button('Изменить группу', color=VkKeyboardColor.NEGATIVE)
        self.keyboard.add_button('Рассылка', color=VkKeyboardColor.DEFAULT)
        # Keyboard for next five days
        self.fiveDaysKeyboard = VkKeyboard(one_time=False, inline=True)
        for i in range(5):
            if i == 0:
                color = VkKeyboardColor.POSITIVE
            elif (TimeCatcher.datelist[i])[0] in ['L', 'P']:
                color = VkKeyboardColor.NEGATIVE
            else:
                color = VkKeyboardColor.DEFAULT
            if i > 0:
                self.fiveDaysKeyboard.add_line()
            self.fiveDaysKeyboard.add_button(
                f"{(TimeCatcher.datelist[i])[0]}: {(TimeCatcher.datelist[i])[1]}.{(TimeCatcher.datelist[i])[2]}.{(TimeCatcher.datelist[i])[3]}", color)
        # Keyboard with days of week
        self.weekDaysKeyboard = VkKeyboard(one_time=False, inline=True)
        for i in TimeCatcher.keyboardNumDays:
            if TimeCatcher.keyboardNumDays.index(i) == TimeCatcher.todayWeekDay():
                color = VkKeyboardColor.POSITIVE
            elif TimeCatcher.keyboardNumDays.index(i) in [5, 6]:
                color = VkKeyboardColor.NEGATIVE
            else:
                color = VkKeyboardColor.DEFAULT
            if i == 'R':
                self.weekDaysKeyboard.add_line()
            self.weekDaysKeyboard.add_button(i, color=color)


class Bot:

    def __init__(self, vk):
        self.vk = vk  # Getting VKApi options from server

    def sendMsg(self, vkid, msg):  # Sending message without keyboard
        self.vk.method('messages.send', {'user_id': vkid, 'random_id': get_random_id(), 'message': msg})

    def sendKeyboard(self, keyboard, vkid, msg):
        self.vk.method('messages.send', {'user_id': vkid, 'random_id': get_random_id(), 'message': msg,
                                         'keyboard': keyboard.get_keyboard()})  # Sending message with keyboard


class SQL:

    def __init__(self):
        pass

    def getConnection(self):
        mysql_l = os.environ['MYSQL_LOGIN']
        mysql_p = os.environ["MYSQL_PASS"]  # Getting login and password from service there bot is deployed
        return pymysql.connect(host='eu-cdbr-west-02.cleardb.net',
                               user=mysql_l,
                               password=mysql_p,
                               db='heroku_0ccfbccd1823b55',
                               cursorclass=DictCursor)  # Database connection settings

    def getUserGroup(vkid):
        conn = self.getConnection()
        with conn.cursor() as cursor:  # Getting user's group at school from database
            cursor.execute("SELECT `thkruhm` FROM `users` WHERE `vkid` = %s", (vkid,))
            row = cursor.fetchone()
            cursor.close()
            conn.close()
            if row is None:
                return None
            return row['thkruhm']

    def setUserGroup(vkid, group):
        conn = self.getConnection()
        usergroup = self.getUserGroup(vkid)
        with conn.cursor() as cursor:
            if usergroup is not None:  # If group currently is specified by user
                cursor.execute("UPDATE `users` SET `thkruhm`=%s WHERE `vkid`=%s", (group, vkid))
            else:  # If group isn't specified, user will be added to database
                cursor.execute("INSERT INTO `users`(`vkid`, `thkruhm`, `sendStatus`) VALUES (%s, %s, 1)", (vkid, group))
            conn.commit()
            cursor.close()
            conn.close()

    def sendStatus(vkid):
        conn = self.getConnection()
        with conn.cursor() as cursor:
            cursor.execute("SELECT sendStatus FROM users WHERE vkid = %s", (vkid,))
            row = cursor.fetchone()
            sendstatus = row['sendStatus']
            if sendstatus == 1:
                cursor.execute("UPDATE `users` SET `sendStatus`=0 WHERE vkid=%s", (vkid,))
                conn.commit()
                cursor.close()
                conn.close()
                return "Рассылка была успешно выключена."
            cursor.execute("UPDATE `users` SET `sendStatus`=1 WHERE `vkid`=%s", (vkid,))
            conn.commit()
            cursor.close()
            conn.close()
            return "Рассылка была успешно включена."


class Changes:
    def __init__(self):
        pass

    def parseChanges(self):
        r = requests.get('http://www.tthk.ee/tunniplaani-muudatused/')  # Schools's site
        html = r.text
        soup = BeautifulSoup(html, 'html.parser')
        table = soup.findChildren('table')
        changes = []
        for item in table:
            my_table = item
            rows = my_table.find_all('tr')
            for row in rows:
                change = []
                cells = row.find_all('td')
                for cell in cells:
                    if cell.text not in ["\xa0", "Kuupäev", "Rühm", "Tund", "Õpetaja",
                                         "Ruum"]:  # Rows, that we don't need
                        data = cell.text
                        change.append(data)
                if change != []:
                    changes.append(change)
        return changes

    def convertChanges(self, i, date):
        if date is False:
            if len(i) == 6:
                changeList = f"🦆 Группа: {i[2]} ⏰ Урок: {i[3]} \n👨‍🏫 Преподаватель: {i[4]}\nКабинет: {i[5]}\n"
            elif len(i) > 2 and i[3].lower() in "jääb ära":
                changeList = f"🦆 {i[2]}\n❌ Не состоится\n"
            elif len(i) > 4 and i[4].lower() in "jääb ära":
                changeList = f"🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n❌ Не состоится\n"
            elif len(i) > 4 and i[4].lower() in "söögivahetund":
                changeList = f"🦆 Группа: {i[2]}\n ⏰ Урок: {i[3]}\n🆒 Обеденный перерыв\n"
            elif len(i) > 5 and i[5].lower() in "iseseisev töö kodus":
                changeList = f"🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n🏠 Самостоятельная работа дома\n"
            elif len(i) > 5 and i[5].lower() in "iseseisev töö":
                changeList = f"🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n📋 Самостоятельная работа\n"
            elif len(i) > 5 and i[5].lower() in ["", " "]:
                changeList = f"🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n👨‍🏫 Преподаватель: {i[4]}\n"
            else:
                changeList = f"🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n"
        else:
            if len(i) == 6:
                changeList = 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":
                changeList = f"🗓 {i[0]} Дата: {i[1]}\n🦆 {i[2]}\n❌ Не состоится\n"
            elif len(i) > 4 and i[4].lower() in "jääb ära":
                changeList = f"🗓 {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n❌ Не состоится\n"
            elif len(i) > 4 and i[4].lower() in "söögivahetund":
                changeList = 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":
                changeList = f"🗓 {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n🏠 Самостоятельная работа дома\n"
            elif len(i) > 5 and i[5].lower() in "iseseisev töö":
                changeList = f"🗓 {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n📋 Самостоятельная работа\n"
            elif len(i) > 5 and i[5].lower() in ["", " "]:
                changeList = f"🗓 {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n👨‍🏫 Преподаватель: {i[4]}\n"
            else:
                changeList = f"🗓 В {i[0]} Дата: {i[1]}\n🦆 Группа: {i[2]} ⏰ Урок: {i[3]}\n"
        return changeList

    def makeChanges(self, data):
        changes = self.parseChanges()  # Changes in array from the school website
        changeList = []
        if data[-3:] in TimeCatcher.getGroupList():  # Group for 4 years (like 2017-2020)
            for line in changes:
                if data.lower() in line[2].lower():
                    changeList.append(
                        self.convertChanges(line, True))  # Takes converted lines of changes from makeChanges func
            if len(changeList) > 0:
                refChanges = f"Для группы 🦆 {data} на данный момент следующие изменения в расписании:\n"  # Head of the message
                for i in changeList:
                    refChanges += f"{i}\n"
                return refChanges
            return f"Для группы 🦆 {data} на данный момент изменений в расписании нет."
        if data[-4:] == str(datetime.date.today().year):
            data = re.split(r':\s', data)
            if len(data) > 1:
                data = data[1]
            else:
                data = data[0]
            for line in changes:
                if data.lower() in line[2].lower():
                    changeList.append(self.convertChanges(line, False))
            if len(changeList) > 0:
                refChanges = f"В учебном заведении на 🗓 {data} следующие изменения в расписании:\n"
                for i in changeList:
                    refChanges += f"{i}\n"
                return refChanges
            return f"В данный момент нет изменний в расписании на 🗓 {data}."
        if data in TimeCatcher.keyboardNumDays:
            for line in changes:
                if data.lower() in line[0].lower():
                    changeList.append(self.convertChanges(line, False))
            if len(changeList) > 0:
                refChanges = f"В учебном заведении на 🗓 {TimeCatcher.dayOfWeek[data]} следующие изменения в расписании:\n"
                for i in changeList:
                    refChanges += f"{i}\n"
                return refChanges
            return f"В данный момент изменений в расписании нет на 🗓 {TimeCatcher.dayOfWeek[data]}."
        return "Нет изменений в расписании по запросу, который вы ввели."


class COVID:
    def __init__(self):
        self.url = 'https://raw.githubusercontent.com/okestonia/koroonakaart/master/koroonakaart/src/data.json'
        # Link for JSON

    def getData(self):
        if self.url.lower().startswith('http'):
            r = requests.get(self.url)
            data = r.json()  # json module loads from the link
            covid = [data['confirmedCasesNumber'], data['testsAdministeredNumber'], data['recoveredNumber'],
                     data['deceasedNumber'], data['activeCasesNumber']]  # Getting correct rows.
            covid = f"🦠 COVID-19 в Эстонии:\n☣ {covid[0]} случаев заражения из 🧪 {covid[1]} тестов.\n😷 {covid[4]} " \
                    f"болеет на данный момент и 💉 {covid[2]} выздоровели\n☠ {covid[3]} человек умерло.\n"
            return covid
        raise ValueError from None

access_token = os.environ["ACCESS_TOKEN"]
server = Server(access_token)  # Access token for VKApi
server.start()