refactor: move game_handler to game.py

Author: brandonzorn

tfinance/game.py

@@ -1,5 +1,15 @@
-from telegram import InlineKeyboardButton, InlineKeyboardMarkup, Update
-from telegram.ext import ContextTypes, CallbackContext
+from telegram import (
+    InlineKeyboardButton,
+    InlineKeyboardMarkup,
+    Update,
+)
+from telegram.ext import (
+    ContextTypes,
+    CallbackContext,
+    ConversationHandler,
+    CommandHandler,
+    CallbackQueryHandler,
+)
 
 from database import Database
 from exceptions import (
@@ -201,3 +211,15 @@ async def game_results(context: CallbackContext):
         # Удаляем пройденные прогнозы
         db.delete_predictions(user)
         user.prediction = db.get_predictions(user)
+
+
+game_handler = ConversationHandler(
+    entry_points=[CommandHandler("game", game_menu)],
+    states={
+        1: [
+            CallbackQueryHandler(higher_game, pattern="^1$"),
+            CallbackQueryHandler(lower_game, pattern="^2$"),
+        ],
+    },
+    fallbacks=[CommandHandler("game", game_menu)],
+)

tfinance/main.py

@@ -1,5 +1,5 @@
-# -------------------------- Основной файл приложения -------------------------- #
-# --------------- Импорт необходимых библиотек, функций, классов --------------- #
+# ------------------------ Основной файл приложения ------------------------ #
+# ------------- Импорт необходимых библиотек, функций, классов ------------- #
 # Встроенные библиотеки.
 import datetime
 import logging
@@ -9,9 +9,7 @@
 
 # Работа с telegram-bot-api.
 from telegram.ext import (
-    CallbackQueryHandler,
     CommandHandler,
-    ConversationHandler,
     Application,
     ContextTypes,
 )
@@ -23,7 +21,7 @@
 from database import Database
 from exceptions import EmptyDataFrameError, WrongPeriodError
 from functions import create_user
-from game import game_menu, game_results, higher_game, lower_game
+from game import game_handler, game_results
 from graphics.visualize import do_stock_image
 from stock import check_stock, get_all_stocks, load_stocks
 from config import BOT_TOKEN, TIMEZONE
@@ -218,16 +216,6 @@ def main():
         ),
     )
 
-    game_handler = ConversationHandler(
-        entry_points=[CommandHandler("game", game_menu)],
-        states={
-            1: [
-                CallbackQueryHandler(higher_game, pattern="^1$"),
-                CallbackQueryHandler(lower_game, pattern="^2$"),
-            ],
-        },
-        fallbacks=[CommandHandler("game", game_menu)],
-    )
     application.add_handler(game_handler)
 
     application.add_handler(CommandHandler("daily", daily))