TheHamkerCat 5 jaren geleden
bovenliggende
commit
18840c0396
3 gewijzigde bestanden met toevoegingen van 20 en 13 verwijderingen
  1. 16 11
      wbb/modules/quotly.py
  2. 1 1
      wbb/modules/reverse.py
  3. 3 1
      wbb/modules/stickers.py

+ 16 - 11
wbb/modules/quotly.py

@@ -89,17 +89,22 @@ async def quotly_func(client, message: Message):
             if arg[1] < 2 or arg[1] > 10:
                 return await m.edit("Argument must be between 2-10.")
             count = arg[1]
-            messages = [i for i in await client.get_messages(
-                message.chat.id,
-                [
-                    i
-                    for i in range(
-                        message.reply_to_message.message_id,
-                        message.reply_to_message.message_id + count,
-                    )
-                ],
-                replies=0,
-            ) if not i.empty]
+            messages = [
+                i
+                for i in await client.get_messages(
+                    message.chat.id,
+                    [
+                        i
+                        for i in range(
+                            message.reply_to_message.message_id,
+                            message.reply_to_message.message_id
+                            + count,
+                        )
+                    ],
+                    replies=0,
+                )
+                if not i.empty
+            ]
         else:
             if getArg(message) != "r":
                 return await m.edit(

+ 1 - 1
wbb/modules/reverse.py

@@ -149,6 +149,6 @@ async def reverse_image_search(_, message: Message):
         pass
 
     await m.edit(
-        f"**Result**: [{text}]({location})",
+        text,
         disable_web_page_preview=True,
     )

+ 3 - 1
wbb/modules/stickers.py

@@ -134,6 +134,9 @@ async def kang(client, message: Message):
                 sticker_emoji,
             )
         elif doc:
+            if doc.file_size > 10000000:
+                return await msg.edit("File too large")
+
             temp_file_path = await app.download_media(doc)
             image_type = imghdr.what(temp_file_path)
             if image_type not in SUPPORTED_TYPES:
@@ -149,7 +152,6 @@ async def kang(client, message: Message):
                 raise Exception(
                     f"Something went wrong while resizing the sticker (at {temp_file_path}); {e}"
                 )
-                return False
             sticker = await create_sticker(
                 await upload_document(
                     client, temp_file_path, message.chat.id