screenshot-scout/screenshot_scout.py

57 lines
1.9 KiB
Python

import asyncio
import httpx
from playwright.async_api import async_playwright, TimeoutError as PlaywrightTimeoutError
SCREENSHOT_PATH = "/tmp/screenshot.png"
TELEGRAM_BOT_TOKEN = "8865512961:AAG4VVJk7S58oAF5Ps00xgwfoPjXobbcm1E"
TELEGRAM_CHAT_ID = "-1004385538803"
TELEGRAM_API_URL = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendPhoto"
TARGET_URL = "https://www.google.com"
async def capture_screenshot() -> bool:
try:
async with async_playwright() as p:
browser = await p.chromium.launch(headless=True)
page = await browser.new_page()
await page.goto(TARGET_URL, wait_until="networkidle", timeout=30000)
await page.screenshot(path=SCREENSHOT_PATH, full_page=True)
await browser.close()
print(f"[+] Screenshot saved to {SCREENSHOT_PATH}")
return True
except PlaywrightTimeoutError:
print(f"[-] Timeout loading {TARGET_URL}")
return False
except Exception as e:
print(f"[-] Error: {e}")
return False
def send_screenshot() -> bool:
try:
with open(SCREENSHOT_PATH, "rb") as photo:
with httpx.Client(timeout=30.0) as client:
resp = client.post(
TELEGRAM_API_URL,
data={"chat_id": TELEGRAM_CHAT_ID, "caption": f"Screenshot of {TARGET_URL}"},
files={"photo": ("screenshot.png", photo, "image/png")},
)
if resp.status_code == 200 and resp.json().get("ok"):
print("[+] Screenshot sent to Telegram!")
return True
else:
print(f"[-] Telegram error: {resp.text}")
return False
except Exception as e:
print(f"[-] Send failed: {e}")
return False
async def main():
if await capture_screenshot():
send_screenshot()
if __name__ == "__main__":
asyncio.run(main())