將接受命令行 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')