microsoft_edge.py 文件源码

def get_driver(self, task):
        """Get the webdriver instance"""
        from selenium import webdriver
        capabilities = webdriver.DesiredCapabilities.EDGE.copy()
        extension_src = os.path.join(os.path.abspath(os.path.dirname(__file__)),
                                     'support', 'edge', 'extension')
        extension_dir = os.path.join(os.environ.get('LOCALAPPDATA'), 'Packages',
                                     'Microsoft.MicrosoftEdge_8wekyb3d8bbwe',
                                     'LocalState', 'wptagent')
        if not os.path.isdir(extension_dir):
            os.makedirs(extension_dir)
        files = os.listdir(extension_src)
        for file_name in files:
            try:
                src = os.path.join(extension_src, file_name)
                if os.path.isfile(src):
                    shutil.copy(src, extension_dir)
            except Exception:
                pass
        capabilities['extensionPaths'] = [extension_dir]
        driver = webdriver.Edge(executable_path=self.path, capabilities=capabilities)
        return driver