python - 为执行脚本添加一个参数

时间:2016-12-30 06:07:58

标签: python

考虑我的bin中有以下代码,如下所示(filename:emp_dsb):

import sys

from employee_detail_collector.EmpCollector import main

if __name__ == '__main__':
    sys.exit(main())

在我的命令行中,我将执行“emp_dsb”,以便上面的代码将执行“employee_detail_collector.EmpCollector”中的main函数

代码(employee_detail_collector.EmpCollector)main():

def main():
    try:
        path = const.CONFIG_FILE
        empdsb = EmpDashboard(path)
    except SONKPIExceptions as e:
        logger.error(e.message)
    except Exception as e:
        logger.error(e)

现在我需要为emp_dsb添加一些参数,就像“emp_dsb create_emp ”应该调用一组新的功能来创建一个员工,这也需要添加到同一个主服务器中()

有人看,让我知道你的想法,如果不清楚,请告诉我,以便我会努力使其更清晰。

2 个答案:

答案 0 :(得分:1)

使用命令行参数的标准方法是:

import sys

if __name__ == '__main__':

    print(sys.argv)

阅读sys.argv的文档。

然后有更好的方式,如内置argparse和第三方docoptclick

答案 1 :(得分:1)

我个人会使用'argparse'模块。 Here is the link to a dead simple code sample

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("echo")
args = parser.parse_args()
print(args.echo)