21 lines
419 B
Python
21 lines
419 B
Python
|
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]
|
|||
|
|
|
|||
|
|
# <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ͷģʽ
|
|||
|
|
options = webdriver.ChromeOptions()
|
|||
|
|
options.add_argument('headless')
|
|||
|
|
|
|||
|
|
# <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ҳ
|
|||
|
|
driver = webdriver.Chrome(options=options)
|
|||
|
|
driver.get(url)
|
|||
|
|
|
|||
|
|
# <20><>ͼ<EFBFBD><CDBC><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
|
driver.save_screenshot(output_file)
|
|||
|
|
driver.quit()
|