6s

system_module/db.py

@@ -14,6 +14,10 @@ class DB:
     def add(self, data_add):
         self.db.insert(data_add)
     
+    def add_raw(self, data_add):
+        # self.db.insert(data_add)
+        pass
+    
     def get_all(self):
         return self.db.all()
     

system_module/gelbooru.py

@@ -1,16 +1,36 @@
 import requests
+from datetime import datetime
 
 class GB:
-    def __init__(self, api_id, api_hash):
+    def __init__(self, api_id, api_hash, headers):
         self.api_id = api_id
         self.api_hash = api_hash
         self.root_url = "https://gelbooru.com/index.php"
+        self.HEADERS = headers
         
     def get_from_id(self, id_post):
+        params = {"page": "dapi", "s": "post", "q": "index", "json": 1, "id": id_post, "api_key":self.api_id, "user_id":self.api_hash}
+        response = requests.get(self.root_url, params=params, headers=self.HEADERS, timeout=15)
+        url_file = response.json()["post"][0]["file_url"]
+        file_name = response.json()["post"][0]["image"]
+        if os.path.exists("dw/"+file_name):
+            message = f"Файл '{file_name}' уже существует. Пропускаем."
+            print(message)
+            return response.json()["post"][0], url_file, file_name
         pass
     
     def get_from_md5(self, md5):
         pass
     
     def get_binary_file(self, id_post):
-        pass
+        pass
+
+    def gelbooru_date_parse(self, date_str):
+        if not self.date_str:
+            return 0, "Unknown"
+        try:
+            dt = datetime.strptime(self.date_str, "%a %b %d %H:%M:%S %z %Y")
+            return dt.timestamp(), dt.strftime("%Y-%m-%d")
+        except Exception as e:
+            print(f"Ошибка парсинга даты '{self.date_str}': {e}")
+            return 0, "Unknown"

system_module/logger.py

@@ -1,13 +1,12 @@
 import logging
 
 class Logger:
-    def __init__(self, name):
-        self.name = name
+    def __init__(self):
         self.logger_file = logging.getLogger(__name__)
         self.logger_cli = logging.getLogger(__name__)
         formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
         
-        file_handler = logging.FileHandler("logs/"+self.name+".log", encoding="utf-8")
+        file_handler = logging.FileHandler("logs.log", encoding="utf-8")
         file_handler.setFormatter(formatter)
         
         console_handler = logging.StreamHandler()