Commit a96392d1 authored by liyang's avatar liyang

fix:debug ptt

parent 8229ccc5
......@@ -28,42 +28,17 @@ def create(option=None, using_user_data=True, web_browser="firefox"):
# 安装或升级 chromedriver
# chromedriver_autoinstaller.install()
def get_user_data_dir():
"""
:return:
"""
# 获取现有Chrome浏览器用户数据目录
user_data_dir = os.path.join(os.path.abspath("../"), 'network-assets-reptile', 'user_data')
# if platform.system() == 'Windows':
# if web_browser == "firefox":
# user_data_dir = os.path.join(os.environ['USERPROFILE'], 'AppData', 'Local', 'Mozilla', 'Firefox',
# 'Profiles')
# else:
# user_data_dir = os.path.join(os.environ['USERPROFILE'], 'AppData', 'Local', 'Google', 'Chrome',
# 'User Data')
# elif platform.system() == 'Linux':
# if web_browser == "firefox":
# user_data_dir = os.path.join(os.path.expanduser('~'), '.config', 'Firefox', 'Profiles','huqg7mpy.default-release')
# else:
# user_data_dir = os.path.join(os.path.expanduser('~'), '.config', 'google-chrome')
# elif platform.system() == 'Darwin':
# if web_browser == "firefox":
# user_data_dir = os.path.join(os.path.expanduser("~"), 'Library', 'Application Support', 'Firefox',
# 'Profiles','huqg7mpy.default-release')
# else:
# user_data_dir = os.path.join(os.path.expanduser("~"), 'Library', 'Application Support', 'Google',
# 'Chrome')
# else:
# raise Exception('Unsupported operating system')
return user_data_dir
options = ""
browser = ""
# 获取现有Chrome浏览器用户数据目录
user_data_dir = os.path.join(os.path.abspath("../"), 'network-assets-reptile', 'user_data')
if web_browser == "firefox":
options = webdriver.FirefoxOptions()
else:
options = webdriver.ChromeOptions()
if option is not None:
for value in option:
options.add_argument(value)
......@@ -71,11 +46,11 @@ def create(option=None, using_user_data=True, web_browser="firefox"):
if using_user_data:
# 添加用户数据目录参数,启用浏览器的持久性会话,可以保存登录状态和Cookie
if web_browser == "firefox":
firefox_profile_path = get_user_data_dir() # 将此处替换为你的Firefox用户数据目录路径
firefox_profile_path = user_data_dir # 将此处替换为你的Firefox用户数据目录路径
profile = FirefoxProfile(profile_directory=firefox_profile_path)
options.profile = profile
else:
options.add_argument(f'--user-data-dir={get_user_data_dir()}')
options.add_argument(f'--user-data-dir={user_data_dir}')
# if sys.platform.startswith('linux'):
# print("当前系统是 Linux")
......@@ -94,9 +69,11 @@ def create(option=None, using_user_data=True, web_browser="firefox"):
extend = ""
if platform.system() == 'Windows':
extend = ".exe"
geckodriver_path = os.path.join(os.path.abspath("./"), "drive", "geckodriver", extend)
print(f"windows系统需要设置驱动程序路径:{geckodriver_path}")
options.binary_location = geckodriver_path
else:
extend = ""
geckodriver_path = os.path.join(os.path.abspath("./"), "drive", "geckodriver", extend)
browser = webdriver.Firefox(options=options)
else:
# 创建Chrome浏览器对象并传入选项
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment