Add debug print for packname and bot_username in kang.py to diagnose sticker pack errors

DragMusic/plugins/management/kang.py

@@ -128,7 +128,7 @@ async def kang(client, message: Message):
         e = format_exc()
         return print(e)
 #-------
-    bot_username = BOT_USERNAME.lstrip("@")
+    bot_username = BOT_USERNAME.lstrip("@").strip("_")
     packnum = 0
     packname = f"f{message.from_user.id}_by_{bot_username}"
     limit = 0
@@ -136,6 +136,9 @@ async def kang(client, message: Message):
         while True:
             if limit >= 50:
                 return await msg.delete()
+            # Debug print for packname and bot_username
+            print(f"[DEBUG] packname: {packname}")
+            print(f"[DEBUG] bot_username: {bot_username}")
             stickerset = await get_sticker_set_by_name(client, packname)
             if not stickerset:
                 stickerset = await create_sticker_set(

DragMusic/plugins/management/quotly.py

@@ -0,0 +1,272 @@
+from io import BytesIO
+from pyrogram import filters
+from pyrogram.types import Message
+from DragMusic import app
+import httpx
+
+# -----------------------------------------------------------------
+fetch = httpx.AsyncClient(
+    http2=True,
+    verify=False,
+    headers={
+        "Accept-Language": "id-ID",
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36 Edge/107.0.1418.42",
+    },
+    timeout=httpx.Timeout(20),
+)
+# ------------------------------------------------------------------------
+class QuotlyException(Exception):
+    pass
+# --------------------------------------------------------------------------
+async def get_message_sender_id(ctx: Message):
+    if ctx.forward_date:
+        if ctx.forward_sender_name:
+            return 1
+        elif ctx.forward_from:
+            return ctx.forward_from.id
+        elif ctx.forward_from_chat:
+            return ctx.forward_from_chat.id
+        else:
+            return 1
+    elif ctx.from_user:
+        return ctx.from_user.id
+    elif ctx.sender_chat:
+        return ctx.sender_chat.id
+    else:
+        return 1
+# -----------------------------------------------------------------------------------------
+async def get_message_sender_name(ctx: Message):
+    if ctx.forward_date:
+        if ctx.forward_sender_name:
+            return ctx.forward_sender_name
+        elif ctx.forward_from:
+            return (
+                f"{ctx.forward_from.first_name} {ctx.forward_from.last_name}"
+                if ctx.forward_from.last_name
+                else ctx.forward_from.first_name
+            )
+        elif ctx.forward_from_chat:
+            return ctx.forward_from_chat.title
+        else:
+            return ""
+    elif ctx.from_user:
+        if ctx.from_user.last_name:
+            return f"{ctx.from_user.first_name} {ctx.from_user.last_name}"
+        else:
+            return ctx.from_user.first_name
+    elif ctx.sender_chat:
+        return ctx.sender_chat.title
+    else:
+        return ""
+# ---------------------------------------------------------------------------------------------------
+async def get_message_sender_username(ctx: Message):
+    if ctx.forward_date:
+        if (
+            not ctx.forward_sender_name
+            and not ctx.forward_from
+            and ctx.forward_from_chat
+            and ctx.forward_from_chat.username
+        ):
+            return ctx.forward_from_chat.username
+        elif (
+            not ctx.forward_sender_name
+            and not ctx.forward_from
+            and ctx.forward_from_chat
+            or ctx.forward_sender_name
+            or not ctx.forward_from
+        ):
+            return ""
+        else:
+            return ctx.forward_from.username or ""
+    elif ctx.from_user and ctx.from_user.username:
+        return ctx.from_user.username
+    elif (
+        ctx.from_user
+        or ctx.sender_chat
+        and not ctx.sender_chat.username
+        or not ctx.sender_chat
+    ):
+        return ""
+    else:
+        return ctx.sender_chat.username
+# ------------------------------------------------------------------------
+async def get_message_sender_photo(ctx: Message):
+    if ctx.forward_date:
+        if (
+            not ctx.forward_sender_name
+            and not ctx.forward_from
+            and ctx.forward_from_chat
+            and ctx.forward_from_chat.photo
+        ):
+            return {
+                "small_file_id": ctx.forward_from_chat.photo.small_file_id,
+                "small_photo_unique_id": ctx.forward_from_chat.photo.small_photo_unique_id,
+                "big_file_id": ctx.forward_from_chat.photo.big_file_id,
+                "big_photo_unique_id": ctx.forward_from_chat.photo.big_photo_unique_id,
+            }
+        elif (
+            not ctx.forward_sender_name
+            and not ctx.forward_from
+            and ctx.forward_from_chat
+            or ctx.forward_sender_name
+            or not ctx.forward_from
+        ):
+            return ""
+        else:
+            return (
+                {
+                    "small_file_id": ctx.forward_from.photo.small_file_id,
+                    "small_photo_unique_id": ctx.forward_from.photo.small_photo_unique_id,
+                    "big_file_id": ctx.forward_from.photo.big_file_id,
+                    "big_photo_unique_id": ctx.forward_from.photo.big_photo_unique_id,
+                }
+                if ctx.forward_from.photo
+                else ""
+            )
+    elif ctx.from_user and ctx.from_user.photo:
+        return {
+            "small_file_id": ctx.from_user.photo.small_file_id,
+            "small_photo_unique_id": ctx.from_user.photo.small_photo_unique_id,
+            "big_file_id": ctx.from_user.photo.big_file_id,
+            "big_photo_unique_id": ctx.from_user.photo.big_photo_unique_id,
+        }
+    elif (
+        ctx.from_user
+        or ctx.sender_chat
+        and not ctx.sender_chat.photo
+        or not ctx.sender_chat
+    ):
+        return ""
+    else:
+        return {
+            "small_file_id": ctx.sender_chat.photo.small_file_id,
+            "small_photo_unique_id": ctx.sender_chat.photo.small_photo_unique_id,
+            "big_file_id": ctx.sender_chat.photo.big_file_id,
+            "big_photo_unique_id": ctx.sender_chat.photo.big_photo_unique_id,
+        }
+# ---------------------------------------------------------------------------------------------------
+async def get_text_or_caption(ctx: Message):
+    if ctx.text:
+        return ctx.text
+    elif ctx.caption:
+        return ctx.caption
+    else:
+        return ""
+# ---------------------------------------------------------------------------------------------------
+async def pyrogram_to_quotly(messages, is_reply):
+    if not isinstance(messages, list):
+        messages = [messages]
+    payload = {
+        "type": "quote",
+        "format": "png",
+        "backgroundColor": "#1b1429",
+        "messages": [],
+    }
+    for message in messages:
+        the_message_dict_to_append = {}
+        if message.entities:
+            the_message_dict_to_append["entities"] = [
+                {
+                    "type": entity.type.name.lower(),
+                    "offset": entity.offset,
+                    "length": entity.length,
+                }
+                for entity in message.entities
+            ]
+        elif message.caption_entities:
+            the_message_dict_to_append["entities"] = [
+                {
+                    "type": entity.type.name.lower(),
+                    "offset": entity.offset,
+                    "length": entity.length,
+                }
+                for entity in message.caption_entities
+            ]
+        else:
+            the_message_dict_to_append["entities"] = []
+        the_message_dict_to_append["chatId"] = await get_message_sender_id(message)
+        the_message_dict_to_append["text"] = await get_text_or_caption(message)
+        the_message_dict_to_append["avatar"] = True
+        the_message_dict_to_append["from"] = {}
+        the_message_dict_to_append["from"]["id"] = await get_message_sender_id(message)
+        the_message_dict_to_append["from"]["name"] = await get_message_sender_name(message)
+        the_message_dict_to_append["from"]["username"] = await get_message_sender_username(message)
+        the_message_dict_to_append["from"]["type"] = message.chat.type.name.lower()
+        the_message_dict_to_append["from"]["photo"] = await get_message_sender_photo(message)
+        if message.reply_to_message and is_reply:
+            the_message_dict_to_append["replyMessage"] = {
+                "name": await get_message_sender_name(message.reply_to_message),
+                "text": await get_text_or_caption(message.reply_to_message),
+                "chatId": await get_message_sender_id(message.reply_to_message),
+            }
+        else:
+            the_message_dict_to_append["replyMessage"] = {}
+        payload["messages"].append(the_message_dict_to_append)
+    r = await fetch.post("https://bot.lyo.su/quote/generate.png", json=payload)
+    if not r.is_error:
+        return r.read()
+    else:
+        raise QuotlyException(await r.aread())
+# ------------------------------------------------------------------------------------------
+def isArgInt(txt) -> list:
+    count = txt
+    try:
+        count = int(count)
+        return [True, count]
+    except ValueError:
+        return [False, 0]
+# ---------------------------------------------------------------------------------------------------
+@app.on_message(filters.command(["q"]) & filters.reply)
+async def msg_quotly_cmd(client, ctx: Message):
+    args = ctx.text.split()
+    is_reply = False
+    quote_range = 1
+    # /q r or /q <number>
+    if len(args) > 1:
+        if args[1].lower() == "r":
+            is_reply = True
+        else:
+            try:
+                quote_range = int(args[1])
+                if quote_range < 2 or quote_range > 10:
+                    return await ctx.reply("Invalid range (2-10)")
+            except ValueError:
+                return await ctx.reply("Invalid argument. Use /q r or /q <number> (2-10)")
+    if quote_range > 1:
+        try:
+            messages = [
+                i
+                for i in await client.get_messages(
+                    chat_id=ctx.chat.id,
+                    message_ids=range(
+                        ctx.reply_to_message.id,
+                        ctx.reply_to_message.id + quote_range
+                    ),
+                    replies=-1,
+                )
+                if not i.empty and not i.media
+            ]
+        except Exception:
+            return await ctx.reply_text("🤷🏻‍♂️")
+        try:
+            make_quotly = await pyrogram_to_quotly(messages, is_reply=is_reply)
+            bio_sticker = BytesIO(await make_quotly)
+            bio_sticker.name = "misskatyquote_sticker.webp"
+            return await ctx.reply_sticker(bio_sticker)
+        except Exception:
+            return await ctx.reply("🤷🏻‍♂️")
+    else:
+        try:
+            messages_one = await client.get_messages(
+                chat_id=ctx.chat.id, message_ids=ctx.reply_to_message.id, replies=-1
+            )
+            messages = [messages_one]
+        except Exception:
+            return await ctx.reply("🤷🏻‍♂️")
+        try:
+            make_quotly = await pyrogram_to_quotly(messages, is_reply=is_reply)
+            bio_sticker = BytesIO(await make_quotly)
+            bio_sticker.name = "misskatyquote_sticker.webp"
+            return await ctx.reply_sticker(bio_sticker)
+        except Exception as e:
+            return await ctx.reply(f"ERROR: {e}")