rabin 7 lat temu
rodzic
commit
4c75070456
2 zmienionych plików z 8 dodań i 2 usunięć
  1. 1 0
      service/__load__.py
  2. 7 2
      service/ssgdfs.py

+ 1 - 0
service/__load__.py

@@ -7,6 +7,7 @@
 """
 import unittest
 import requests
+from pyvirtualdisplay import Display
 from selenium import webdriver
 from time import sleep
 from selenium.webdriver.common.keys import Keys

+ 7 - 2
service/ssgdfs.py

@@ -71,7 +71,10 @@ class Ssgdfs(object):
 			return True
 
 	def open(self):
-		self.driver = webdriver.Remote(command_executor=Demeter.config['setting']['phantomjs'], desired_capabilities=DesiredCapabilities.PHANTOMJS)
+		self.display = Display(visible=0, size=(800, 600))
+		self.display.start()
+		#self.driver = webdriver.Remote(command_executor=Demeter.config['setting']['phantomjs'], desired_capabilities=DesiredCapabilities.PHANTOMJS)
+		self.driver = webdriver.Firefox()
 
 	def login(self):
 		self.driver.get(self.config['login_link'])
@@ -153,4 +156,6 @@ class Ssgdfs(object):
 		WebDriverWait(self.driver, 100).until(lambda x: x.find_element_by_class_name(name).is_displayed())
 
 	def close(self):
-		self.driver.close()
+		self.driver.quit()
+		#self.driver.close()
+		self.display.stop()