sintamar commited on
Commit
6b74d80
·
verified ·
1 Parent(s): 49c405a

Update extract.py

Browse files
Files changed (1) hide show
  1. extract.py +3 -4
extract.py CHANGED
@@ -18,12 +18,11 @@ def take_webdata(url):
18
  page_title = wd.title
19
  screenshot = wd.get_screenshot_as_png()
20
  html = wd.execute_script("return document.documentElement.outerHTML;")
21
- print(html)
22
-
23
  except WebDriverException as e:
24
- return Image.new('RGB', (1, 1)), page_title
25
  finally:
26
  if wd:
27
  wd.quit()
28
 
29
- return Image.open(BytesIO(screenshot)) , page_title
 
18
  page_title = wd.title
19
  screenshot = wd.get_screenshot_as_png()
20
  html = wd.execute_script("return document.documentElement.outerHTML;")
21
+
 
22
  except WebDriverException as e:
23
+ return page_title
24
  finally:
25
  if wd:
26
  wd.quit()
27
 
28
+ return html , page_title