从命令行调用python模块函数

Question

我想知道是否有办法从命令行调用模块函数。我的问题是我有一个名为find_mutations的文件包。这要求输入文件以特定方式排序，所以我做了一个功能。我希望用户能够选择正常运行代码：

$ python -m find_mutations -h

或运行特定的sort_files函数：

$ python -m find_mutations.sort_files -h

这样的事情可能吗？如果是这样，sort_files函数是否需要在自己的条带中？我是否需要在__init__.py脚本中添加内容？包正确安装并运行正常，我只想添加它。

Answer 1

您可以添加标志来运行该功能。以下示例不是使用命令参数的好方法，但演示了这个想法。

import sys

if __name__ == "__main__":
    if '-s' in sys.argv:
        sort_files()

添加一个标志来运行特定功能，在本例中为-s以对文件进行排序。