shell命令运行带有参数的python脚本，该参数是一个列表[重复]

Question

我需要使用shell命令来运行python脚本

但是args是python脚本的列表：

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import sys

def main(task_list):
    print('task_list:',task_list)

if __name__=='__main__':
main(sys.argv[1])

当我使用shell python时

python scrpit.py [("store", "IDCODE", 18116485, 88779)]

系统显示："（'有一个错误"

如何将列表转换为脚本？

（“存储”、“IDCODE”、18116485、88779）是一个参数

Answer 1

这里有一个您可能喜欢的替代方案：将其作为单个字符串发送，并将其作为编码的json导入。只需“导入json”并将main（）调用更改为如下所示：

main(json.JSONDecoder().decode(sys.argv[1]))

并调用带有单引号内参数的脚本，但省略（），使其看起来像正确的json编码：

python scrpit.py '["store", "IDCODE", 18116485, 88779]'

这给了我一个输出，看起来像：

('task_list:', [u'store', u'IDCODE', 18116485, 88779])

这也将允许您发送更复杂的结构，如：

python try.py '{"p2": {"a": 1, "b": 2}, "p1": ["a", "B", "c"]}'
('task_list:', {u'p2': {u'a': 1, u'b': 2}, u'p1': [u'a', u'B', u'c']})

如果不分析json解码器的内部，我不能说它是否会直接打开任何安全漏洞，但我可以说，您必须小心使用解码的值，以确保不会添加任何自己的值。

希望这能有所帮助。

Answer 2

一个更干净的方法来做你所要求的（主要是如果你需要其他参数）是从另一个参数中识别你的列表并枚举允许的类型（如果需要转换），这看起来像这样：

import sys, getopt
sh_params, args = getopt.getopt(sys.argv[1:], "l:")

def convert_type(elem):
    for fn in (int, lambda e: e.strip('"')): # Lambda is the string conversion function and int is auto-cast fct
        try:
            return fn(elem)
        except ValueError:
            pass
    return elem

my_list = []
for param, val in sh_params:
    if param == "-l":
        my_list = [convert_type(elem) for elem in val.strip('[]').split(', ')]
print(my_list)

这是命令

$ python3 scrpit.py -l '["store", "IDCODE", 18116485, 88779]'
$ ['store', 'IDCODE', 18116485, 88779]

记住，PM 2Ring的替代方案更好！

Answer 3

shell向程序传递一个字符串数组，Python将其转换为字符串列表。shell不知道Python语法。你可以把你的程序叫做

$ python scrpit.py '[("store", "IDCODE", 18116485, 88779)]'

然后使用ast.literal\u eval将其解析为一个列表。

import sys
import ast

if len(sys.argv) == 2:
    data = ast.literal_eval(sys.argv[1])
    print(data)

输出

[('store', 'IDCODE', 18116485, 88779)]

但是，将arg作为单独的字符串接收会更加健壮。

import sys

if len(sys.argv) == 5:
    name, code, num1, num2 = sys.argv[1:]
    num1, num2 = int(num1), int(num2)
    data = [name, code, num1, num2]
    print(data)

就像

$ python scrpit.py "store" "IDCODE" 18116485 88779

输出是

['store', 'IDCODE', 18116485, 88779]