Question

我想创建一个硒脚本并使用argparse从命令行中选择浏览器。这就是我所拥有的-当我运行test.py chrome时，什么也没发生。

test.py：

https://repl.it/repls/ProbableHeavenlySystem

from selenium import webdriver
import argparse


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('chrome')
    parser.add_argument('firefox')
    parser.parse_args()
    args = parser.parse_args()


def pick_browser(args):
    if args.chrome == 'chrome':
        return args.webdriver.Chrome(executable_path='C:/pathto/chromedriver.exe')
elif args.firefox == 'firefox':
    return args.webdriver.Firefox(
        executable_path='C:/pathto/geckodriver.exe')


if __name__ == '__main__':
    main()

感谢您的帮助！

Answer 1

您可以使用sys.argv：

import sys

print(sys.argv)

在命令行类型python script.py chrome和脚本文件中：

import sys

print(sys.argv[1]) # prints chrome

Here，您可以找到一个不错的教程。

selenium + argparse-使用命令行参数更改浏览器

1 个答案: