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

Question

I've got a script (foo.py) which I originally developed to be called from the command line with named arguments. I'd now like to refactor it so it may called both from the command line and via import as well. Calling the script as is after import throws an unexpected keyword argument error, as written (since the main function itself has no arguments). I was under the impression that the __name__=="__main__" chunk would've provided the same experience in both calls, but I've obviously misunderstood. How could I refactor this so it can take the named arguments interactively and upon import by another script?

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()

bar.py

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