Question

当我导入我的python脚本之一并运行当前脚本时，它似乎正在运行并显示导入脚本的输出，这确实是不寻常的行为。我刚刚将其导入脚本中，但未在主代码中真正调用其任何功能。我如何避免这种行为的发生？

如果我在主脚本中传递-d标志，它将仅在主脚本中运行常规代码

如果我在主脚本中传递-t标志，它将仅从导入的python脚本运行代码

main.py

import os
import argparse
import functions as funcs
import generate_json as gen_json
from test_compare_filesets import tester as imptd_tester


def get_json_location():
    path = os.getcwd() + '/Testdata'
    return path


def main():

 parser = argparse.ArgumentParser()
 parser.add_argument("-d", "--export-date", action="store_true", required=True)
 parser.add_argument("-t", "--execute-test", action="store_true", required=False)
 args = parser.parse_args()
 date = args.export_date
 testt = args.execute_test

 yml_directory = os.listdir('yaml/')
 yml_directory.remove('export_config.yaml')



 with open('dates/' + date + '.json', 'w') as start:
  start.close()

 for yml in yml_directory :
   print("Running export for " + yml)
   yml_file = os.path.join('yaml/' + yml)

   json_path = get_json_location()
   yml = funcs.read_config(yml_file)
   data_folder = date
   gen_json.generate_data_report(json_path , yml , data_folder)


if __name__ == '__main__':


   main()

test_files.py

import generate_report as generate_reportt


def compare_filesets(file_names, previous_data, current_data):
    for item in file_names:
        print(item + generate_reportt.compare(previous_data.get(item), current_data.get(item)) + "\n")


def test_filesets():

 '''
Test for scenario 1
'''

dict_1 = generate_reportt.read_file_into_dict("dates/2018-01-01.json")
dict_2 = generate_reportt.read_file_into_dict("dates/2018-01-02.json")
print(" Test 1 ")
compare_filesets(file_names=['a.json', 'b.json', 'c.json'],
                 previous_data=dict_1,
                 current_data=dict_2
                 )

Answer 1

这就是使用以下语句的原因：

if __name__ == "__main__":
    main()

非常重要。您将需要将此代码添加到要导入的脚本中，并将所有被调用的代码放在该脚本的main()函数中。脚本的变量__name__根据是否导入脚本而变化。如果您不导入脚本并运行它，则该脚本的__name__变量将为"__main__"。但是，如果将其导入，__name__变量将变成脚本的文件名，因此该脚本的main()函数中的所有内容都将不会运行。

有关更多信息：What does if __name__ == "__main__": do?

如何停止导入的python脚本运行，但仅在代码中调用时运行

1 个答案: