`V 2.2.0` few more changes left

Powers/__init__.py

@@ -1,3 +1,4 @@
+import shutil
 from datetime import datetime
 from importlib import import_module as imp_mod
 from logging import (INFO, WARNING, FileHandler, StreamHandler, basicConfig,
@@ -21,6 +22,9 @@ LOGDIR = f"{__name__}/logs"
 # Make Logs directory if it does not exixts
 if not path.isdir(LOGDIR):
     mkdir(LOGDIR)
+else:
+    shutil.rmtree(LOGDIR)
+    mkdir(LOGDIR)
 
 LOGFILE = f"{LOGDIR}/{__name__}_{LOG_DATETIME}_log.txt"
 
@@ -59,9 +63,9 @@ except Exception as ef:
 # time zone
 TIME_ZONE = pytz.timezone(Config.TIME_ZONE)
 
-path = "./Version"
+Vpath = "./Version"
 version = []
-for i in listdir(path):
+for i in listdir(Vpath):
     if i.startswith("version") and i.endswith("md"):
         version.append(i)
     else:
@@ -134,6 +138,9 @@ Defult_dev = set(defult_dev)
 DEVS = DEVS_USER | Defult_dev
 DEV_USERS = list(DEVS)
 
+CHROME_BIN = Config.CHROME_BIN
+CHROME_DRIVER = Config.CHROME_DRIVER
+
 # Plugins, DB and Workers
 DB_URI = Config.DB_URI
 DB_NAME = Config.DB_NAME
@@ -148,6 +155,14 @@ HELP_COMMANDS = {}  # For help menu
 UPTIME = time()  # Check bot uptime
 
 
+#Make dir
+youtube_dir = "./Youtube/"
+if not path.isdir(youtube_dir):
+    mkdir(youtube_dir)
+else:
+    shutil.rmtree(youtube_dir)
+    mkdir(youtube_dir)
+
 scheduler = AsyncIOScheduler(timezone=TIME_ZONE)
 
 

Powers/plugins/bans.py

@@ -487,7 +487,7 @@ async def skick_usr(c: Gojo, m: Message):
         return
 
     if user_id == Config.BOT_ID:
-        await m.reply_text("Huh, why would I kick myself?")
+        await m.reply_text("Nuh Hu, why would I kick myself?")
         await m.stop_propagation()
 
     SUPPORT_STAFF = get_support_staff()

Powers/plugins/formatting.py

@@ -39,14 +39,7 @@ async def markdownhelp(_, m: Message):
     return
 
 
-@Gojo.on_callback_query(filters.regex("^formatting."))
-async def get_formatting_info(c: Gojo, q: CallbackQuery):
-    cmd = q.data.split(".")[1]
-    kb = ikb([[("Back", "back.formatting")]])
-
-    if cmd == "md_formatting":
-        
-        txt = """<b>Markdown Formatting</b>
+md_txt = """<b>Markdown Formatting</b>
 You can format your message using <b>bold</b>, <i>italic</i>, <u>underline</u>, <strike>strike</strike> and much more. Go ahead and experiment!
 
 **Note**: It supports telegram user based formatting as well as html and markdown formattings.
@@ -66,19 +59,74 @@ If you would like to send buttons on the same row, use the <code>:same</code> fo
 <code>[button 2](buttonurl://example.com:same)</code>
 <code>[button 3](buttonurl://example.com)</code>
 This will show button 1 and 2 on the same line, while 3 will be underneath."""
+
+async def get_splited_formatting(msg, page=1):
+    msg = msg.split("\n")
+    l = len(msg)
+    new_msg = ""
+    total = l // 10
+    first = 10 * (page - 1)
+    last = 10 * page
+    if not first:
+        for i in msg[first:last]:
+            new_msg += f"{i}\n"
+        kb = [
+            [
+                ("Next page ▶️", f"next_format_{page+1}")
+            ]
+        ]
+    else:
+        first += 1
+        if page == total:
+            for i in msg[first:]:
+                new_msg += f"{i}\n"
+            kb = [
+                [
+                    ("◀️ Previous page", f"next_format_{page-1}")
+                ]
+            ]
+        else:
+            for i in msg[first:last]:
+                new_msg += f"{i}\n"
+            kb = [
+                    [
+                        ("◀️ Previous page", f"next_format_{page-1}"),
+                        ("Next page ▶️", f"next_format_{page+1}")
+                    ]
+                ]
+
+
+    kb = ikb(kb, True, "back.formatting")
+
+    return new_msg, kb
+
+@Gojo.on_callback_query(filters.regex(r"^next_format_.*[0-9]$"))
+async def change_formatting_page(c: Gojo, q: CallbackQuery):
+    page = q.data.split("_")[-1]
+    txt, kb = await get_splited_formatting(md_txt, int(page))
+    await q.edit_message_caption(txt, reply_markup=kb,parse_mode=enums.ParseMode.HTML,)
+    return
+
+@Gojo.on_callback_query(filters.regex("^formatting."))
+async def get_formatting_info(c: Gojo, q: CallbackQuery):
+    cmd = q.data.split(".")[1]
+    kb = ikb([[("Back", "back.formatting")]])
+
+    if cmd == "md_formatting":
+        
         try:
             await q.edit_message_caption(
-                caption=txt,
+                caption=md_txt,
                 reply_markup=kb,
                 parse_mode=enums.ParseMode.HTML,
             )
         except MediaCaptionTooLong:
-            kb = ikb([[("Back", "DELETEEEE")]])
-            await c.send_message(
-                chat_id=q.message.chat.id,
-                text=txt,
+            txt, kb = await get_splited_formatting(md_txt)
+            await q.edit_message_caption(
+                caption=txt,
+                reply_markup=kb,
                 parse_mode=enums.ParseMode.HTML,
-                reply_markup=kb)
+            )
     elif cmd == "fillings":
         await q.edit_message_caption(
             caption="""<b>Fillings</b>

Powers/plugins/locks.py

@@ -36,7 +36,9 @@ l_t = """
 - `forwardall` = Forwarding from channel and user
 - `forwardu` = Forwarding from user
 - `forwardc` = Forwarding from channel
-- `links | url` = Lock links"""
+- `links | url` = Lock links
+- `bot` = Adding bot will be forbidden
+"""
 
 
 @Gojo.on_message(command("locktypes"))

Powers/plugins/start.py

@@ -18,8 +18,9 @@ from Powers.utils.custom_filters import command
 from Powers.utils.extras import StartPic
 from Powers.utils.kbhelpers import ikb
 from Powers.utils.parser import mention_html
-from Powers.utils.start_utils import (gen_cmds_kb, gen_start_kb, get_help_msg,
-                                      get_private_note, get_private_rules)
+from Powers.utils.start_utils import (gen_cmds_kb, gen_start_kb,
+                                      get_private_note, get_private_rules,
+                                      iter_msg)
 from Powers.utils.string import encode_decode
 from Powers.vars import Config
 
@@ -85,7 +86,7 @@ async def start(c: Gojo, m: Message):
                 await get_private_rules(c, m, help_option)
                 return
 
-            help_msg, help_kb = await get_help_msg(c, m, help_option)
+            help_msg, help_kb = await iter_msg(c, m, help_option)
 
             if not help_msg:
                 return
@@ -219,7 +220,7 @@ async def help_menu(c: Gojo, m: Message):
     if len(m.text.split()) >= 2:
         textt = m.text.replace(" ", "_",).replace("_", " ", 1)
         help_option = (textt.split(None)[1]).lower()
-        help_msg, help_kb = await get_help_msg(c, m, help_option)
+        help_msg, help_kb = await iter_msg(c, m, help_option)
 
         if not help_msg:
             LOGGER.error(
@@ -291,6 +292,64 @@ Commands available:
     return
 
 
+async def get_divided_msg(plugin_name: str, page:int=1, back_to_do = None):
+    msg = HELP_COMMANDS[plugin_name]["help_msg"]
+    msg = msg.split("\n")
+    l = len(msg)
+    new_msg = ""
+    total = l // 10
+    first = 10 * (page - 1)
+    last = 10 * page
+
+    if not first:
+        for i in msg[first:last]:
+            new_msg += f"{i}\n"
+        kb = [
+            [
+                ("Next page ▶️", f"iter_page_{plugin_name}_{(back_to_do+'_') if back_to_do else ''}{page+1}")
+            ]
+        ]
+    else:
+        first += 1
+        if page == total:
+            for i in msg[first:]:
+                new_msg += f"{i}\n"
+            kb = [
+                [
+                    ("◀️ Previous page", f"iter_page_{plugin_name}_{(back_to_do+'_') if back_to_do else ''}{page-1}")
+                ]
+            ]
+        else:
+            for i in msg[first:last]:
+                new_msg += f"{i}\n"
+            kb = [
+                    [
+                        ("◀️ Previous page", f"iter_page_{plugin_name}_{(back_to_do+'_') if back_to_do else ''}{page-1}"),
+                        ("Next page ▶️", f"iter_page_{plugin_name}_{(back_to_do+'_') if back_to_do else ''}{page+1}")
+                    ]
+                ]
+    if back_to_do:  
+        kb = ikb(kb, True, back_to_do)
+    else:
+        kb = ikb(kb)
+
+    return new_msg, kb
+    
+@Gojo.on_callback_query(filters.regex(r"^iter_page_.*[0-9]$"))
+async def helppp_page_iter(c: Gojo, q: CallbackQuery):
+    data = q.data.split("_")
+    plugin_ = data[2]
+    try:
+        back_to = data[-2]
+    except:
+        back_to = None
+    curr_page = int(data[-1])
+    msg, kb = await get_divided_msg(plugin_, curr_page, back_to_do=back_to)
+
+    await q.edit_message_caption(msg, reply_markup=kb)
+    return
+
+
 @Gojo.on_callback_query(filters.regex("^bot_curr_info$"))
 async def give_curr_info(c: Gojo, q: CallbackQuery):
     start = time()
@@ -323,17 +382,20 @@ async def get_module_info(c: Gojo, q: CallbackQuery):
             reply_markup=ikb(help_kb, True, todo="commands"),
         )
     except MediaCaptionTooLong:
-        kb = ikb([[("Back", "DELETEEEE")]])
-        await c.send_message(chat_id=q.message.chat.id, text=help_msg, reply_markup=kb)
+        caption, kb = await get_divided_msg(f"plugins.{module}", back_to_do="commands")
+        await q.edit_message_caption(
+            caption,
+            parse_mode=enums.ParseMode.MARKDOWN,
+            reply_markup=kb
+        )
     await q.answer()
     return
 
-DEV_USERS = get_support_staff("dev")
-SUDO_USERS = get_support_staff("sudo")
-
 
 @Gojo.on_callback_query(filters.regex("^give_bot_staffs$"))
 async def give_bot_staffs(c: Gojo, q: CallbackQuery):
+    DEV_USERS = get_support_staff("dev")
+    SUDO_USERS = get_support_staff("sudo")  
     try:
         owner = await c.get_users(OWNER_ID)
         reply = f"<b>🌟 Owner:</b> {(await mention_html(owner.first_name, OWNER_ID))} (<code>{OWNER_ID}</code>)\n"

Powers/plugins/web_con.py

@@ -1,20 +1,21 @@
 import asyncio
 import os
+import shutil
 from traceback import format_exc
 
 from pyrogram import filters
 from pyrogram.types import CallbackQuery
 from pyrogram.types import InlineKeyboardButton as IKB
 from pyrogram.types import InlineKeyboardMarkup as IKM
-from pyrogram.types import Message
+from pyrogram.types import InputMediaPhoto, InputMediaVideo, Message
 
-from Powers import (LOGGER, RMBG, Audd, genius_lyrics, is_audd,
-                    is_genius_lyrics, is_rmbg)
+from Powers import LOGGER, RMBG, genius_lyrics, is_genius_lyrics, is_rmbg
 from Powers.bot_class import Gojo
 from Powers.utils.custom_filters import command
 from Powers.utils.http_helper import *
 from Powers.utils.sticker_help import toimage
 from Powers.utils.web_helpers import *
+from Powers.utils.web_scrapper import INSTAGRAM, SCRAP_DATA
 
 # @Gojo.on_message(command(["songname","insong","songinfo","whichsong","rsong","reversesong"]))
 # • /whichsong (/songname, /songinfo, /insong, /rsong, /reversesong) : Reply to file to get the song playing in it.
@@ -254,15 +255,15 @@ async def song_down_up(c: Gojo, m: Message):
     except IndexError:
         await m.reply_text("**USAGE**\n /song [song name | link]")
         return
-    if splited.startswith("https://youtube.com"):
-        is_direct = True
-        query = splited.split("?")[0]
+    _id = get_video_id(splited)
+    if not _id:
+        await m.reply_text("Invalid youtube link")
+        return
     else:
-        is_direct = False
-        query = splited
+        query = _id
     XnX = await m.reply_text("⏳")
     try:
-        await youtube_downloader(c,m,query,is_direct,"a")
+        await youtube_downloader(c,m,query, "a")
         await XnX.delete()
         return
     except KeyError:
@@ -281,15 +282,15 @@ async def video_down_up(c: Gojo, m: Message):
     except IndexError:
         await m.reply_text("**USAGE**\n /vsong [song name | link]")
         return
-    if splited.startswith("https://youtube.com"):
-        is_direct = True
-        query = splited.split("?")[0]
+    _id = get_video_id(splited)
+    if not _id:
+        await m.reply_text("Invalid youtube link")
+        return
     else:
-        is_direct = False
-        query = splited
+        query = _id
     XnX = await m.reply_text("⏳")
     try:
-        await youtube_downloader(c,m,query,is_direct,"v")
+        await youtube_downloader(c,m,query,"v")
         await XnX.delete()
         return
     except KeyError:
@@ -306,27 +307,44 @@ async def download_instareels(c: Gojo, m: Message):
     try:
         reel_ = m.command[1]
     except IndexError:
-        await m.reply_text("Give me an link to download it...")
+        await m.reply_text("Give me an instagram link to download it...")
         return
-    if not reel_.startswith("https://www.instagram.com/reel/"):
-        await m.reply_text("In order to obtain the requested reel, a valid link is necessary. Kindly provide me with the required link.")
+    
+    insta = INSTAGRAM(reel_)
+
+    if not insta.is_correct_link():
+        await m.reply_text("The link you have provided is not of instagram")
         return
-    OwO = reel_.split(".",1)
-    Reel_ = ".dd".join(OwO)
-    try:
-        await m.reply_video(Reel_)
+
+    to_edit = await m.reply_text("Trying to fetch data from the link")
+
+    content = insta.get_all()
+
+    if type(content) == str:
+        await to_edit.edit_text(content)
         return
-    except Exception:
-        try:
-            await m.reply_photo(Reel_)
-            return
-        except Exception:
-            try:
-                await m.reply_document(Reel_)
-                return
-            except Exception:
-                await m.reply_text("I am unable to reach to this reel.")
-                return
+    elif not content:
+        await to_edit.edit_text("Failed to get any media from the link")
+
+    videos = content["video"]
+    images = content["image"]
+
+    to_delete = await to_edit.edit_text("Found media in the link trying to download and upload them please wait")
+
+    to_send = []
+    if images:
+        scrapped_images = SCRAP_DATA(images).get_images()
+        for i in scrapped_images:
+            to_send.append(InputMediaPhoto(i))
+    if videos:
+        scrapped_videos = SCRAP_DATA(videos).get_videos()
+        for i in scrapped_videos:
+            to_send.append(InputMediaVideo(i))
+
+    await m.reply_media_group(to_send)
+    await to_delete.delete()
+    shutil.rmtree("./scrapped/")
+    
 
 __PLUGIN__ = "web support"
 

Powers/utils/custom_filters.py

@@ -154,7 +154,7 @@ async def admin_check_func(_, __, m: Message or CallbackQuery):
         return False
 
     # Telegram and GroupAnonyamousBot
-    if m.sender_chat:
+    if m.sender_chat and m.sender_chat.id == m.chat.id:
         return True
 
     if not m.from_user:

Powers/utils/extras.py

@@ -518,6 +518,7 @@ DECIDE = ("Yes.", "No.", "Maybe.", "Who the hell cares?",
           "No one give a damn about it")
 
 INSULT_STRINGS = [
+    "Tell me\nYou are gay becauce you are you or you are you that's why you are gay?"
     "`Owww ... Such a stupid idiot.`",
     "`Don't drink and type.`",
     "`Command not found. Just like your brain.`",

Powers/utils/sticker_help.py

@@ -85,11 +85,10 @@ async def create_sticker(
     return raw.types.InputStickerSetItem(document=sticker, emoji=emoji)
 
 
-STICKER_DIMENSIONS = (512, 512)
-
 
 async def resize_file_to_sticker_size(file_path: str, length: int = 512, width: int = 512) -> str:
     im = Image.open(file_path)
+    STICKER_DIMENSIONS = (length, width)
     if (im.width, im.height) < STICKER_DIMENSIONS:
         size1 = im.width
         size2 = im.height

Powers/utils/web_helpers.py

@@ -6,10 +6,10 @@ from urllib import parse
 from pyrogram.types import InlineKeyboardButton as IKB
 from pyrogram.types import InlineKeyboardMarkup as IKM
 from pyrogram.types import Message
-# import yt_dlp
-from pytube import YouTube
-from youtubesearchpython.__future__ import Video, VideosSearch
+from pytube import YouTube, extract
+from youtubesearchpython.__future__ import VideosSearch
 
+from Powers import youtube_dir
 from Powers.bot_class import LOGGER, MESSAGE_DUMP, Gojo
 from Powers.utils.http_helper import *
 from Powers.utils.sticker_help import resize_file_to_sticker_size
@@ -45,6 +45,15 @@ async def get_file_size(file: Message):
             size = size/1024
             return f"{round(size)} gb"
 
+def get_video_id(url):
+    try:
+        _id = extract.video_id(url)
+        if not _id:
+            return None
+        else:
+            return _id
+    except:
+        return None
 
 def get_duration_in_sec(dur: str):
     duration = dur.split(":")
@@ -57,14 +66,9 @@ def get_duration_in_sec(dur: str):
 # Gets yt result of given query.
 
 
-async def song_search(query, is_direct, max_results=1):
+async def song_search(query, max_results=1):
     yt_dict = {}
     try:
-        if is_direct:
-            vid = Video.getInfo(query)
-            query = vid["title"]
-        else:
-            query = query
         videos = VideosSearch(query, max_results)
         results = await videos.next()
     except Exception as e:
@@ -93,7 +97,7 @@ async def song_search(query, is_direct, max_results=1):
                 "uploader": i["channel"]["name"]
             }
             try:
-                thumb = {"thumbnail": i["richThumbnail"]["url"]}
+                thumb = {"thumbnail": i["thumbnails"][0]["url"]}
             except Exception:
                 thumb = {"thumbnail": None}
             dict_form.update(thumb)
@@ -103,45 +107,6 @@ async def song_search(query, is_direct, max_results=1):
             pass
     return yt_dict
 
-"""song_opts = {
-    "format": "bestaudio",
-    "addmetadata": True,
-    "key": "FFmpegMetadata",
-    "writethumbnail": True,
-    "prefer_ffmpeg": True,
-    "geo_bypass": True,
-    "nocheckcertificate": True,
-    "postprocessors": [
-        {
-            "key": "FFmpegExtractAudio",
-            "preferredcodec": "mp3",
-            "preferredquality": "480",
-        }
-    ],
-    "outtmpl": "%(id)s.mp3",
-    "quiet": True,
-    "logtostderr": False,
-}
-
-
-video_opts = {
-    "format": "best",
-    "addmetadata": True,
-    "key": "FFmpegMetadata",
-    "prefer_ffmpeg": True,
-    "geo_bypass": True,
-    "nocheckcertificate": True,
-    "postprocessors": [
-        {
-            "key": "FFmpegVideoConvertor",
-            "preferedformat": "mp4",
-        }
-    ],
-    "outtmpl": "%(id)s.mp4",
-    "logtostderr": False,
-    "quiet": True,
-}"""
-
 
 async def youtube_downloader(c: Gojo, m: Message, query: str, is_direct: bool, type_: str):
     if type_ == "a":
@@ -164,6 +129,9 @@ async def youtube_downloader(c: Gojo, m: Message, query: str, is_direct: bool, t
     except KeyError:
         return
     yt = YouTube(query)
+    if yt.age_restricted:
+        await m.reply_text("This video is age restricted")
+        return
     dicti = dicti[1]
     f_name = dicti["title"]
     views = dicti["views"]
@@ -204,19 +172,18 @@ Downloaded by: @{c.me.username}
     if song:
         audio_stream = yt.streams.filter(only_audio=True).first()
         f_path = audio_stream.download()
-        # file_path = f"./youtube_downloads/{f_name.strip()}.mp3"
-        file_path = f"./{f_name.strip()}.mp3"
+        file_path = f"{youtube_dir}{f_name.strip()}.mp3"
         os.rename(f_path, file_path)
-        await m.reply_audio(file_path, caption=cap, reply_markup=kb, duration=vid_dur, thumb=thumb, title=f_name)
-        # os.remove(f_path)
+        await m.reply_audio(file_path, caption=cap, reply_markup=kb, duration=vid_dur, thumb=thumb, title=f_name,performer=uploader)
         os.remove(file_path)
         os.remove(thumb)
         return
     elif video:
         video_stream = yt.streams.get_highest_resolution()
         file_path = video_stream.download()
-        # file_path = f"./youtube_downloads/{f_name}.mp4"
+        new_file_path = f"{youtube_dir}{f_name}.mp4"
+        os.rename(file_path, new_file_path)
         await m.reply_video(file_path, caption=cap, reply_markup=kb, duration=vid_dur, thumb=thumb)
-        os.remove(file_path)
+        os.remove(new_file_path)
         os.remove(thumb)
         return

Powers/utils/web_scrapper.py

@@ -0,0 +1,207 @@
+import os
+import re
+import time
+from typing import List
+
+import requests
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.chrome.service import Service
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.expected_conditions import \
+    presence_of_element_located
+from selenium.webdriver.support.wait import WebDriverWait
+
+from Powers import *
+
+
+class SCRAP_DATA:
+    """Class to get and handel scrapped data"""
+
+    def __init__(self, urls: List[str] or str) -> None:
+        self.urls = urls
+        self.path = "./scrapped/"
+        if not os.path.isdir(self.path):
+            os.makedirs("./scrapped/")
+
+    def get_images(self) -> list:
+        images = []
+        if isinstance(self.urls, str):
+            requested = requests.get(self.urls)
+            try:
+                name = self.path + f"img_{time.time()}.jpg"
+                with open(name, "wb") as f:
+                    f.write(requested.content)
+                images.append(name)
+            except Exception as e:
+                LOGGER.error(e)
+                requested.close()
+        else:
+            for i in self.urls:
+                if i:
+                    requested = requests.get(i)
+                else:
+                    continue
+                try:
+                    name = self.path + f"img_{time.time()}.jpg"
+                    with open(name, "wb") as f:
+                        f.write(requested.content)
+                    images.append(name)
+                except Exception as e:
+                    LOGGER.error(e)
+                    requested.close()
+                    continue
+        return images
+
+    def get_videos(self) -> list:
+        videos = []
+        if isinstance(self.urls, str):
+            if i:
+                requested = requests.get(i)
+            else:
+                return []
+            try:
+                name = self.path + f"vid_{time.time()}.mp4"
+                with open(name, "wb") as f:
+                    f.write(requested.content)
+                videos.append(name)
+            except Exception as e:
+                LOGGER.error(e)
+                requested.close()
+        else:
+            for i in self.urls:
+                if i:
+                    requested = requests.get(i)
+                else:
+                    continue
+                try:
+                    name = self.path + f"vid_{time.time()}.mp4"
+                    with open(name, "wb") as f:
+                        f.write(requested.content)
+                    videos.append(name)
+                except Exception as e:
+                    LOGGER.error(e)
+                    requested.close()
+                    continue
+        return videos
+
+
+class DRIVER:
+    """Class to make selenium driver"""
+
+    def __init__(self) -> None:
+        self.BIN = CHROME_BIN
+        self.CHROME_DRIVER = CHROME_DRIVER
+
+    def initialize_driver(self):
+        if not self.BIN:
+            LOGGER.error(
+                "ChromeBinaryErr: No binary path found! Install Chromium or Google Chrome.")
+            return (
+                None,
+                "ChromeBinaryErr: No binary path found! Install Chromium or Google Chrome.",
+            )
+
+        try:
+            options = Options()
+            options.binary_location = self.BIN
+            options.add_argument("--disable-dev-shm-usage")
+            options.add_argument("--ignore-certificate-errors")
+            options.add_argument("--disable-gpu")
+            options.add_argument("--headless=new")
+            options.add_argument("--test-type")
+            options.add_argument("--no-sandbox")
+
+            service = Service(self.CHROME_DRIVER)
+            driver = webdriver.Chrome(options, service)
+            return driver, None
+        except Exception as e:
+            LOGGER.error(f"ChromeDriverErr: {e}")
+            return None, f"ChromeDriverErr: {e}"
+
+    def driver_close(self, driver: webdriver.Chrome):
+        driver.close()
+        driver.quit()
+
+
+class INSTAGRAM(DRIVER):
+    """Class to scrap data from instagram"""
+
+    def __init__(self, url: str) -> None:
+        self.url = url
+        self.article = "article._aa6a"
+        self.ul_class = "_acay"
+        self.image_class = "x5yr21d"
+        self.video_class = "x1lliihq"
+        self.next_button = "button._afxw"
+        self.return_dict = {"image": [], "video": []}
+        super().__init__()
+
+    def is_correct_link(self):
+        return bool((re.compile(r"^https?://(?:www\.)?instagram\.com/")).match(self.url))
+
+    def get_all(self):
+        driver, error = self.initialize_driver()
+        if not driver:
+            return error
+
+        driver.get(self.url)
+        wait = WebDriverWait(driver, 30)
+        if "reel" in self.url:
+            element = wait.until(
+                presence_of_element_located((By.TAG_NAME, "video")))
+            reels = element.get_attribute("src")
+            self.driver_close(driver)
+            self.return_dict.get("video").append(reels)
+            return self.return_dict
+        elif bool((re.compile(r"^https?://(?:www\.)?instagram\.com/p/")).match(self.url)):
+            image_links = []
+            video_links = []
+            try:
+                element = wait.until(presence_of_element_located(
+                    (By.CLASS_NAME, self.ul_class)))
+
+                while True:
+                    sub_element = element.find_elements(
+                        By.CLASS_NAME, self.image_class)
+                    for i in sub_element:
+                        url = i.get_attribute("src")
+                        image_links.append(url)
+
+                    sub_element = element.find_elements(
+                        By.CLASS_NAME, self.video_class)
+                    for i in sub_element:
+                        url = i.get_attribute("src")
+                        video_links.append(url)
+
+                    try:
+                        driver.find_element(
+                            By.CSS_SELECTOR, self.next_button).click()
+                    except:  # Failed to either find the element or click on next i.e. no more media left in post
+                        break
+            except:
+                element = wait.until(presence_of_element_located(
+                    (By.CSS_SELECTOR, self.article)))
+                try:
+                    sub_element = element.find_element(By.TAG_NAME, "img")
+                    image_links.append(sub_element.get_attribute("src"))
+                except:
+                    sub_element = element.find_element(By.TAG_NAME, "video")
+                    video_links.append(sub_element.get_attribute("src"))
+
+            self.driver_close(driver)
+            # To remove duplicates here I am converting into set
+            if image_links:
+                image_links = list(set(image_links))
+            if video_links:
+                video_links = list(set(video_links))
+                for i in video_links:
+                    image_links.remove(i)
+
+            self.return_dict.get("image").extend(image_links)
+            self.return_dict.get("video").extend(video_links)
+            return self.return_dict
+
+        else:
+            return {}
+

Powers/vars.py

@@ -1,3 +1,4 @@
+from email.policy import default
 from os import getcwd
 
 from prettyconf import Configuration
@@ -34,9 +35,11 @@ class Config:
         int(i)
         for i in config(
             "WHITELIST_USERS",
-            default="",
+            default=""
         ).split(None)
     ]
+    CHROME_BIN = config("CHROME_BIN", "/app/.apt/usr/bin/google-chrome")
+    CHROME_DRIVER = config("CHROME_DRIVER", default="/app/.chromedriver/bin/chromedriver")
     GENIUS_API_TOKEN = config("GENIUS_API", default=None)
     # AuDD_API = config("AuDD_API",default=None)
     RMBG_API = config("RMBG_API", default=None)
@@ -80,3 +83,5 @@ class Development:
     TIME_ZONE = 'Asia/Kolkata'
     BDB_URI = ""
     WORKERS = 8
+    CHROME_BIN = "/app/.apt/usr/bin/google-chrome"
+    CHROME_DRIVER = "/app/.chromedriver/bin/chromedriver"

README.md

@@ -213,6 +213,10 @@ If all works well, the bot should send a message to the MESSAGE_DUMP Group!--->
 
 `WHITELIST_USERS`: A space-separated list of user IDs whitelisted, cannot be restricted.
 
+`CHROME_BIN`: Location of your chrome bin.
+
+`CHROME_DRIVER`: Location of your chrome driver.
+
 ⚠️ **Note:** In case you are passing more than one value separate them using whitespace (space) for example If I want to pass more than one PREFIX_HANDLER
 I'll pass it like `'/' '.' '!'` this.
 

app.json

@@ -104,6 +104,16 @@
         "required": false,
         "value": "8"
       },
+      "CHROME_DRIVER": {
+        "description": "Location of chrome driver",
+        "required": false,
+        "value": "/app/.chromedriver/bin/chromedriver"
+      },
+      "CHROME_BIN" : {
+        "description": "Location of chrome bin",
+        "required": false,
+        "value": "/app/.apt/usr/bin/google-chrome"
+      },
       "ENV": {
         "description": "Setting this to ANYTHING will enable environment variables. Leave it as it is",
         "required": true,
@@ -114,6 +124,15 @@
       {
         "url":"https://github.com/jonathanong/heroku-buildpack-ffmpeg-latest"
       },
+      {
+        "url": "https://github.com/chrismytton/heroku-buildpack-jq"
+      },
+      {
+        "url":"https://github.com/heroku/heroku-buildpack-google-chrome"
+      },
+      {
+        "url":"https://github.com/heroku/heroku-buildpack-chromedriver"
+      },
       {
         "url": "heroku/python"
       }

requirements.txt

@@ -27,6 +27,7 @@ regex==2023.12.25; python_version >= "3.6"
 requests==2.31.0
 rfc3986==1.5.0; python_version >= "3.7"
 search-engine-parser==0.6.8
+selenium==4.18.1
 six==1.16.0; python_version >= "2.7" and python_full_version < "3.0.0" or python_full_version >= "3.3.0"
 sniffio==1.3.0; python_full_version >= "3.6.2" and python_version >= "3.7"
 soupsieve==2.4; python_version >= "3.6" and python_full_version >= "3.6.0"