Feature/func 6

.gitignore

@@ -131,7 +131,19 @@ dmypy.json
 # List of user ids & answers
 *_list.txt
 *_response.txt
+*_file.txt
 *_patterns.txt
 
 # Session journals
-*.session*
+*.session*
+
+# SQLite
+*.db
+
+# Bucket objects
+birthdays/
+*.jpg
+*.png
+
+# Dynaconf
+settings.toml

Containerfile

@@ -2,7 +2,8 @@ FROM python:3.9-alpine as builder
 
 WORKDIR /opt
 
-COPY ["bot.py", "/opt/"]
+ADD ["/src/", "/opt/"]
+
 COPY requirements.txt requirements.txt
 
 RUN pip3 install --no-cache-dir -r requirements.txt

docker-compose.yaml

@@ -5,15 +5,20 @@ services:
     container_name: herold_bot
     image: h0d0user/herold_bot:latest
     restart: always
-    network_mode: host
+    networks:
+      - herold_net
     environment:
       - API_ID=YOURAPIID
       - API_HASH=YOURAPIHASH
       - SESSION=YOURSESSION
     volumes:
-      - "/root/herold_bot/circulation_ids_list.txt:/opt/circulation_ids_list.txt"
-      - "/root/herold_bot/circulation_response.txt:/opt/circulation_response.txt"
-      - "/root/herold_bot/new_year_patterns.txt:/opt/new_year_patterns.txt"
+      - "/root/herold_bot/herold_database.db:/opt/herold_database.db"
+      - "/root/herold_bot/congrats_list.txt:/opt/congrats_list.txt"
       - "/root/herold_bot/JustTalk.session:/opt/JustTalk.session"
+      - "/root/herold_bot/settings.toml:/opt/settings.toml"
+
 volumes:
-  herold_bot:
+  herold_volume:
+
+networks:
+  herold_net:

requirements.txt

@@ -1,3 +1,7 @@
 black
 isort
-telethon~=1.25.4
+boto3~=1.26.114
+dynaconf~=3.1.12
+telethon~=1.25.4
+SQLAlchemy~=2.0.7
+botocore~=1.29.114

src/bot.py

@@ -0,0 +1,86 @@
+import asyncio
+import logging
+import os
+import random
+import time
+from datetime import datetime
+from pathlib import Path
+from typing import List
+
+from telethon import TelegramClient
+
+import bucket
+import db
+
+# Environment variables
+API_ID = os.environ["API_ID"]
+API_HASH = os.environ["API_HASH"]
+SESSION = os.environ["SESSION"]
+
+client = TelegramClient(SESSION, API_ID, API_HASH)
+
+# Logging
+logging.basicConfig(
+    format="%(asctime)s - %(levelname)s - %(message)s", level=logging.INFO
+)
+logging.basicConfig(
+    format="%(asctime)s - %(levelname)s - %(message)s", level=logging.ERROR
+)
+
+
+def load_congrats_from_file(file: Path) -> List[str]:
+    """
+    Load congrats from the file
+    Try to load from file, if exception caught, send message about err
+    :return:
+    """
+    try:
+        with open(Path(file), "r", encoding="utf-8") as file:
+            result = file.readlines()
+            logging.info(f"Uploaded response from the {file.name} done successfully.")
+            return result
+    except FileNotFoundError as file_not_found_err:
+        logging.error(f"Err while load file - {file_not_found_err}")
+        return None
+
+
+CONGRATULATIONS = load_congrats_from_file(Path("congrats_list.txt"))
+
+
+async def send_congratulations():
+    """
+    Check if someone has a birthday on this date then send congratulation
+    Wait for one day to check date
+    :return:
+    """
+    current_date = datetime.today().strftime("%m.%d")
+    if db.get_tg_id(current_date) is None:
+        logging.info(f"Today - {current_date} - are no one to congratulate")
+        time.sleep(86400)
+    else:
+        user_data = await client.get_entity(db.get_tg_id(current_date))
+        logging.info(
+            f"Today - {current_date} - going to congratulate {user_data.first_name} - {user_data.username}"
+        )
+        await client.send_message(
+            db.get_tg_id(current_date), random.choice(CONGRATULATIONS)
+        )
+        async with client.action(user_data.id, "typing"):
+            await asyncio.sleep(random.randrange(2, 5))
+            await client.send_file(
+                db.get_tg_id(current_date),
+                bucket.download_file_from_bucket("birthdays", Path("/home/")),
+            )
+    time.sleep(400)
+    bucket.prune_directory(Path("/home/"))
+    time.sleep(86000)
+
+
+async def main():
+    while True:
+        await send_congratulations()
+
+
+if __name__ == "__main__":
+    with client:
+        client.loop.run_until_complete(main())