用于将函数输出与传递给脚本的参数进行比较的 Python 脚本

Question

我编写了一个 python 脚本来进行 api 调用。该脚本返回一个整数(数字)。我想将此输出与将传递给脚本的参数进行比较。

例如，如果 x 是脚本返回的数字，我想执行如下脚本 python test.py 20 并将 x 与 20 进行比较。

请帮忙。

下面是脚本:

import json
import os, sys
import urllib2

def main():
    data = json.load(urllib2.urlopen('some url'))
    val = data.keys()[0]
    print(val)

if __name__ == "__main__":
    try:
        main()
    except Exception as e:
        print ('!!FAIL {0}!!!'.format(e))

Answer 1

您可以通过 sys.argv 访问命令行参数列表:

import json
import os, sys
import urllib2

def main(expected_value):
    data = json.load(urllib2.urlopen('some url'))
    matched = data.keys()[0] == expected_value
    print('expected value {}'.format('matched' if matched else 'mismatch'))
    # and you could return a bool...
    return matched

if __name__ == "__main__":
    try:
        main(sys.argv[1])    # pass the first command line argument to main()
    except Exception as e:
        print('!!FAIL {0}!!!'.format(e))