Question

我正在研究一些Selenium脚本，以测试不同设备，浏览器和平台上的网站。我可以使用相同的代码来使用脚本，除了我定义命令执行程序URL和浏览器功能的两行。我试图构建一个脚本，我可以使用命令行参数定义这些行。

这是我的代码：

from selenium import webdriver
import time
import sys
import getopt
def main(argv):
    #define desired browser capabilities
    desktopCapabilities = {'browserName': 'chrome'} #change browserName to: 'firefox' 'chrome' 'safari' 'internet explorer'
    iosCapabilities = {'platformName': 'iOS' ,'platformVersion': '8.1' ,'deviceName': 'iPad Air','browserName': 'Safari'}
    androidCapabilities = {'chromeOptions': {'androidPackage': 'com.android.chrome'}}
    # Establish the command executor URL
    desktopExecutor = 'http://127.0.0.1:4444/wd/hub'
    iosExecutor = 'http://127.0.0.1:4723/wd/hub'
    androidExecutor = 'http://127.0.0.1:9515'

    cmdExecutor = desktopExecutor
    browserCapabilities = desktopCapabilities
    try:
      opts, args = getopt.getopt(argv,"he:c:",["executor=","capabilities="])
    except getopt.GetoptError:
      print 'test.py -e <executor> -c <capabilities>'
      sys.exit(2)
    for opt, arg in opts:
      if opt == '-h':
         print 'test.py -e <executor> -c <capabilities>'
         sys.exit()
      elif opt in ("-e", "--executor"):
         cmdExecutor = arg
      elif opt in ("-c", "--capabilities"):
         browserCapabilities = arg

    print 'Command executor is:', cmdExecutor
    print 'Desired capabilities are:', browserCapabilities
    driver = webdriver.Remote(command_executor=cmdExecutor, desired_capabilities=browserCapabilities)
    driver.get("http://google.com")
    time.sleep(5)
    driver.quit()
if __name__ == "__main__":
   main(sys.argv[1:])

如果我不通过命令行添加任何参数，则此代码按预期运行。如果我用：

运行它也可以

python test.py -e 'http://127.0.0.1:4444/wd/hub'

如果我使用以下命令运行它会中断，因为-c不作为字典传递：

python test.py -e 'http://127.0.0.1:4444/wd/hub' -c {'browserName': 'firefox'}

如何通过以下方式运行此操作：

python test.py -e iosExecutor -c iosCapabilities

这是我运行上述命令时得到的输出：

 python my_script.py -e iosExecutor --capabilities iosCapabilities
Command executor is: iosExecutor
Desired capabilities are: iosCapabilities
Traceback (most recent call last):
  File "my_script.py", line 38, in <module>
    main(sys.argv[1:])
  File "my_script.py", line 33, in main
    driver = webdriver.Remote(command_executor=cmdExecutor, desired_capabilities=browserCapabilities)
  File "/Library/Python/2.7/site-packages/selenium/webdriver/remote/webdriver.py", line 62, in __init__
    raise WebDriverException("Desired Capabilities must be a dictionary")
selenium.common.exceptions.WebDriverException: Message: Desired Capabilities must be a dictionary

基本上它正在运行，好像我像这样通过第33行：

driver = webdriver.Remote(command_executor="iosExecutor", desired_capabilities="iosCapabilities")

如果我使用＆＃34; iosExecutor＆＃34;硬编码第15行和第16行，它也可以工作。和＆＃34; iosCapabilites＆＃34;所以这告诉我它是如何从CLI传递信息的。

任何建议都会很棒。我对这个（编程）很陌生，所以我猜测可能有更好的方法来做到这一点，但谷歌还没有为我清除它。

Answer 1

使用argparse可以让事情变得更轻松。对于capabilities，您可以使用json.loads或ast.literal_eval作为type，例如，在此完成：

对于executor，要么将字符串作为字符串传递，要么定义用户友好的映射，例如：

EXECUTORS = {
    'desktop': 'http://127.0.0.1:4444/wd/hub',
    'ios': 'http://127.0.0.1:4723/wd/hub',
    'android': 'http://127.0.0.1:9515'
}

以下是代码最终的样子：

import ast
import time
import argparse

from selenium import webdriver


EXECUTORS = {
    'desktop': 'http://127.0.0.1:4444/wd/hub',
    'ios': 'http://127.0.0.1:4723/wd/hub',
    'android': 'http://127.0.0.1:9515'
}

parser = argparse.ArgumentParser(description='My program.')
parser.add_argument('-c', '--capabilities', type=ast.literal_eval)
parser.add_argument('-e', '--executor', type=str, choices=EXECUTORS)

args = parser.parse_args()

driver = webdriver.Remote(command_executor=EXECUTORS[args.executor], desired_capabilities=args.capabilities)
driver.get("http://google.com")
time.sleep(5)
driver.quit()

示例脚本运行：

$ python test.py -e android -c "{'chromeOptions': {'androidPackage': 'com.android.chrome'}}"
$ python test.py -e ios -c "{'platformName': 'iOS' ,'platformVersion': '8.1' ,'deviceName': 'iPad Air','browserName': 'Safari'}" 
$ python test.py -e desktop -c "{'browserName': 'chrome'}"

而且，作为奖励，您可以获得由argparse神奇地制作的内置帮助：

$ python test.py --help
usage: test.py [-h] [-c CAPABILITIES] [-e {android,ios,desktop}]

My program.

optional arguments:
  -h, --help            show this help message and exit
  -c CAPABILITIES, --capabilities CAPABILITIES
  -e {android,ios,desktop}, --executor {android,ios,desktop}

Python：命令行参数与硬编码行的传递方式不同

1 个答案: