背景:1)任何一个Python程序文件既可以直接执行,也可以作为模块导入再使用其中的对象;2)对于大型系统开发,一般不会把所有代码放到单个文件中,而是根据功能将其分类并分散多个模块中,在编写小型项目时最好也能养成这样的好习惯。
本文介绍Python自定义模块中对象的导入和使用。
假设当前工作目录为C:\Python36,创建一个子目录child,然后在其中创建一个Python程序文件add.py,其中的代码为:
def add(x, y):
return x+y
这时,文件夹结构如图所示:
现在我们启动IDLE交互编程模式,默认工作目录是C:\Python36,执行下面的代码:
>>> import child
>>> child.add.add(3,5)
Traceback (most recent call last):
File "<pyshell#17>", line 1, in <module>
child.add.add(3,5)
AttributeError: module 'child' has no attribute 'add'
错误提示显示,child模块中没有可用的add,这是因为child文件夹被认为是一个包,而add.py是包中的子模块,并没有随着child一起导入。继续执行下面的代码:
>>> import child.add
>>> child.add.add(3,5)
8
自定义模块中的对象成功被导入并能够正常使用,也就是说,如果要使用的对象在子模块中,应该单独使用import来导入子模块。或者使用下面的方法:
>>> from child import add
>>> add.add(3,5)
8
接下来在IDLE中单击菜单“Restart Shell”恢复初始状态,然后执行下面的代码:
>>> from child import *
>>> add.add(3,5)
Traceback (most recent call last):
File "<pyshell#24>", line 1, in <module>
add.add(3,5)
NameError: name 'add' is not defined
错误提示显示并不存在add这样的名字,也就是说命令from child import *并没有导入add模块。现在在child子文件夹中创建一个Python程序文件__init__.py,其中内容为:
__all__ = ['add']
此时文件夹结构变为:
然后回到IDLE中执行刚才的代码:
>>> from child import *
>>> add.add(3,5)
8
结果正常。原因在于,如果文件夹作为包来使用,并且其中包含__init__.py文件时,__init__.py文件中的特殊列表成员__all__用来指定from ... import *时哪些子模块或对象会被自动导入。