这可能是一个愚蠢的问题,但我有一个python脚本,当前使用argparser接受了一堆争论,我想将这个脚本作为一个模块加载到另一个python脚本中,这很好。但我不知道如何调用模块,因为没有定义任何函数;如果我只是从cmd调用它,我还能像我一样调用吗?
这是子脚本:
import argparse as ap
from subprocess import Popen, PIPE
parser = ap.ArgumentParser(
description='Gathers parameters.')
parser.add_argument('-f', metavar='--file', type=ap.FileType('r'), action='store', dest='file',
required=True, help='Path to json parameter file')
parser.add_argument('-t', metavar='--type', type=str, action='store', dest='type',
required=True, help='Type of parameter file.')
parser.add_argument('-g', metavar='--group', type=str, action='store', dest='group',
required=False, help='Group to apply parameters to')
# Gather the provided arguements as an array.
args = parser.parse_args()
... Do stuff in the script
这是我要调用子脚本的父脚本;它还使用arg解析器并执行其他逻辑
from configuration import parameterscript as paramscript
# Can I do something like this?
paramscript('parameters/test.params.json', test)
在配置目录中,我还创建了一个空的 init .py文件。
答案 0 :(得分:9)
parse_args
的第一个参数是参数列表。默认情况下,None
表示使用sys.argv
。所以你可以这样安排你的脚本:
import argparse as ap
def main(raw_args=None):
parser = ap.ArgumentParser(
description='Gathers parameters.')
parser.add_argument('-f', metavar='--file', type=ap.FileType('r'), action='store', dest='file',
required=True, help='Path to json parameter file')
parser.add_argument('-t', metavar='--type', type=str, action='store', dest='type',
required=True, help='Type of parameter file.')
parser.add_argument('-g', metavar='--group', type=str, action='store', dest='group',
required=False, help='Group to apply parameters to')
# Gather the provided arguements as an array.
args = parser.parse_args(raw_args)
print(vars(args))
# Run with command line arguments precisely when called directly
# (rather than when imported)
if __name__ == '__main__':
main()
然后在其他地方:
from first_module import main
main(['-f', '/etc/hosts', '-t', 'json'])
输出:
{'group': None, 'file': <_io.TextIOWrapper name='/etc/hosts' mode='r' encoding='UTF-8'>, 'type': 'json'}
答案 1 :(得分:2)
可能有一种更简单,更pythonic的方法,但这是使用子进程模块的一种可能性:
示例:强>
<强> child_script.py 强>
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-n", "--name", help="your name")
args = parser.parse_args()
print("hello there {}").format(args.name)
然后另一个Python脚本可以这样调用该脚本:
<强> calling_script.py:强>
import subprocess
# using Popen may suit better here depending on how you want to deal
# with the output of the child_script.
subprocess.call(["python", "child_script.py", "-n", "Donny"])
执行上面的脚本会得到以下输出:
"hello there Donny"
答案 2 :(得分:1)
其中一个选项是将其称为子进程调用,如下所示:
import subprocess
childproc = subprocess.Popen('python childscript.py -file yourjsonfile')
op, oe = childproc.communicate()
print op