Use session for bot

borisevich-a-v · Dec 25, 2024 · 62632a6 · 62632a6
1 parent 29a2d23
commit 62632a6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/src/aggregator/bot/create_bot.py b/src/aggregator/bot/create_bot.py
@@ -7,7 +7,8 @@
 from typing_extensions import NamedTuple
 
 from aggregator.bot.warden.warden import NotAllowed, Warden
-from aggregator.config import ADMIN, AGGREGATOR_CHANNEL, TELEGRAM_API_HASH, TELEGRAM_API_ID, TELEGRAM_BOT_TOKEN
+from aggregator.config import ADMIN, AGGREGATOR_CHANNEL, TELEGRAM_API_HASH, TELEGRAM_API_ID, TELEGRAM_BOT_TOKEN, \
+    BOT_SESSION
 from aggregator.posts_storage import NoNewPosts, PostStorage
 from aggregator.telegram_slow_client import TelegramSlowClient
 
@@ -45,9 +46,7 @@ def get_request_pattern(post_storage: PostStorage) -> re.Pattern:
 
 def create_bot(post_storage: PostStorage, warden: Warden) -> TelegramClient:
     logger.info("Creating bot")
-    bot = TelegramSlowClient(StringSession(), TELEGRAM_API_ID, TELEGRAM_API_HASH, min_request_interval=0.005).start(
-        bot_token=TELEGRAM_BOT_TOKEN
-    )
+    bot = TelegramSlowClient(StringSession(BOT_SESSION), TELEGRAM_API_ID, TELEGRAM_API_HASH, min_request_interval=0.005)
 
     # TODO: should we parse album here? nea
     @bot.on(events.NewMessage(chats=AGGREGATOR_CHANNEL))

diff --git a/src/aggregator/config.py b/src/aggregator/config.py
@@ -26,6 +26,8 @@ def load_env_list(env_name: str) -> list[str]:
 TELEGRAM_API_HASH = str(os.environ.get("TELEGRAM_API_HASH"))
 
 CLIENT_SESSION = str(os.environ.get("CLIENT_SESSION"))
+BOT_SESSION = str(os.environ.get("BOT_SESSION"))
+
 DB_CONNECTION_STRING = str(os.environ.get("DB_CONNECTION_STRING"))
 
 ADMIN = int(os.environ.get("ADMIN", 0))

diff --git a/utilities/generate_session.py b/utilities/generate_session.py
@@ -17,7 +17,7 @@ def generate_bot_session():
     print("Generating bot session")
     with TelegramClient(StringSession(), TELEGRAM_API_ID, TELEGRAM_API_HASH).start(bot_token=TELEGRAM_BOT_TOKEN) as bot:
         print("Bot session:")
-        print(bot.sesson.save())
+        print(bot.session.save())
 
 
 if len(sys.argv) != 2: