1

我正在构建一个命令行工具,它在一个主 *.py 文件(我们称之为“main_file.py”)中执行一些 python 脚本(k2_fig1 - k2_fig3)。在这个“main_file.py”中,用户必须为数据库连接填写一些参数(用户名、dbname 等)。
现在我不知道如何将这些参数传递给我正在导入的每个 python 脚本。我必须为这些导入的文件编写什么代码?
这是我的“main_file.py”代码:

import argparse
def main():
  parser = argparse.ArgumentParser()

  parser.add_argument('-D', '--database', action="store", type=str, dest="my_dbname",  required=True, help="DB name")

  parser.add_argument('-U', '--username', action="store", type=str, dest="my_username", required=True, help="DB username")

  args = parser.parse_args()


  # Import different scripts
  import k2_fig1
  import k2_fig2
  import k2_fig3

if __name__ == '__main__':
  main()
4

2 回答 2

0

在不了解 k2fig_1 等其他任何信息的情况下,您需要使用subprocess而不是导入它们来调用它们。

import argparse
import subprocess

def main():
  parser = argparse.ArgumentParser()
  parser.add_argument('-D', '--database', action="store", type=str, dest="my_dbname",  required=True, help="DB name")
  parser.add_argument('-U', '--username', action="store", type=str, dest="my_username", required=True, help="DB username")

  args = parser.parse_args()
  for script in ['k2_fig1', 'k2_fig2', 'k2_fig3']:
      subprocess.call([script, '-U', args.my_username, '-D', args.my_dbname])

if __name__ == '__main__':
  main()
于 2013-01-19T15:25:28.673 回答
0

我认为最好的方法是将命名空间属性复制到“配置”模块::

import argparse

from . import config
from . import other

def update_obj(dst, src):
    for key, value in src.items():
        setattr(dst, key, value)

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('-D', '--database')
    parser.add_argument('-U', '--username')

    args = parser.parse_args('-D foo'.split())

    update_obj(config, args)

和“其他模块”::

from . import config

def some_func():
    assert config.database == 'foo'
于 2013-01-19T17:38:39.767 回答