def save_screenshot(self, img_name=""):
screenshot_date_path = "../err_png/" + datetime.today().strftime(
"%Y-%m-%d")
if not os.path.exists(screenshot_date_path):
os.mkdir(screenshot_date_path)
current_date = datetime.today().strftime("%Y-%m-%d-%H-%M-%S")
screenshot_date_file = os.path.join(screenshot_date_path,
f"{current_date}_{img_name}.png")
log.info(screenshot_date_file)
self.driver.save_screenshot(screenshot_date_file)
log.info("screenshot save ok")
def finds(self, ele, timeout=7):
try:
WebDriverWait(self.driver, timeout).until(
EC.visibility_of_element_located((By.CSS_SELECTOR, ele)))
except:
WebDriverWait(self.driver, timeout).until(
EC.presence_of_element_located((By.CSS_SELECTOR, ele)))
try:
elements = self.driver.find_elements(By.CSS_SELECTOR, ele)
log.info("find {ele} element")
except Exception as e:
self.save_screenshot(ele)
log.error(f"{ele} not found")
log.exception(e)
return elements
def sleep(secs):
time.sleep(secs)
log.info(f"wait {secs} second")
def open_url(self, url):
self.driver.get(url)
log.info(f"open {url}")
def setup(self):
log.info("strat TestBaidu")
self.index = Index()
def goto_index(self):
self.open_url("https://work.weixin.qq.com/wework_admin/frame")
log.info("start get cookies")
self.get_cookies()
log.info("get cookies ok")
return self