Change /rice command to automatically forward [RICE] messages

wbb/modules/rice.py

@@ -1,27 +1,26 @@
 from pyrogram import filters
 from pyrogram.types import Message
 from wbb import app
-from wbb.utils import cust_filter
 
 
-__MODULE__ = "Rice"
-__HELP__ = "/rice - To Forward Your Linux Rice To DE_WM's" \
-           "Rice Gallery only works in [DE_WM](t.me/de_wm)"
-
-
-@app.on_message(filters.chat('DE_WM') & cust_filter.command(commands=("rice")))
+@app.on_message(filters.chat("DE_WM")
+                & filters.media
+                & filters.regex(r"^\[RICE\] ")
+                & ~filters.forwarded
+                & ~filters.edited)
 async def rice(_, message: Message):
-    if bool(message.reply_to_message) is True:
-        app.set_parse_mode("markdown")
-        user_id = message.reply_to_message.from_user.id
-        await message.reply_to_message.forward('RiceGallery')
-        await message.reply_text(
-            f"[Your](tg://user?id={user_id}) Rice Forwared"
-            " To [Rice Gallery](https://t.me/RiceGallery)",
-            disable_web_page_preview=True)
-
-    elif bool(message.reply_to_message) is False:
-        await message.reply_text(
-            "Reply To A Message With /rice, Just Hitting /rice "
-            + "Won't Do Anything Other Than Proving Everyone That "
-            + "You Are A Spammer Who Is Obsessed To 'BlueTextMustClickofobia")
+    """Forward media and media_group messages which starts with [RICE] with
+    space and description in DE_WM group to RiceGallery channel
+    edited or forwarded messages won't be forwarded
+    """
+    reply_text = ("Successfully forwarded to "
+                  "[Rice Gallery](https://t.me/RiceGallery)")
+    if message.media_group_id:
+        message_id = message.message_id
+        media_group = await app.get_media_group("DE_WM", message_id)
+        await app.forward_messages("RiceGallery", "DE_WM",
+                                   [m.message_id for m in media_group])
+        await message.reply_text(reply_text, disable_web_page_preview=True)
+    else:
+        await message.forward("RiceGallery")
+        await message.reply_text(reply_text, disable_web_page_preview=True)