2.2 Мелкие улучшения и аннотации файлы

SQLite3/bd_func/__init__.py

@@ -14,7 +14,7 @@ from .status_user import *
 
 
 # Основная обработка SQL
-async def base_sql(message: types.Message):
+async def base_sql(message: types.Message) -> None:
     tg_id = message.from_user.id
     usernames = username(message)
     first_name = message.from_user.first_name

SQLite3/bd_func/bd_add_user.py

@@ -10,7 +10,7 @@ __all__ = ("add_user",)
 
 # Функция добавления пользователя с последовательным user_id
 async def add_user(tg_id: int, username: str, first_name: str, last_name: str,
-                   role: str, status: str, user: str, bd_name: str = BotVar.bd_names):
+                   role: str, status: str, user: str, bd_name: str = BotVar.bd_names) -> None:
     with sqlite3.connect(bd_name) as db:
         cursor = db.cursor()
 

SQLite3/bd_func/bd_get_user.py

@@ -9,7 +9,7 @@ __all__ = ("get_user",)
 
 
 # Функция для получения данных о пользователе
-async def get_user(tg_id: int, bd_name: str = BotVar.bd_names):
+async def get_user(tg_id: int, bd_name: str = BotVar.bd_names) -> None:
     with sqlite3.connect(bd_name) as db:
         cursor = db.cursor()
         cursor.execute("SELECT * FROM users WHERE tg_id = ?", (tg_id,))

SQLite3/bd_func/bd_update_user.py

@@ -10,7 +10,7 @@ __all__ = ("update_user",)
 
 # Функция обновления пользователя
 async def update_user(tg_id: int, username: str = None, first_name: str = None, last_name: str = None,
-                      bd_name: str = BotVar.bd_names, role: str = None, user: str = None):
+                      bd_name: str = BotVar.bd_names, role: str = None, user: str = None) -> None:
     updates = []
     params = []
 

SQLite3/bd_func/bd_update_user_msg.py

@@ -13,7 +13,7 @@ __all__ = ("update_user_messages",)
 
 
 # Функция обновления статистики сообщений пользователя
-async def update_user_messages(message: types.Message, bd_name: str = BotVar.bd_names):
+async def update_user_messages(message: types.Message, bd_name: str = BotVar.bd_names) -> None:
     with sqlite3.connect(bd_name) as db:
         cursor = db.cursor()
         user_id = message.from_user.id  # Используем user_id напрямую

SQLite3/bd_func/bd_user_create.py

@@ -9,7 +9,7 @@ __all__ = ("create_user_db",)
 
 
 # Функция создания базы данных
-async def create_user_db(bd_name: str = BotVar.bd_names):
+async def create_user_db(bd_name: str = BotVar.bd_names) -> None:
     with sqlite3.connect(bd_name) as db:
         cursor = db.cursor()
 

SQLite3/bd_func/status_user.py

@@ -6,7 +6,7 @@ from aiogram import types
 from ProjectsFiles import BotVar
 
 # Функция проверки статуса пользователя
-async def status_user(message: types.Message, bd_path: str = BotVar.bd_names):
+async def status_user(message: types.Message, bd_path: str = BotVar.bd_names) -> str:
     # Подключение к базе данных
     bd = sqlite3.connect(bd_path)
     tg_id = message.from_user.id
@@ -30,9 +30,9 @@ async def status_user(message: types.Message, bd_path: str = BotVar.bd_names):
 
     if row:
         user_type = row[0]  # предполагаем, что в столбце 'user' находится только одно значение
-        status = status_map.get(user_type, "Ошибка!")  # Получаем статус или "Ошибка!"
+        status: str = status_map.get(user_type, "Ошибка!")  # Получаем статус или "Ошибка!"
     else:
-        status = "Пользователь не найден"
+        status: str = "Пользователь не найден"
 
     # Закрываем соединение с базой данных
     bd.close()