from selenium import webdriver
import sys

if len(sys.argv) != 3:
    print("Usage: python screenshot.py <url> <output_file>")
    sys.exit(1)

url = sys.argv[1]
output_file = sys.argv[2]

# 设置无头模式
options = webdriver.ChromeOptions()
options.add_argument('headless')

# 启动浏览器并打开网页
driver = webdriver.Chrome(options=options)
driver.get(url)

# 截图并保存
driver.save_screenshot(output_file)
driver.quit()