我有两个python脚本。主脚本调用生成.py它有一个从argsparse获取3个参数的主函数。我想做的是,进口生成.py进入主.py. 从中调用主函数生成.py但是将三个参数传递给this,这样它就可以用这些参数执行main函数。在
这三个参数是path、files和data。在
路径只是指向数据的文件路径
文件本质上是循环遍历目录中的文件的for循环
data是一个包含json文件的文件夹
到目前为止我的主.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()
在生成.py具体如下:
^{pr2}$