forked from STI/Astroport.ONE
43 lines
1.2 KiB
Python
Executable File
43 lines
1.2 KiB
Python
Executable File
import asyncio
|
|
import sys
|
|
from pyppeteer import launch
|
|
|
|
async def take_screenshot(url, output_file, width, height):
|
|
browser = await launch(
|
|
headless=True,
|
|
args=['--disable-gpu'] # Add this line to disable GPU
|
|
)
|
|
page = await browser.newPage()
|
|
|
|
try:
|
|
await page.setViewport({'width': width, 'height': height}) # Set the viewport size
|
|
await page.goto(url, waitUntil='networkidle2')
|
|
await page.screenshot({'path': output_file})
|
|
except Exception as e:
|
|
print(f"An error occurred: {e}")
|
|
finally:
|
|
await browser.close()
|
|
|
|
def main():
|
|
if len(sys.argv) != 5:
|
|
print("Usage: python page_screenshot.py <URL> <output_file> <width> <height>")
|
|
sys.exit(1)
|
|
|
|
url = sys.argv[1]
|
|
output_file = sys.argv[2]
|
|
width = int(sys.argv[3])
|
|
height = int(sys.argv[4])
|
|
|
|
loop = asyncio.new_event_loop()
|
|
asyncio.set_event_loop(loop)
|
|
|
|
try:
|
|
loop.run_until_complete(asyncio.wait_for(take_screenshot(url, output_file, width, height), timeout=30))
|
|
except asyncio.TimeoutError:
|
|
print("Timeout: The operation took too long to complete and has been terminated.")
|
|
finally:
|
|
loop.close()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|