base--PageBase

import time,os
from selenium import  webdriver
from cwp.util.logger import logger


logger = logger(logger='base_page').getlog()

class PageBase(object):
    def __init__(self,driver = None):
        if driver is None:
            os.system('taskkill / im chromedriver.exe /f')
            self.driver = webdriver.Chrome()
        else:
            self.driver = driver


    def get_element(self,*locator):
        logger.info('查找元素 %s'%str(locator))
        return self.driver.find_element(*locator)

    def wait_element(self,*locator):
        ele = None
        count = 0
        while ele is None:
            count = count + 1
            try:
                ele = self.driver.find_element(*locator)
            except:
                pass
            flag = ele is not None
            logger.info("查找元素第%d次 %s %s" %(count,flag,str(locator)))
            time.sleep(0.1)
            if count > 99:
                logger.info("没有找到元素 %s " %str(locator))
                break
        return ele