Update Yre.py

shizuchat/modules/Yre.py

@@ -1,43 +1,52 @@
-import asyncio
-from os import remove
-from pyrogram import Client, filters
-from pyrogram.enums import ChatType
-from pyrogram.types import Message
-from datetime import datetime
-from time import time
-from pyrogram.errors import MessageDeleteForbidden, RPCError
-from asyncio import sleep
-from pyrogram import Client, enums
-from pyrogram.types import Message, User
-from pyrogram import Client, enums, filters
-from pyrogram import Client, enums, filters, raw
-from pyrogram.errors.exceptions.bad_request_400 import ChatNotModified
-from shizuchat import shizuchat
-from pyrogram.types import ChatPermissions, Message
-from os import getenv
-from dotenv import load_dotenv
 import random
+import asyncio
+from shizuchat import shizuchat as app
+from pyrogram import filters  
+
+
+# Reactions list 
+reactions = [
+    '👍', '👎', '❤️', '🔥', '🥰', '👏', '😁', '🤔', '🤯', '😱', '🤬', '😢', '🎉', '🤩', '🤮', '💩', '🙏', '👌',
+    '🕊', '🤡', '🥱', '🥴', '😍', '🐳', '❤️‍🔥', '🌚', '🌭', '💯', '🤣', '⚡️', '🍌', '🏆', '💔', '🤨', '😐',
+    '🍓', '🍾', '💋', '🖕', '😈', '😴', '🤓', '👻', '👨‍💻', '👀', '🎃', '🙈', '😇', '😨', '🤝', '✍️', '🤗',
+    '🫡', '🎅', '🎄', '☃️', '💅', '🤪', '🗿', '🆒', '💘', '🙉', '🦄', '😘', '💊', '🙊', '😎', '👾', '🤷‍♂️',
+    '🤷‍♀️', '😭', '🤫', '💃', '🕺', '👋', '🐷', '🌹', '💖', '🌈', '🖤', '😡', '😳', '🥳', '🤖', '🦸', '🦹',
+    '🧙‍♂️', '🧙‍♀️', '🧝‍♂️', '🧝‍♀️', '🧛‍♂️', '🧛‍♀️', '🧟‍♂️', '🧟‍♀️', '🧞‍♂️', '🧞‍♀️', '🧜‍♂️', '🧜‍♀️',
+    '🧚‍♂️', '🧚‍♀️', '🐶', '🐱', '🐭', '🐹', '🐰', '🦊', '🐻', '🐼', '🐻‍❄️', '🐨', '🐯', '🦁', '🐮', '🐷',
+    '🐸', '🐵', '🙈', '🙉', '🙊', '🐒', '🐔', '🐧', '🐦', '🐤', '🐣', '🐥', '🦆', '🦅', '🦉', '🦜', '🐓', '🦃',
+    '🐬', '🐟', '🐠', '🐡', '🦈', '🐙', '🐚', '🐌', '🐞', '🐜', '🦋', '🐝', '🐧', '🦗', '🕷', '🕸', '🦕', '🦖',
+    '🦎', '🐢', '🐍', '🦂', '🦟', '🦠', '🐲', '🐉', '🦜', '🐳', '🐋', '🐬'
+]
 
-CMDS = [" ","sukh"]
 
-@shizuchat.on_message(filters.command(CMDS, prefixes=[""]))
-async def handle_incoming_messages(client, message):
-    reactions = ['👍','❤️','🔥','🥰','👏','😁','🤩','👌','🥱','😍','❤️‍🔥','💯','🤣','⚡️','😴','👀','🙈','🤝','🤗','🤪','💘','😘','😎']
-    if not await react_to_message(client, message, random.choice(reactions)):
-        print("All positive reactions failed.")
-        return
+# Global variable to track reaction status (default off)
+is_reaction_on = False  # Make sure this is defined globally
 
-async def react_to_message(client, message, reactions):
-    for reaction in reactions:
-        try:
-            if hasattr(message, 'id'):
-                await client.send_reaction(message.chat.id, message.id, reactions)
-                return True
-            else:
-                print("Message object does not have id attribute.")
-                return False
-        except Exception as e:
-            print(f"An error occurred: {e}")
-            continue
-    return False
+@app.on_message(filters.command(["reaction", "react", "eaction", "eact"], prefixes=["/", "!", ".", "R", "r"]))
+async def toggle_reaction(client, message):
+    global is_reaction_on
+    command_parts = message.text.split()
+    if len(command_parts) == 2:
+        if command_parts[1].lower() == "on":
+            is_reaction_on = True
+            await message.reply_text("Reaction spam is now ON! 😈")
+        elif command_parts[1].lower() == "off":
+            is_reaction_on = False
+            await message.reply_text("Reaction spam is now OFF! 😌")
+        else:
+            await message.reply_text("Invalid command. Use /reaction on or /reaction off")
+    else:
+        await message.reply_text("Invalid command. Use /reaction on or /reaction off")
 
+@app.on_message()
+async def auto_react(client, message):
+    global is_reaction_on
+    if is_reaction_on:
+        # Check the type of the message and react accordingly
+        if message.text or message.sticker or message.audio or message.voice or message.video or message.document or message.animation:
+            reaction = random.choice(reactions)
+            try:
+                await message.react(reaction)
+                await asyncio.sleep(0.01)  # Small delay to avoid rate limiting
+            except Exception as e:
+                print(f"An error occurred: {e}")