我正在构建一个命令行工具,它在一个主 *.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()