python的主函数形式的简单介绍( 三 )


导入模块或解释器执行
在会话(或其他模块)中导入此文件时,Python解释器将执行相同的步骤 。
Python解释器导入文件后 , 您可以使用已导入模块中定义的任何变量 , 类或函数 。为了证明这一点,我们将使用可交互的Python解释器 。启动解释器,然后键入import best_practices:
导入best_practices.py后唯一的输出来自process_data()函数外定义的print() 。导入模块或解释器执行与基于命令行执行类似 。
使用__name__控制代码的执行
如何实现基于命令行而不使用Python解释器导入文件来执行呢?
您可以使用__name__来决定执行上下文,并且当__name__等于"__main__"时才执行process_data() 。在best_practices.py文件中添加如下代码:
这段代码添加了一个条件语句来检验__name__的值 。当值为"__main__"时 , 条件为True 。记住当__name__变量的特殊值为"__main__"时意味着Python解释器会执行脚本而不是将其导入 。
条件语块内添加了四行代码(第12,13 , 14和15行):
现在,在命令行中运行best_practices.py , 并观察输出的变化:
首先 , 输出显示了process_data()函数外的print()的调用结果 。
之后,data的值被打印 。因为当Python解释器将文件作为脚本执行时 , 变量__name__具有值"__main__" , 因此条件语句被计算为True 。
接下来 , 脚本将调用process_data()并传入data进行修改 。当process_data执行时,将输出一些状态信息 。最终 , 将输出modified_data的值 。
现在您可以验证从解释器(或其他模块)导入best_practices.py后发生的事情了 。如下示例演示了这种情况:
注意,当前结果与将条件语句添加到文件末尾之前相同 。因为此时__name__变量的值为"best_practices" , 因此条件语句结果为False,Python将不执行process_data() 。
创建名为main()的函数来包含要运行的代码
现在,您可以编写作为脚本由从命令行执行并导入且没有副作用的Python代码 。接下来,您将学习如何编写代码并使其他程序员能轻松地理解其含义 。
许多语言,如C,C++,Java以及其他的一些语言,都会定义一个叫做main()的函数 , 当编译程序时 , 操作系统会自动调用该函数 。此函数通常被称为入口点(entry point),因为它是程序进入执行的起始位置 。
相比之下,Python没有一个特殊的函数作为脚本的入口点 。实际上在Python中可以将入口点定义成任何名称 。
尽管Python不要求将函数命名为main(),但是最佳的做法是将入口点函数命名为main() 。这样方便其他程序员定位程序的起点 。
此外,main()函数应该包含Python解释器执行文件时要运行的任何代码 。这比将代码放入条件语块中更好,因为用户可以在导入模块时重复使用main()函数 。
修改best_practices.py文件如下所示:
在这个示例中,定义了一个main()函数,它包含了上面的条件语句块 。之后修改条件语块执行main() 。如果您将此代码作为脚本运行或导入,将获得与上一节相同的输出 。
在main()中调用其他函数
另一种常见的实现方式是在main()中调用其他函数,而不是直接将代码写入main() 。这样做的好处在于可以实现将几个独立运行的子任务整合 。
例如,某个脚本有如下功能:
如果在单独的函数中各自实现这些子任务 , 您(或其他用户)可以很容易地实现代码重用 。之后您可以在main()函数中创建默认的工作流 。
您可以根据自己的情况选择是否使用此方案 。将任务拆分为多个函数会使重用更容易,但会增加他人理解代码的难度 。

推荐阅读