chore: add __all__ for explicit module exports

Author: brandonzorn

tfinance/blast.py

@@ -36,3 +36,9 @@ async def daily(update: Update, _):
     else:
         await update.message.reply_text("Ежедневная рассылка включена")
     db.user_daily_notify(user)
+
+
+__all__ = [
+    "daily",
+    "notify_assignees",
+]

tfinance/database.py

@@ -353,3 +353,8 @@ def add_point(self, user: User):
         )
         user.points += 1
         self.con.commit()
+
+
+__all__ = [
+    "Database",
+]

tfinance/exceptions.py

@@ -14,3 +14,11 @@ class EmptyDataFrameError(Exception):  # Пустой дата фрейм
 
 class WrongPeriodError(Exception):  # Неверный период
     pass
+
+
+__all__ = [
+    "StockSelectedAlready",
+    "PredictionAlreadySet",
+    "EmptyDataFrameError",
+    "WrongPeriodError",
+]

tfinance/functions.py

@@ -13,3 +13,8 @@ def create_user(update: Update) -> User:
         user_data["language_code"],
         user_data["is_bot"],
     )
+
+
+__all__ = [
+    "create_user",
+]

tfinance/game.py

@@ -223,3 +223,9 @@ async def game_results(context: CallbackContext):
     },
     fallbacks=[CommandHandler("game", game_menu)],
 )
+
+
+__all__ = [
+    "game_handler",
+    "game_results",
+]

tfinance/graphics/visualize.py

@@ -63,3 +63,9 @@ def check_stock_prices(stock_name) -> bool:
     curr_price = stock["Close"][-1]
 
     return curr_price > prev_price
+
+
+__all__ = [
+    "check_stock_prices",
+    "do_stock_image",
+]

tfinance/main.py

@@ -26,6 +26,8 @@
 from stock import check_stock, get_all_stocks, load_stocks
 from config import BOT_TOKEN, TIMEZONE
 
+__all__ = []
+
 # Запускаем логирование
 Path("logs").mkdir(exist_ok=True)
 logging.basicConfig(

tfinance/models.py

@@ -21,3 +21,8 @@ def __init__(
         self.prediction = prediction
         self.selected_stock = selected_stock
         self.points: int = 0
+
+
+__all__ = [
+    "User",
+]

tfinance/stock.py

@@ -60,3 +60,10 @@ def get_all_stocks():
         } for i in stocks
     ]
     save_stocks("stocks.json", stocks)
+
+
+__all__ = [
+    "check_stock",
+    "get_all_stocks",
+    "load_stocks",
+]