MAX: пуши через firebase (особо не тестил, вроде работает)

src/common/push.py

@@ -0,0 +1,92 @@
+import asyncio
+import logging
+import time
+import firebase_admin
+from firebase_admin import credentials, messaging
+
+class PushService:
+    def __init__(self, credentials_path):
+        self.logger = logging.getLogger(__name__)
+
+        if not credentials_path:
+            self.logger.warning("Огненная база сегодня не работает, укажите путь к файлу с ключами")
+            self.enabled = False
+            return
+
+        cred = credentials.Certificate(credentials_path)
+        firebase_admin.initialize_app(cred)
+        self.enabled = True
+        self.logger.info("Огненная база инициализирована")
+
+    async def send(self, push_token, data):
+        """Отправка пуша"""
+        if not self.enabled:
+            return None
+
+        str_data = {k: str(v) for k, v in data.items() if v is not None}
+
+        message = messaging.Message(
+            data=str_data,
+            token=push_token,
+            android=messaging.AndroidConfig(
+                priority="high",
+            ),
+        )
+
+        try:
+            loop = asyncio.get_event_loop()
+            response = await loop.run_in_executor(None, messaging.send, message)
+            self.logger.debug(f"Отправил пуш: {response}")
+            return response
+        except messaging.UnregisteredError:
+            self.logger.warning(f"Пуш-токен не зарегистрирован: {push_token}")
+            return None
+        except Exception as e:
+            self.logger.error(f"Не удалось отправить пуш: {e}")
+            return None
+
+    async def send_to_user(self, db_pool, phone, sender_id=None, msg_id=None,
+                           chat_id=None, text="", is_group=False):
+        """Отправка пушей на все устройства пользователя"""
+        if not self.enabled:
+            return
+
+        # Получаем имя отправителя
+        user_name = ""
+        if sender_id:
+            async with db_pool.acquire() as conn:
+                async with conn.cursor() as cursor:
+                    await cursor.execute(
+                        "SELECT firstname, lastname FROM users WHERE id = %s",
+                        (sender_id,)
+                    )
+                    sender = await cursor.fetchone()
+                    if sender:
+                        firstname = sender.get("firstname", "")
+                        lastname = sender.get("lastname", "")
+                        user_name = f"{firstname} {lastname}".strip()
+
+        now_ms = str(int(time.time() * 1000))
+        msg_type = "ChatMessage" if is_group else "Message"
+        data = {
+            "type": msg_type,
+            "msgid": str(msg_id) if msg_id else "0",
+            "suid": str(sender_id) if sender_id else None,
+            "mc": str(chat_id) if chat_id else None,
+            "msg": text,
+            "userName": user_name,
+            "ttime": now_ms,
+            "ctime": now_ms,
+        }
+
+        # Получаем все пуш-токены пользователя
+        async with db_pool.acquire() as conn:
+            async with conn.cursor() as cursor:
+                await cursor.execute(
+                    "SELECT push_token FROM tokens WHERE phone = %s AND push_token IS NOT NULL",
+                    (phone,)
+                )
+                rows = await cursor.fetchall()
+
+        for row in rows:
+            await self.send(row["push_token"], data)