Update screenlong.py

userbot/plugins/screenlong.py

@@ -1,63 +1,89 @@
 """Take screenshot of any website
-Syntax: .screenlong <Website URL>"""
+
+Syntax: .screencapture <Website URL>"""
+
+
 
 import io
-import traceback
-from datetime import datetime
-from selenium import webdriver
+
+import requests
+
 from telethon import events
+
 from userbot.utils import admin_cmd
 
 
-@borg.on(admin_cmd("screenlong (.*)"))
+
+
+
+@borg.on(admin_cmd("screencapture (.*)"))
+
 async def _(event):
+
     if event.fwd_from:
+
         return
-    if Config.GOOGLE_CHROME_BIN is None:
-        await event.edit("need to install Google Chrome. Module Stopping.")
+
+    if Config.SCREEN_SHOT_LAYER_ACCESS_KEY is None:
+
+        await event.edit("Need to get an API key from https://screenshotlayer.com/product \nModule stopping!")
+
         return
+
     await event.edit("Processing ...")
-    start = datetime.now()
-    try:
-        chrome_options = webdriver.ChromeOptions()
-        chrome_options.add_argument('--ignore-certificate-errors')
-        chrome_options.add_argument("--test-type")
-        chrome_options.add_argument("--headless")
-        # https://stackoverflow.com/a/53073789/4723940
-        chrome_options.add_argument('--no-sandbox')
-        chrome_options.add_argument('--disable-dev-shm-usage')
-        chrome_options.binary_location = Config.GOOGLE_CHROME_BIN
-        await event.edit("Starting Google Chrome BIN")
-        driver = webdriver.Chrome(chrome_options=chrome_options)
-        input_str = event.pattern_match.group(1)
-        driver.get(input_str)
-        await event.edit("Calculating Page Dimensions")
-        height = driver.execute_script("return Math.max(document.body.scrollHeight, document.body.offsetHeight, document.documentElement.clientHeight, document.documentElement.scrollHeight, document.documentElement.offsetHeight);")
-        width = driver.execute_script("return Math.max(document.body.scrollWidth, document.body.offsetWidth, document.documentElement.clientWidth, document.documentElement.scrollWidth, document.documentElement.offsetWidth);")
-        await event.edit("Painting web-page")
-        driver.set_window_size(width + 100, height + 100)
-        # Add some pixels on top of the calculated dimensions 
-        # for good measure to make the scroll bars disappear
-        im_png = driver.get_screenshot_as_png()
-        # saves screenshot of entire page
-        driver.close()
-        await event.edit("Stopping Google Chrome BIN")
-        message_id = event.message.id
-        if event.reply_to_msg_id:
-            message_id = event.reply_to_msg_id
-        with io.BytesIO(im_png) as out_file:
-            out_file.name = "@UniBorg.ScreenCapture.PNG"
-            await borg.send_file(
-                event.chat_id,
-                out_file,
-                caption=input_str,
-                force_document=True,
-                reply_to=message_id,
-                allow_cache=False,
-                silent=True
-            )
-        end = datetime.now()
-        ms = (end - start).seconds
-        await event.edit(f"Completed screencapture Process in {ms} seconds")
-    except Exception:
-        await event.edit(traceback.format_exc())
+
+    sample_url = "https://api.screenshotlayer.com/api/capture?access_key={}&url={}&fullpage={}&viewport={}&format={}&force={}"
+
+    input_str = event.pattern_match.group(1)
+
+    response_api = requests.get(sample_url.format(
+
+        Config.SCREEN_SHOT_LAYER_ACCESS_KEY,
+
+        input_str,
+
+        "1",
+
+        "2560x1440",
+
+        "PNG",
+
+        "1"
+
+    ))
+
+    # https://stackoverflow.com/a/23718458/4723940
+
+    contentType = response_api.headers['content-type']
+
+    if "image" in contentType:
+
+        with io.BytesIO(response_api.content) as screenshot_image:
+
+            screenshot_image.name = "screencapture.png"
+
+            try:
+
+                await borg.send_file(
+
+                    event.chat_id,
+
+                    screenshot_image,
+
+                    caption=input_str,
+
+                    force_document=True,
+
+                    reply_to=event.message.reply_to_msg_id
+
+                )
+
+                await event.delete()
+
+            except Exception as e:
+
+                await event.edit(str(e))
+
+    else:
+
+        await event.edit(response_api.text)