Merge branch 'beta' into main

Powers/plugins/downloads.py

@@ -0,0 +1,38 @@
+# Add instagrapi & pytube in requirements.txt
+from pyrogram import Client, filters
+import urllib.request
+from instagrapi import Client as InstaClient
+from pytube import YouTube
+from Powers.bot_class import Gojo as app
+
+insta_client = InstaClient()
+
+@app.on_message(filters.command("download"))
+def download_media(client, message):
+    chat_id = message.chat.id
+    platform = message.text.split()[1]
+    client.send_message(chat_id, "Enter your choice 1 for Instagram 2 For YouTube:")
+    if platform == "1":
+        client.send_message(chat_id, "Please enter the Instagram post or story URL:")
+        insta_link = message.text.split()[2]
+
+        media = insta_client.media_info(insta_link)
+
+        video_url = media.video_versions[0].url
+        urllib.request.urlretrieve(video_url, "video.mp4")
+
+        client.send_video(chat_id, video="video.mp4")
+
+    elif platform == "2":
+        client.send_message(chat_id, "Please enter the YouTube link:")
+        yt_link = message.text.split()[2]
+
+        yt = YouTube(yt_link)
+        stream = yt.streams.get_highest_resolution()
+
+        stream.download(output_path="./", filename="video.mp4")
+
+        client.send_video(chat_id, video="video.mp4")
+
+    else:
+        client.send_message(chat_id, "Invalid platform choice. Please enter 1 for Instagram or 2 for YouTube.")

Powers/plugins/flood.py

@@ -350,15 +350,17 @@ async def flood_watcher(c: Gojo, m: Message):
     if not len(dic):
         z = {c_id : {u_id : [[],[]]}}
         dic.update(z)
-    for i in dic.keys():
-        if c_id != i:
-            z = {c_id : {u_id : [[],[]]}}
-            dic.update(z)
+    try:
+      dic[c_id] # access and check weather the c_id present or not
+    except KeyError:
+      z = {c_id : {u_id : [[],[]]}}
+      dic.update(z)
 
-    for i in dic[c_id].keys():
-        if u_id != i:
-            z = {u_id : [[],[]]}
-            dic[c_id].update(z) # make the dic something like {c_id : {u_id : [[for time],[for msg]]}}
+    try:
+      dic[c_id][u_id]
+    except KeyError:
+      z = {u_id : [[],[]]}
+      dic[c_id].update(z) # make the dic something like {c_id : {u_id : [[for time],[for msg]]}}
     sec = round(time.time())
     try:
         dic[c_id][u_id][0].append(sec)

Powers/plugins/search.py

@@ -156,7 +156,7 @@ async def anime_search(c: Gojo, m: Message):
         LOGGER.error(format_exc())
         return
 
-@Gojo.on_message(command('anime'))
+@Gojo.on_message(command('stack'))
 async def stack_search(c: Gojo, m: Message):
     split = m.text.split(None, 1)
     if len(split) == 1: