MySQL no longer supported, antispam thing remaded

README.md

@@ -43,7 +43,7 @@
 
 Теперь заходим в директорию бота (`cd dan63047pythonbot/`) и устанавливаем необходимые библиотеки:
 
-`pip3 install vk-api pyowm Wikipedia-API PyMySQL --user`
+`pip3 install vk-api pyowm Wikipedia-API --user`
 
 Эта команда установит все необходимые библиотеки для работы бота.
 
@@ -58,18 +58,11 @@ admins = [276193568, 629085267] # Список ид аккаунтов адми
 vk_service_token = "vk_service_token" # Сервисный ключ доступа для команды !image. Оставьте None, если хотите выключить эту команду
 album_for_command = 269199619 # Цифровой ид альбома для команды !image. Оставьте None, если хотите выключить эту команду
 openweathermap_api_key = 'openweathermap_api_key' # Ключ OpenWeather API. Оставьте None, если хотите выключить команду !weather
-use_database = True # False, если вы не планируете использовать MySQL базу данных, иначе следующие 4 строки обязательны
-mysql_host = 'pythonanywhere_nickname.mysql.pythonanywhere-services.com' # ссылка на хост БД. Вместо pythonanywhere_nickname должен быть ваш никнейм на pythonanywhere
-mysql_user = 'pythonanywhere_nickname' # Ваш никнейм на pythonanywhere
-mysql_pass = 'qwerty123' # Пароль, созданный вами для БД
-mysql_db = 'pythonanywhere_nickname$default' # Название вашей БД. Вместо pythonanywhere_nickname должен быть ваш никнейм на pythonanywhere. Так же вместо default может быть ваше название БД
 ```
 ## Если вы будете использовать команду !image
 Разберёмся с костылём для метода `photos.get`: Идем на `vk.com/dev`, нажимаем на *Мои приложения* и кликаем по *Создать приложение*. Новое приложение может называться как угодно, от него нам необходимо всего лишь сервисный ключ доступа, который можно найти, если нажать на *Настройки*
 ## Если вы будете использовать команду !weather
 Перейдите по `openweathermap.org`, пройдите там регистрацию и создайте API ключ
-## Если вы решите использовать MySQL БД (давно не тестировалось, может не работать)
-Вам необходимо будет купить тариф на pythonanywhere, где есть поддержка MySQL базы данных, с которой бот будет работать. Тыкаем на *Databases* в шапке сайта. Создаём пароль для вашей базы данных. По умолчанию будет создана база данных с названием `your_nickname$default`, но вы можете создать БД со своим названием.
 ## Запуск бота
 Если аккаунт платный: в pythonanywhere тыкаем на *Tasks*, создаём и запускаем в *Always-on tasks* задачу `python3.8 /home/pythonanywhere_nickname/dan63047pythonbot/longpulling.py`, где вместо pythonanywhere_nickname должен быть ваш никнейм на pythonanywhere. После небольшого тупления pythonanywhere запустит вашего бота.
 
@@ -94,9 +87,7 @@ mysql_db = 'pythonanywhere_nickname$default' # Название вашей БД.
 * [vk_api](https://github.com/python273/vk_api) — модуль для создания скриптов для социальной сети Вконтакте
 * [pyowm](https://github.com/csparpa/pyowm) — модуль для получения погоды через OpenWeather API
 * [Wikipedia-API](https://github.com/martin-majlis/Wikipedia-API) — модуль для получения статей из Wikipedia
-* [PyMySQL](https://github.com/PyMySQL/PyMySQL/) — Pure Python MySQL Client
 # Дополнительно
-
 Автор бота: [Даниил Михайлов](https://vk.com/dan63047)
 
 Буду рад помощи и поддержке

dan63047VKbot.py

@@ -43,6 +43,7 @@ def log(warning, text):
         print(msg)
 
 bot = {}
+SPAMMER_LIST = {}
 errors_array = {"access": "Отказано в доступе",
                 "miss_argument": "Отсуствует аргумент", 
                 "command_off": "Команда отключена",
@@ -94,156 +95,61 @@ except Exception:
 class Database_worker():
 
     def __init__(self):
-        if config.use_database:
-            log(False, "Trying to connect to database")
-            try:
-                self._CON = pymysql.connect(
-                    host=config.mysql_host,
-                    user=config.mysql_user,
-                    password=config.mysql_pass,
-                    db=config.mysql_db,
-                    charset='utf8mb4',
-                    cursorclass=DictCursor
-                )
-                cur = self._CON.cursor()
-            except Exception as e:
-                debug_array['db_warnings'] += 1
-                log(True, f"Unable to connect to database: {str(e)}")
-            try:
-                cur.execute("SELECT * FROM bot_users")
-            except:
-                cur.execute("CREATE TABLE bot_users ( id INT UNSIGNED NOT NULL AUTO_INCREMENT PRIMARY KEY, chat_id INT UNSIGNED, awaiting VARCHAR(128), access TINYINT, midnight BOOL, new_post BOOL, admin_mode BOOL, game_wins INT UNSIGNED, game_defeats INT UNSIGNED, game_draws INT UNSIGNED, banned BOOL)")
-            cur.close()
-            log(False, "Database connection established")
-        else:
-            log(False, "Bot will use JSON file as database")
         try:
             with open("data.json", "r") as data:
                 self._DATA_DIST = json.load(data)
                 data.close()
         except Exception:
-                log(True, "data.json is not exist, it will be created soon")
+            log(True, "data.json is not exist, it will be created")
-                self._DATA_DIST = {"users": {}}
+            self._DATA_DIST = {"users": {}, "spammers": []}
+            open("data.json", "w").write(json.dumps(self._DATA_DIST))
 
     def set_new_user(self, peer_id, midnight=False, awaiting=None, access=1, new_post=False, admin_mode=False, game_wins=0, game_defeats=0, game_draws=0, banned=False):
-        if(config.use_database):
+        self._DATA_DIST['users'][peer_id] = {"awaiting": awaiting, "access": access, "midnight": midnight, "new_post": new_post, "admin_mode": admin_mode, "game_wins": game_wins, "game_defeats": game_defeats, "game_draws": game_draws, "banned": banned}
-            try:
-                cur = self._CON.cursor()
-                cur.execute("INSERT INTO bot_users (chat_id, awaiting, access, midnight, new_post, admin_mode, game_wins, game_defeats, game_draws, banned) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)",
-                            (peer_id, awaiting, access, midnight, new_post, admin_mode, game_wins, game_defeats, game_draws, banned))
-                self._CON.commit()
-                cur.close()
-            except Exception as e:
-                debug_array['db_warnings'] += 1
-                log(True, f"Unable to add new user in database: {str(e)}")
-        else:
-            self._DATA_DIST['users'][peer_id] = {"awaiting": awaiting, "access": access, "midnight": midnight, "new_post": new_post,
-                                                 "admin_mode": admin_mode, "game_wins": game_wins, "game_defeats": game_defeats, "game_draws": game_draws, "banned": banned}
         open("data.json", "w").write(json.dumps(self._DATA_DIST))
 
     def get_all_users(self):
-        if not config.use_database:
         with open("data.json", "r") as data:
             self._DATA_DIST = json.load(data)
             data.close()
         return self._DATA_DIST['users']
-        try:
-            cur = self._CON.cursor()
-            cur.execute("SELECT * FROM bot_users")
-            result = cur.fetchall()
-            cur.close()
-            return result
-        except Exception as e:
-            debug_array['db_warnings'] += 1
-            log(True, f"Unable to load user from database: {str(e)}")
 
     def get_from_users(self, from_id):
-        if not config.use_database:
         with open("data.json", "r") as data:
             self._DATA_DIST = json.load(data)
             data.close()
         if not self._DATA_DIST['users'].get(str(from_id)):
             self.set_new_user(str(from_id))
         return self._DATA_DIST['users'][str(from_id)]
-        try:
-            cur = self._CON.cursor()
-            cur.execute(
-                "SELECT * FROM bot_users WHERE chat_id = %s", (from_id))
-            result = cur.fetchall()
-            cur.close()
-            return result
-        except Exception as e:
-            debug_array['db_warnings'] += 1
-            log(True, f"Unable to load user from database: {str(e)}")
 
     def get_game_stat(self):
-        if not config.use_database:
         return self._DATA_DIST['users']
-        try:
-            cur = self._CON.cursor()
-            cur.execute(
-                "SELECT chat_id, game_wins, game_draws, game_defeats FROM bot_users WHERE game_wins > 0 OR game_draws > 0 OR game_defeats > 0")
-            result = cur.fetchall()
-            cur.close()
-            return result
-        except Exception as e:
-            debug_array['db_warnings'] += 1
-            log(True, f"Unable to load stats from database: {str(e)}")
-            # Info: dist cannot return only the necessary keys
 
     def update_user(self, chat_id, thing, new_value):        
-        if(config.use_database):
-            try:
-                cur = self._CON.cursor()
-                cur.execute(f"UPDATE bot_users SET {thing} = %s WHERE bot_users.chat_id = %s;", (new_value, chat_id))
-                self._CON.commit()
-                cur.close()
-            except Exception as e:
-                debug_array['db_warnings'] += 1
-                log(True, f"Unable to update info about user in database: {str(e)}")
-        else:
-            if not self._DATA_DIST['users'][str(chat_id)].get(thing):
-                if thing == "spam_list":
-                    self._DATA_DIST['users'][str(chat_id)][thing] = []
         self._DATA_DIST['users'][str(chat_id)][thing] = new_value
         open("data.json", "w").write(json.dumps(self._DATA_DIST))
 
     def delete_user(self, chat_id):
-        if(config.use_database):
-            try:
-                cur = self._CON.cursor()
-                cur.execute(
-                    "DELETE FROM bot_users, game_defeats WHERE chat_id = %s", (chat_id))
-                self._CON.commit()
-                cur.close()
-                return True
-            except Exception as e:
-                debug_array['db_warnings'] += 1
-                log(True, f"Unable to delete user from database: {str(e)}")
-                return False
-        else:
         self._DATA_DIST['users'].pop(str(chat_id))
         open("data.json", "w").write(json.dumps(self._DATA_DIST))
 
+    def add_spammer(self, user_id):
+        SPAMMER_LIST.append(int(user_id))
+        self._DATA_DIST["spammers"].append(int(user_id))
+        open("data.json", "w").write(json.dumps(self._DATA_DIST))
+
+    def remove_spammer(self, user_id):
+        SPAMMER_LIST.pop(int(user_id))
+        self._DATA_DIST["spammers"].pop(int(user_id))
+        open("data.json", "w").write(json.dumps(self._DATA_DIST))
+
+    def read_spammers(self):
+        return self._DATA_DIST["spammers"]
+
 
 db = Database_worker()
 
 
-def load_users():
-    try:
-        log(False, "Reading database")
-        get_info = db.get_all_users()
-        for i in get_info:
-            if config.use_database:
-                bot[int(i['chat_id'])] = VkBot(int(i['chat_id']), bool(i['midnight']), i['awaiting'], int(
-                    i['access']), bool(i['new_post']), bool(i['admin_mode']), bool(i['banned']))
-            else:
-                bot[int(i)] = VkBot(int(i), bool(get_info[i]['midnight']), get_info[i]['awaiting'], int(
-                    get_info[i]['access']), bool(get_info[i]['new_post']), bool(get_info[i]['admin_mode']), bool(get_info[i]['banned']))
-    except Exception as lol:
-        debug_array['bot_warnings'] += 1
-        log(True, f"Problem with creating objects: {str(lol)}")
-
 
 def toFixed(numObj, digits=0):
     return f"{numObj:.{digits}f}"
@@ -306,7 +212,6 @@ class VkBot:
         """
         log(False, f"[BOT_{peer_id}] Created new bot-object")
         self._CHAT_ID = peer_id
-        self._SPAMMER_LIST = {}
         self._AWAITING_INPUT_MODE = awaiting
         self._ACCESS_TO_ALL = access
         self._MIDNIGHT_EVENT = midnight
@@ -375,10 +280,8 @@ class VkBot:
                 action = event.message.action
                 if action['type'] == 'chat_invite_user' or action['type'] == 'chat_invite_user_by_link':
                     user_info = vk.method('users.get', {'user_ids': action["member_id"], 'fields': 'verified,last_seen,sex'})
-                    chat_info = db.get_from_users(int(self._CHAT_ID))
+                    if int(action["member_id"]) in SPAMMER_LIST:
-                    if chat_info.get("spam_list"):
+                        self.send(f'[id{action["member_id"]}|Данный пользователь] находится в антиспам базе. Исключаю...')
-                        if int(action["member_id"]) in chat_info["spam_list"]:
-                            self.send(f'Привет, {user_info[0]["first_name"]}. К сожалению, администраторы этой беседы признали тебя спамером, поэтому мне придётся выгнать тебя отсюда')
                         vk.method("messages.removeChatUser", {"chat_id": int(self._CHAT_ID)-2000000000, "member_id": action["member_id"]})
                         return
                     self.send(f'Добро пожаловать в беседу, {user_info[0]["first_name"]} {user_info[0]["last_name"]}')
@@ -386,6 +289,10 @@ class VkBot:
                     pass
                     # user_info = vk.method('users.get', {'user_ids': action["member_id"], 'fields': 'verified,last_seen,sex'})
                     # self.send(f'{user_info[0]["first_name"]} {user_info[0]["last_name"]} покинул беседу')
+            if event.message.peer_id > 2000000000 and int(user_id) in SPAMMER_LIST:
+                self.send(f'[id{user_id}|Данный пользователь] находится в антиспам базе. Исключаю...')
+                vk.method("messages.removeChatUser", {"chat_id": int(self._CHAT_ID)-2000000000, "member_id": user_id})
+                return
         if self._AWAITING_INPUT_MODE:
             if message == "Назад":
                 self.change_await()
@@ -645,14 +552,10 @@ class VkBot:
                         message = message[1].split(' ', 1)
                         victum = re.search(r'id\d+', message[1])
                         victum = victum[0][2:]
-                        if db.get_from_users(int(self._CHAT_ID)).get("spam_list"):
+                        if message[0] == "add" or message[0] == "добавить":
-                            chat_spammers_list = db.get_from_users(int(self._CHAT_ID))["spam_list"]
-                        else:
-                            chat_spammers_list = []
-                        if message[0] == "add" or message [0] == "добавить":
                             if int(victum) != int(config.owner_id):
-                                if int(victum) not in chat_spammers_list:
+                                if int(victum) not in SPAMMER_LIST:
-                                    chat_spammers_list.append(int(victum))
+                                    db.add_spammer(int(victum))
                                     respond["text"] = "Теперь он считается спамером"
                                     log(False, f"[BOT_{self._CHAT_ID}] user {victum} added to spammer list")
                                 else:
@@ -661,15 +564,14 @@ class VkBot:
                                 log(False, f"[BOT_{self._CHAT_ID}] can't add to spammer list owner")
                         elif message[0] == "remove" or message[0] == "удалить":
                             if int(victum) != int(config.owner_id):
-                                if int(victum) in chat_spammers_list:
+                                if int(victum) in SPAMMER_LIST:
-                                    chat_spammers_list.pop(int(victum))
+                                    db.remove_spammer(int(victum))
                                     respond["text"] = "Теперь он не считается спамером"
                                     log(False, f"[BOT_{self._CHAT_ID}] user {victum} removed to spammer list")
                                 else:
                                     respond["text"] = "Его нет в этой базе"  
                             else:
                                 log(False, f"[BOT_{self._CHAT_ID}] can't restore owner")
-                        db.update_user(int(self._CHAT_ID), "spam_list", chat_spammers_list)
                     except IndexError:
                         respond['text'] = errors_array["miss_argument"]
                     except Exception as e:
@@ -758,14 +660,7 @@ class VkBot:
             return answer
 
     def game(self, thing, user_id):
-        data = db.get_from_users(user_id)
+        d = db.get_from_users(user_id)
-        if (config.use_database):
-            if len(data) == 0:
-                create_new_bot_object(user_id)
-                data = db.get_from_users(user_id)
-            d = data[0]
-        else:
-            d = data
         if thing == "статистика":
             try:
                 winrate = (d['game_wins']/(d['game_wins'] +

longpulling.py

@@ -41,17 +41,19 @@ def bots():
                     log_msg += atch
                 dan63047VKbot.log(False, log_msg)
                 dan63047VKbot.debug_array['messages_get'] += 1
-                if int(event.message.peer_id) not in dan63047VKbot.bot:
+                if event.message.peer_id not in dan63047VKbot.bot:
+                    u = dan63047VKbot.db.get_all_users()
+                    if str(event.message.peer_id) not in u:
                         dan63047VKbot.create_new_bot_object(event.message.peer_id)
+                    else:
+                        i = dan63047VKbot.db.get_from_users(event.message.peer_id)
+                        dan63047VKbot.bot[event.message.peer_id] = dan63047VKbot.VkBot(event.message.peer_id, bool(i['midnight']), i['awaiting'], int(i['access']), bool(i['new_post']), bool(i['admin_mode']), bool(i['banned']))
                 dan63047VKbot.bot[event.message.peer_id].get_message(event)
             elif event.type == dan63047VKbot.VkBotEventType.WALL_POST_NEW:
                 if event.object.post_type == "post":
                     dan63047VKbot.log(False, f"[NEW_POST] id{event.object.id}")
                     users = dan63047VKbot.db.get_all_users()
                     for i in users:
-                        if (config.use_database):
-                            dan63047VKbot.bot[int(i['chat_id'])].event("post", event.object)
-                        else:
                         dan63047VKbot.bot[int(i)].event("post", event.object)
                 else:
                     dan63047VKbot.log(False, f"[NEW_OFFER] id{event.object.id}")
@@ -75,9 +77,6 @@ def midnight():
             dan63047VKbot.log(False, "[EVENT_STARTED] \"Midnight\"")
             users = dan63047VKbot.db.get_all_users()
             for i in users:
-                if (config.use_database):
-                    dan63047VKbot.bot[int(i['chat_id'])].event("midnight")
-                else:
                 dan63047VKbot.bot[int(i)].event("midnight")
             dan63047VKbot.log(False, "[EVENT_ENDED] \"Midnight\"")
             time.sleep(1)
@@ -85,7 +84,7 @@ def midnight():
             time.sleep(0.50)
 
 
-dan63047VKbot.load_users()
+dan63047VKbot.SPAMMER_LIST = dan63047VKbot.db.read_spammers()
 tread_bots = threading.Thread(target=bots)
 tread_midnight = threading.Thread(target=midnight, daemon=True)
 tread_bots.start()