[英]How to import python script and pass arguments to the main function
我有兩個python腳本。 主腳本調用generate.py,它具有一個主要函數,該函數從argsparse獲取3個參數。 我想做的是,將generate.py導入到main.py中。 從generate.py調用主函數,但將三個參數傳遞給它,以便它可以使用這些參數執行主函數。
這三個參數是path,files和data。
路徑只是數據文件的文件路徑,本質上是for循環遍歷目錄文件的數據data是帶有json文件的文件夾
到目前為止,我的main.py腳本如下所示。 :
def get_json_location(username=os.getlogin()):
first = "/Users/"
last = "/Desktop/data-code/Testdata"
result = first + username + last
return result
def get_config():
parser = argparse.ArgumentParser()
parser.add_argument("-p", "--parameter-file", action="store", required=False)
parser.add_argument("-e", "--export-data-file", action="store", required=False)
parser.add_argument("-d", "--export-date", action="store", required=False)
args = parser.parse_args()
return [args.parameter_file, args.export_data_file, args.export_date]
path, files, data = get_config()
date = get_config()[2]
yml_directory = os.listdir('yaml')
yml_directory.remove('export_config.yaml')
def generate_json():
with open('dates/' + date + '.json', 'w') as start:
start.close()
for yml in yml_directory :
print("Running export for " + yml)
# Here is the script that i am calling but i want to pass the parameters path , file and data_folder
gen_json.main(path,files,data)
if __name__ == '__main__':
generate_json()
generate.py如下:
def main(get_config):
cfg = get_config()[0] # json location
data = get_config()[1] # export_agent_core_agent.yaml
date = get_config()[2] # synthetic data folder - YYYY-MM-DD
json_location = cfg
json_database = data["config"]["database"]
json_collection = data["config"]["collection"]
json_path = "{0}/{1}{2}/{3}/{3}.json".format(json_location, date, json_database, json_collection)
def get_config():
parser = argparse.ArgumentParser()
parser.add_argument("-p", "--parameter-file", action="store", required=True)
parser.add_argument("-e", "--export-data-file", action="store", required=True)
parser.add_argument("-d", "--export-date", action="store", required=True)
args = parser.parse_args()
return [args.parameter_file, funcs.read_config(args.export_data_file), args.export_date]
if __name__ == "__main__":
logging.getLogger().setLevel(logging.INFO)
main(get_config)
考慮重構代碼!
要回答您的問題,您可以將字符串從main.py
給gen_json.main('...')
, gen_json.get_config('...')
。 Argparse將評估該字符串,該字符串需要拆分為一個列表。
您仍然可以使用命令行參數調用generate.py。
請注意,我刪除了不必要的/不正確的代碼/標識以顯示其原理。
generate.py:
import argparse
import sys
def main(cliParams=None):
cfg, data, date = get_config(cliParams)
print (cfg, data, date)
#json_location = cfg
#json_database = data["config"]["database"]
#json_collection = data["config"]["collection"]
#json_path = "{0}/{1}{2}/{3}/{3}.json".format(json_location, date, json_database, json_collection)
#print json_path
def get_config(cliParams=None):
parser = argparse.ArgumentParser()
parser.add_argument("-p", "--parameter-file", action="store", required=True)
parser.add_argument("-e", "--export-data-file", action="store", required=True)
parser.add_argument("-d", "--export-date", action="store", required=True)
if not cliParams:
args = parser.parse_args()
else:
args = parser.parse_args(cliParams.split())
#return [args.parameter_file, funcs.read_config(args.export_data_file), args.export_date]
return [args.parameter_file, args.export_data_file, args.export_date]
if __name__ == "__main__":
logging.getLogger().setLevel(logging.INFO)
main()
main.py:
import os
import generate as gen_json
def generate_json():
path = '/tmp/foo.yml'
data = 'export_agent_core_agent.yaml'
date = '2019-01-15'
argParseString = '-p %s -e %s -d %s' % (path, data, date)
gen_json.main(argParseString)
if __name__ == '__main__':
generate_json()
打印到標准輸出:
>>> /tmp/foo.yml export_agent_core_agent.yaml 2019-01-15
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.