browser.py 文件源码

def __init__(self, browserName='chrome', executable_path=None, remoteAddress=None):
        """
        ???????
        :param   browserName ??????? chrome firefox ie
        :param   executable_path ????
        :param   remoteAddress ????????
        :Usage:
        browser = browser('chrome') # ????
        path = os.path.abspath('./driver/chromedriver.exe')
        browser = browser('chrome',executable_path=path) # ???????????
        browser = browser('chrome',remoteAddress='http://127.0.0.1:4444/wd/hub') # ???????????
        """
        # ??driver????
        driver_path = configinfo.driver_path
        os.environ['PATH'] = os.environ['PATH'] + ';' + driver_path

        if remoteAddress is not None:
            desired_capabilities = {'platform': 'ANY', 'version': '',
                                    'javascriptEnabled': True}
            if browserName.lower() == 'chrome':
                dc['browserName'] = 'chrome'
            elif browserName.lower() == 'firefox':
                dc['browserName'] = 'firefox'
                dc['marionette'] = False
            elif browserName.lower() == 'ie':
                dc['browserName'] = 'internet explorer'
            else:
                log.error('??????????????????')
                raise NameError('?????????chrome firefox ie')

            self.driver = webdriver.Remote(command_executor=remoteAddress,
                                           desired_capabilities=desired_capabilities)
            log.info(browserName + '??????')
        else:
            # ?????????
            if executable_path is not None:
                dir_path = os.path.abspath(executable_path)
                if os.path.isfile(dir_path):
                    dir_path = os.path.dirname(dir_path)
                os.environ['PATH'] = os.environ['PATH'] + ';' + dir_path
                log.info(dir_path + '?????????')
            if browserName.lower() == 'chrome':
                self.driver = webdriver.Chrome()
            elif browserName.lower() == 'firefox':
                self.driver = webdriver.Firefox()
            elif browserName.lower() == 'ie':
                self.driver = webdriver.Ie()
            else:
                log.error('??????????????????')
                raise NameError('?????????chrome firefox ie')
            log.info(browserName + '????')