Merge branch 'main' of https://github.com/izzzzzi/Solo_bot

database.py

@@ -9,6 +9,22 @@ from logger import logger
 
 async def init_db():
     conn = await asyncpg.connect(DATABASE_URL)
+
+    # Таблица для хранения информации о платежах
+    await conn.execute(
+        """
+        CREATE TABLE IF NOT EXISTS payments (
+            id SERIAL PRIMARY KEY,
+            tg_id BIGINT NOT NULL,
+            amount REAL NOT NULL,
+            payment_system TEXT NOT NULL,
+            status TEXT DEFAULT 'success',
+            created_at TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY (tg_id) REFERENCES users(tg_id)
+        )
+        """
+    )
+
     # Таблица для хранения основной информации о пользователях из Telegram
     await conn.execute(
         """
@@ -51,6 +67,21 @@ async def init_db():
         """
     )
 
+    # Таблица для хранения информации о платежах
+    await conn.execute(
+        """
+        CREATE TABLE IF NOT EXISTS payments (
+            id SERIAL PRIMARY KEY,
+            tg_id BIGINT NOT NULL,
+            amount REAL NOT NULL,
+            payment_system TEXT NOT NULL,
+            status TEXT DEFAULT 'success',
+            created_at TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY (tg_id) REFERENCES users(tg_id)
+        )
+        """
+    )
+
     # Таблица для хранения ключей
     await conn.execute(
         """

handlers/utils.py

@@ -42,33 +42,24 @@ async def get_least_loaded_cluster() -> str:
     Returns:
         str: Идентификатор наименее загруженного кластера.
     """
-    cluster_loads: dict[str, int] = {}
+    cluster_loads: dict[str, int] = {cluster_id: 0 for cluster_id in CLUSTERS.keys()}
 
     async with asyncpg.create_pool(DATABASE_URL) as pool:
         async with pool.acquire() as conn:
-            keys = await conn.fetch("SELECT * FROM keys")
+            keys = await conn.fetch("SELECT server_id FROM keys")
 
             for key in keys:
                 cluster_id = key["server_id"]
-                if re.match(r"^cluster\d+$", cluster_id):
-                    cluster_loads[cluster_id] = cluster_loads.get(cluster_id, 0) + 1
+                if cluster_id in cluster_loads:
+                    cluster_loads[cluster_id] += 1
 
-    logger.info(f"Cluster loads: {cluster_loads}")
+    logger.info(f"Cluster loads after database query: {cluster_loads}")
 
     if not cluster_loads:
-        available_clusters = [cluster_id for cluster_id in CLUSTERS.keys() if re.match(r"^cluster\d+$", cluster_id)]
-
-        logger.info(f"Available clusters from config: {available_clusters}")
-
-        if available_clusters:
-            selected_cluster = available_clusters[0]
-            logger.info(f"Returning the first available cluster: {selected_cluster}")
-            return selected_cluster
-
-        logger.warning("No valid clusters found in config, returning 'cluster1'.")
+        logger.warning("No clusters found in database or configuration.")
         return "cluster1"
 
-    least_loaded_cluster = min(cluster_loads, key=lambda k: (cluster_loads.get(k, 0), k))
+    least_loaded_cluster = min(cluster_loads, key=lambda k: (cluster_loads[k], k))
 
     logger.info(f"Least loaded cluster selected: {least_loaded_cluster}")