修改打开网页方法

engine/browser_engine.py

@@ -19,7 +19,7 @@ class BrowserEngine(ReadConfig):
         self.driver = webdriver.Chrome(self._driver_path)
         self.open_browser()
         self.login_znfz()
-        self.get_page()
+        self.format_url()
         self.driver.xpath = self.driver.find_elements_by_xpath
 
     def open_browser(self):
@@ -68,15 +68,19 @@ class BrowserEngine(ReadConfig):
         self.driver.find_element_by_xpath('//*[@id="root"]/div/div/div[2]/form/div[4]/div/div/span/button').click()
         return self.driver
 
-    def get_page(self):
-        time.sleep(1)
-        if self.url in ['LOGIN', 'INDEX']:
-            return
-        if self.url != 'URL':
-            url = self.get_testUrl(self.url)
-            url = url.format(bmsah=self.get_caseId('caseId')) if '{' in url else url
-            self.logger.info("the testUrl : %s" % url)
-            self.driver.get(url)
+    def format_url(self):
+        url = self.get_testUrl(self.url)
+        if '{' in url:
+            self.daoqiezui_url = url.format(bmsah=self.get_caseId('caseId_dqz'),ysay='盗窃罪')
+            self.guyishanghaizui_url = url.format(bmsah=self.get_caseId('caseId_gyshz'),ysay='故意伤害罪')
+            self.DQZ = self.daoqiezui_url
+            self.GYSHZ = self.guyishanghaizui_url
+
+
+    def get_page(self,url):
+
+        self.logger.info("the testUrl : %s" % url)
+        self.driver.get(url)
 
     # def load_image(self):
     #     print(self.image_path)