jonathanjordan21 commited on
Commit
ef19ec0
·
verified ·
1 Parent(s): deeaf90

Update extract.py

Browse files
Files changed (1) hide show
  1. extract.py +5 -0
extract.py CHANGED
@@ -80,6 +80,11 @@ def get_vehicle_info(plate_number: str):
80
 
81
  driver.implicitly_wait(3)
82
 
 
 
 
 
 
83
  page_title = driver.title
84
  screenshot = driver.get_screenshot_as_png()
85
 
 
80
 
81
  driver.implicitly_wait(3)
82
 
83
+
84
+ scroll_height = driver.execute_script("return document.body.scrollHeight")
85
+ driver.set_window_size(1920, scroll_height + 200) # force full-page height
86
+ time.sleep(1)
87
+
88
  page_title = driver.title
89
  screenshot = driver.get_screenshot_as_png()
90