1
0

browser_common.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. from selenium import webdriver
  2. import time
  3. import networkx as nx
  4. import dataset
  5. import pickle
  6. import codecs
  7. from selenium.webdriver.common.keys import Keys
  8. import sys
  9. import os
  10. import time
  11. import re
  12. class JBrowser:
  13. def __init__(self):
  14. os.chdir("C:\\Program Files\\Google\\Chrome\\Application\\")
  15. def set_profile_path(self,profilepath):
  16. self.profilepath=profilepath
  17. option = webdriver.ChromeOptions()
  18. option.add_argument('--disable-web-security')
  19. option.add_argument('--allow-running-insecure-content')
  20. option.add_argument("--user-data-dir=C:\\Users\\jared\\AppData\\Local\\Google\\Chrome\\User Data\\"+self.profilepath+"\\")
  21. self.option=option
  22. driver = webdriver.Chrome(options=option)
  23. self.driver=driver
  24. # executor_url = driver.command_executor._url
  25. # session_id = driver.session_id
  26. def scrolling(self,pgnum):
  27. ub = self.driver.find_element_by_css_selector('body')
  28. for i in range(pgnum):
  29. ub.send_keys(Keys.PAGE_DOWN)
  30. if pgnum>1:
  31. time.sleep(0.5)
  32. def get(self,url):
  33. self.driver.get(url)
  34. def get_driver(self):
  35. return self.driver