Bläddra i källkod

Improved webss

TheHamkerCat 5 år sedan
förälder
incheckning
2e55a0cc4e
2 ändrade filer med 11 tillägg och 22 borttagningar
  1. 0 1
      requirements.txt
  2. 11 21
      wbb/modules/webss.py

+ 0 - 1
requirements.txt

@@ -1,5 +1,4 @@
 pykeyboard
-pyppeteer
 requests
 python-arq
 TgCrypto

+ 11 - 21
wbb/modules/webss.py

@@ -1,11 +1,10 @@
 from wbb import app, app2, SUDOERS
 from wbb.utils.errors import capture_err
+from wbb.utils.fetch import fetch
 from pyrogram import filters
-from pyppeteer import launch
-import os
-import asyncio
 import time
 
+
 __MODULE__ = "WebSS"
 __HELP__ = "/webss | .webss [URL] - Take A Screenshot Of A Webpage"
 
@@ -17,24 +16,15 @@ async def take_ss(_, message):
     if len(message.command) != 2:
         await message.reply_text("Give A Url To Fetch Screenshot.")
         return
-    start_time = time.time()
     url = message.text.split(None, 1)[1]
-    m = await message.reply_text("**Launching Browser**")
-    browser = await launch(
-            headless=True,
-            args=['--no-sandbox', '--disable-setuid-sandbox']
+    start_time = time.time()
+    m = await message.reply_text("**Taking Screenshot**")
+    screenshot = await fetch(f"https://patheticprogrammers.cf/ss?site={url}")
+    await m.edit("**Uploading**")
+    end_time = time.time()
+    await app.send_photo(
+            message.chat.id,
+            photo=screenshot['url'],
+            caption=(f"{url}\n__Took {round(end_time - start_time)} Seconds.__")
             )
-    await m.edit("**Loading Page**")
-    page = await browser.newPage()
-    await page.setViewport({'width': 2000, 'height': 1381, 'deviceScaleFactor': 2.0})
-    await page.goto(url, {'waitUntil': 'load'})
-    await asyncio.sleep(1)
-    await m.edit("**Taking Screenshot**")
-    await page.screenshot({'path': 'webss.png'})
     await m.delete()
-    end_time = time.time()
-    await message.reply_photo(
-            'webss.png',
-            caption=f"`{url}`\n__Took {round(end_time - start_time)} Seconds.__")
-    await browser.close()
-    os.remove('webss.png')