diff --git a/.gitignore b/.gitignore index 7c161b4..5b938af 100644 --- a/.gitignore +++ b/.gitignore @@ -188,8 +188,6 @@ pyvenv.cfg .venv pip-selfcheck.json -*/processed.txt -processed.txt +*.txt *.log *.session -*/*.session diff --git a/logs/.gitinclude b/logs/.gitinclude new file mode 100644 index 0000000..e69de29 diff --git a/main.py b/main.py index 365576a..88683c3 100644 --- a/main.py +++ b/main.py @@ -1,3 +1,4 @@ +import os import asyncio import logging @@ -7,6 +8,17 @@ from modules.telegram import TelegramStarsBot logger = logging.getLogger(__name__) +def setup_logging(): + logging.basicConfig( + level=logging.INFO, + format="%(asctime)s | %(levelname)s | %(name)s | %(message)s", + handlers=[ + logging.FileHandler("logs/bot.log", encoding="utf-8"), + logging.StreamHandler(), + ], + ) + + async def main(): try: config = Config() @@ -21,14 +33,8 @@ async def main(): if __name__ == "__main__": - logging.basicConfig( - level=logging.INFO, - format="%(asctime)s | %(levelname)s | %(name)s | %(message)s", - handlers=[ - logging.FileHandler("logs/bot.log", encoding="utf-8"), - logging.StreamHandler(), - ], - ) + os.makedirs("logs", exist_ok=True) + setup_logging() try: asyncio.run(main()) diff --git a/modules/telegram.py b/modules/telegram.py index e72bcf9..c5d3baf 100644 --- a/modules/telegram.py +++ b/modules/telegram.py @@ -46,7 +46,7 @@ class TelegramLinkExtractor: class TelegramStarsBot: - SESSION_NAME = "stars_bot" + SESSION_NAME = "storage/stars_bot" def __init__(self, config: Config): self.config = config @@ -237,7 +237,7 @@ class TelegramStarsBot: await asyncio.sleep(self.config.check_interval) async def start(self): - is_first_login = not os.path.exists(f"storage/{self.SESSION_NAME}.session") + is_first_login = not os.path.exists(f"{self.SESSION_NAME}.session") if is_first_login: logger.info("Сессия Telegram не найдена. Запускаю процесс входа...")