将接受命令行 arguments 的脚本转换为也可导入

Question

我有一个脚本 (foo.py)，我最初开发它是为了从名为 arguments 的命令行调用。 我现在想重构它，以便它既可以从命令行调用，也可以通过导入调用。 在导入后按原样调用脚本会引发意外的关键字参数错误，如所写（因为主 function 本身没有参数）。 我的印象是 __name__=="__main__" 块会在两次调用中提供相同的体验，但我显然误解了。 我该如何重构它，以便它可以交互地使用命名的 arguments 并通过另一个脚本导入？

foo.py

def parse_args():
  parser = argparse.ArgumentParser()
  parser.add_argument('--token', help='API token')
  parser.add_argument('--namedArgA', help='Switch parameter', action='store_true')
  return parser.parse_args()

def main():
  args = parse_args()
  token = args.token
  # Other stuff with the args
  print(argDict)

if __name__=="__main__":
  main()

酒吧.py

import foo
baz = foo.main(token='myToken',namedArgA='True')