feat: refactor code for flake8

Author: OldTyT

clients/GPTGenerate.py

@@ -1,3 +1,4 @@
+"""GPT generate clients."""
 import time
 from threading import Thread
 
@@ -8,65 +9,63 @@
 from logger import logger
 from models.configs import GlobalConfigs
 
-
-STOP_TYPING = False
+stop_typing = False
 cfg = GlobalConfigs()
 bot = telebot.TeleBot(cfg.telegram_token_bot.get_secret_value())
 
-IGNORE_ERRORS = [
+IGNORE_ERRORS = [  # Hardcode, need refactor
     "A request to the Telegram API was unsuccessful. Error code: 400. Description: Bad Request: message is not modified: specified new message content and reply markup are exactly the same as a current content and reply markup of the message"  # noqa E501
 ]
 
 
 def set_is_typing(chat_id):
+    """Set bot status typing."""
     while True:
-        if STOP_TYPING:
+        if stop_typing:
             return
-        logger.debug(f"STOP_TYPING is False. Seting status typing. Chat id: {chat_id}")
+        logger.debug(f"stop_typing is False. Seting status typing. Chat id: {chat_id}")
         bot.send_chat_action(chat_id, action="typing")
         time.sleep(4)
 
 
 def get_full_len_list(my_list: list):
+    """Get full len list."""
     cnt = 0
     for i in my_list:
         cnt += len(i)
     return cnt
 
 
 def get_str_from_list(my_list: list):
-    return ''.join(str(x) for x in my_list)
+    """Get string from list."""
+    return "".join(str(x) for x in my_list)
 
 
 class GPT4TextGenerate(BaseSettings):
+    """GPT4TextGenerate class."""
 
     def message_responser(self, prompt, chat_id, msg_id):
+        """Set typing bot status and generate text."""
         if chat_id != 0:
             Thread(target=set_is_typing, args=(chat_id,)).start()
-        # try:
         result_generate = self.gpt4_generate(
-            prompt=prompt,
-            chat_id=chat_id,
-            msg_id=msg_id
+            prompt=prompt, chat_id=chat_id, msg_id=msg_id
         )
         if result_generate:
-            STOP_TYPING = True
+            stop_typing = True
             logger.debug(f"Result sent successful for chat id: {chat_id}")
             return True
-        # except Exception as e:
-        #     logger.error(f"Fatal error: {e}")
-        STOP_TYPING = True  # noqa F841
+        stop_typing = True  # noqa F841
         self.send_message("Smth error", chat_id)
         raise RuntimeError
 
-    def gpt4_generate(self, chat_id, prompt, msg_id):
+    def gpt4_generate(self, chat_id, prompt, msg_id):  # noqa: CFQ004,CFQ001
+        """Generate text with prompt and send."""
         prompt = cfg.pre_prompt + prompt
         response = g4f.ChatCompletion.create(
-            # model='gpt-3.5-turbo',
-            model='gpt-4',
+            model="gpt-4",
             messages=[{"role": "user", "content": prompt}],
-            # stream=True,
-            provider=g4f.Provider.ChatgptAi
+            provider=g4f.Provider.ChatgptAi,
         )
         full_text = []
         max_length = 150
@@ -84,7 +83,7 @@ def gpt4_generate(self, chat_id, prompt, msg_id):
                         return True
                     msg_id = self.send_message(
                         text=get_str_from_list(full_text[last_len_list:]),
-                        chat_id=chat_id
+                        chat_id=chat_id,
                     )
                     len_list = len(full_text)
                     cnt_divisions += 1
@@ -94,13 +93,13 @@ def gpt4_generate(self, chat_id, prompt, msg_id):
                             bot.edit_message_text(
                                 chat_id=chat_id,
                                 text=get_str_from_list(full_text[len_list:]),
-                                message_id=msg_id
+                                message_id=msg_id,
                             )
                             last_len_list = len(full_text)
                             if chat_id == 0:
                                 bot.edit_message_text(
                                     inline_message_id=msg_id,
-                                    text=get_str_from_list(full_text[len_list:])
+                                    text=get_str_from_list(full_text[len_list:]),
                                 )
                         except telebot.apihelper.ApiTelegramException as e:
                             if str(e) in IGNORE_ERRORS:
@@ -113,27 +112,23 @@ def gpt4_generate(self, chat_id, prompt, msg_id):
                     bot.edit_message_text(
                         inline_message_id=msg_id,
                         text=get_str_from_list(full_text[len_list:]),
-                        # parse_mode='Markdown'
-                        # Added markdown validator
                     )
                     return True
                 bot.edit_message_text(
                     chat_id=chat_id,
                     text=get_str_from_list(full_text[len_list:]),
                     message_id=msg_id,
-                    # parse_mode='Markdown'
-                    # Added markdown validator
                 )
         except telebot.apihelper.ApiTelegramException as e:
             if str(e) in IGNORE_ERRORS:
                 return True
-            else:
-                raise RuntimeError(e)
+            raise RuntimeError(e)
         return True
 
     def send_message(self, text, chat_id):
+        """Send message with chunk per 4096 sumbol."""
         max_length = 4096
-        chunks = [text[i:i+max_length] for i in range(0, len(text), max_length)]
+        chunks = [text[i: i + max_length] for i in range(0, len(text), max_length)]
         msg_id = 0
         try:
             for chunk in chunks:
@@ -144,5 +139,4 @@ def send_message(self, text, chat_id):
         except telebot.apihelper.ApiTelegramException as e:
             if str(e) in IGNORE_ERRORS:
                 return msg_id
-            else:
-                raise RuntimeError(e)
+            raise RuntimeError(e)

clients/middleware.py

@@ -1,12 +1,13 @@
-import json
+# Need refactor.
 import asyncio
+import json
 
-from sqlalchemy import select
 from pydantic import BaseSettings
+from sqlalchemy import select
 
-from models.history import MessageHistory, CallbackQueryHistory, Chats
 from db.base import get_session
 from logger import logger
+from models.history import CallbackQueryHistory, Chats, MessageHistory
 
 
 class UpdateLastTimeMessage(BaseSettings):
@@ -24,11 +25,7 @@ async def update_time(self, time, chat_id: int):
             chat.message_last_time = time
             chat.message_count += 1
         else:
-            chat = Chats(
-                chat_id=chat_id,
-                message_last_time=time,
-                message_count=1
-            )
+            chat = Chats(chat_id=chat_id, message_last_time=time, message_count=1)
             session.add(chat)
         try:
             await session.commit()
@@ -71,12 +68,14 @@ def save_qc(self, callback_query):
     async def save_cq(self, callback_query):
         session = [session_q async for session_q in get_session()][0]
         logger.debug(callback_query)
-        callback_query_h = CallbackQueryHistory(callback_query=json.loads(str(callback_query)))
+        callback_query_h = CallbackQueryHistory(
+            callback_query=json.loads(str(callback_query))
+        )
         session.add(callback_query_h)
         try:
             await session.commit()
             return True
         except Exception as ex:
             await session.rollback()
-            logger.error(ex)
+            logger.error(f"Error in func CallbackQuerySaver: {ex}")
             return False