ruffed

bot/announce.py

@@ -1,12 +1,13 @@
 from bot.api import telegram_api
-from utils.mention import mention, userdata_extract
-from utils.store import redis
+from utils.mention import userdata_extract
+from state.redis import redis
 import logging
 
 
 logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.INFO)
 
+
 def get_newcomer_message(msg):
     lang = msg["from"].get("language_code", "ru")
     r = "хочет присоединиться к нам здесь" if lang == "ru" else " wants to join us here"
@@ -27,27 +28,36 @@ async def show_announce(msg):
     userphotos_response = await telegram_api("getUserphotos", user_id=from_id)
 
     file_id = ""
-    if isinstance(userphotos_response, dict) and userphotos_response["ok"] and userphotos_response["result"]["total_count"] > 0:
+    if (
+        isinstance(userphotos_response, dict)
+        and userphotos_response["ok"]
+        and userphotos_response["result"]["total_count"] > 0
+    ):
         logger.info("showing button with photo")
         file_id = userphotos_response["result"]["photos"][0][0]["file_id"]
 
-    r = await telegram_api("sendPhoto",
+    r = await telegram_api(
+        "sendPhoto",
         chat_id=chat_id,
         file_id=file_id,
         caption=newcomer_message,
-        reply_to=mid
+        reply_to=mid,
     )
     announce_msg_id = r.get("message_id")
-    await redis.set(f"announce:{chat_id}:{from_id}", announce_message_id)
+    await redis.set(f"announce:{chat_id}:{from_id}", announce_msg_id)
 
 
 async def edit_announce(msg):
     logger.info("editing announce")
     chat_id = str(msg["chat"]["id"])
     from_id = str(msg["from"]["id"])
-    mid = msg.get("message_id", "")
     caption = get_newcomer_message(msg) + msg.get("text").replace("/message ", "")
     announce_message_id = await redis.get(f"announce:{chat_id}:{from_id}")
     if announce_message_id:
-        r = await telegram_api("editMessageCaption", chat_id=chat_id, message_id=int(announce_message_id), caption=caption)
+        r = await telegram_api(
+            "editMessageCaption",
+            chat_id=chat_id,
+            message_id=int(announce_message_id),
+            caption=caption,
+        )
         await redis.set(f"announce:{chat_id}:{from_id}", r.get("message_id"))