9.1 自定义模块
自定义模块你已经会了,平常写的代码放到一个文件里面就是啦!
例如,写个简单的函数,作为一个模块:
#!/usr/bin/python # -*- coding: utf-8 -*- def func(a, b): return a * b class MyClass: def __init__(self, a, b): self.a = a self.b = b def method(self): return self.a * self.b 导入模块: >>> import test >>> test.func(2, 2) 4 >>> c = test.MyClass(2, 2) >>> c.method() 4
是不是很简单!是的,没错,就是这样。
需要注意的是,test就是文件名。另外,模块名要能找到,我的是在当前目录下。
有时经常from...import...,这又是啥呢,来看看:
>>> from test import func, MyClass # 多个函数或类以逗号分隔 >>> test.func(2, 2) Traceback (most recent call last): File "<stdin>", line 1, in <module> NameError: name 'test' is not defined >>> func(2, 2) 4 >>> c = MyClass(2, 2) >>> c.method() 4
看到了吧!如果你不想把模块里的函数都导入,就可以这样。一方面避免导入过多用不到的函数增加负载,另一方面引用时可不加模块名。
如果想调用不加模块名,也想导入所有模块,可以这样:
>>> from test import * >>> func(2, 2) 4 >>> c = MyClass(2, 2) >>> c.method() 4
使用个星号就代表了所有。
提醒:在模块之间引用也是同样的方式。
博客地址:http://lizhenliang.blog.51cto.com
QQ群:323779636(Shell/Python运维开发群)
9.2 作为脚本来运行程序
所有的模块都有一个内置属性__name__,如果import一个模块,那么模块的__name__属性返回值一般是文件名。如果直接运行Python程序,__name__的值将是一个"__mian__"。
举例说明,根据上面程序做一个测试:
#!/usr/bin/python # -*- coding: utf-8 -*- def func(a, b): return a * b class MyClass: def __init__(self, a, b): self.a = a self.b = b def method(self): return self.a * self.b print __name__ # python test.py __main__
与预期一样,打印出了“__main__”,再创建一个test2.py,导入这个模块:
#!/usr/bin/python # -*- coding: utf-8 -*- import test # python test2.py test
打印出了模块名,这个结果输出就是test.py中的print __name__。
所以,我们在test.py里面判断下__name__值等于__main__时说明在手动执行这个程序:
#!/usr/bin/python # -*- coding: utf-8 -*- def func(a, b): return a * b class MyClass: def __init__(self, a, b): self.a = a self.b = b def method(self): return self.a * self.b if __name__ == "__main__": print "我在手动执行这个程序..." # python test.py 我在手动执行这个程序...
此时再运行test2.py试试,是不是打印为空!明白了吧!
9.3 安装第三方模块
在Python中安装外部的模块有几种方式:
1)下载压缩包,通过setuptools工具安装,这个在第一章Python基础知识里面用到过。推荐下载地址:http://pypi.python.org
2)easy_install工具安装,也依赖setuptools。
3)pip工具安装。推荐使用这个方式。
4)直接将压缩包解压到Python模块目录。但常常会出现import失败,不推荐。
5)在Windows下,除了上面几种方式以外,可以直接下载exe文件点击一步步安装。
pip与easy_install安装方式类似,主要区别在于easy_install不支持卸载软件,而pip支持。
推荐使用pip命令安装,简单方便。如果安装失败可以按顺序这么尝试:方式1 --> 方式2 --> 方式4
以安装setuptools举例上面几种安装方式:
方式1: # wget https://pypi.python.org/packages/32/3c/e853a68b703f347f5ed86585c2dd2828a83252e1216c1201fa6f81270578/setuptools-26.1.1.tar.gz # tar zxvf setuptools-26.1.1.tar.gz # cd setuptools-26.1.1 # python setup.py install 方式2: # easy_install setuptools 方式3: # pip install setuptools # pip uninstall setuptools # 卸载 # pip search setuptools # 搜索 方式3: cp -rf setuptools-26.1.1 /usr/local/lib/python2.7/dist-packages
9.4 查看模块帮助文档
前面几个章节已经使用几个内置模块了,比如collections、itertools等,导入与上面一样,这里不再过多说明了。
1)help()函数
当一个模块对其语法不了解时,可以查看帮助,以collections举例:
>>> import collections >>> help(collections) Help on module collections: NAME collections FILE /usr/lib/python2.7/collections.py MODULE DOCS http://docs.python.org/library/collections # 注意:这里是这个模块的帮助文档,很详细的哦! CLASSES __builtin__.dict(__builtin__.object) Counter OrderedDict defaultdict __builtin__.object _abcoll.Callable _abcoll.Container ......
使用help()就能查看这个模块的内部构造,包括类方法、属性等信息。
也可以再对某个方法查看其用法:
>>> help(collections.Counter()) Help on Counter in module collections object: class Counter(__builtin__.dict) | Dict subclass for counting hashable items. Sometimes called a bag | or multiset. Elements are stored as dictionary keys and their counts | are stored as dictionary values. | | >>> c = Counter('abcdeabcdabcaba') # count elements from a string | | >>> c.most_common(3) # three most common elements | [('a', 5), ('b', 4), ('c', 3)] | >>> sorted(c) # list all unique elements | ['a', 'b', 'c', 'd', 'e'] | >>> ''.join(sorted(c.elements())) # list elements with repetitions | 'aaaaabbbbcccdde' | >>> sum(c.values()) # total of all counts | 15 | | >>> c['a'] # count of letter 'a' ......
一般里面都是举例说明,可快速帮助我们回忆使用方法。
2)dir()函数查看对象属性
这个在前面也用到过,能看到对象的方法、属性等信息:
>>> dir(collections) ['Callable', 'Container', 'Counter', 'Hashable', 'ItemsView', 'Iterable', 'Iterator', 'KeysView', 'Mapping', 'MappingView', 'MutableMapping', 'MutableSequence', 'MutableSet', 'OrderedDict', 'Sequence', 'Set', 'Sized', 'ValuesView', '__all__', '__builtins__', '__doc__', '__file__', '__name__', '__package__', '_abcoll', '_chain', '_class_template', '_eq', '_field_template', '_get_ident', '_heapq', '_imap', '_iskeyword', '_itemgetter', '_repeat', '_repr_template', '_starmap', '_sys', 'defaultdict', 'deque', 'namedtuple']• 1. • 2.
3)github上查看模块用法
Python官方模块下载地址http://pypi.python.org,所有的模块在这里都有。
打开网站后,在搜索框搜索你的模块名,在结果找到模块名点进去,会有一个 Home Page的连接,Python大多数模块都是托管在github上面,这个链接就是这个模块在github上面的地址,点击后跳转到github对应的模块页面,里面也有很详细模块使用方法。
9.5 导入模块新手容易出现的问题
还有一个新手经常犯的问题,写一个模块,比如使用itertools模块,为了说明这个测试文件是这个模块,就把文件名写成了这个模块名,于是就造成了下面错误:
#!/usr/bin/python # -*- coding: utf-8 -*- import collections c = collections.Counter() for i in "Hello world!": c[i] += 1 print c # python collections.py Traceback (most recent call last): File "collections.py", line 3, in <module> import collections File "/home/user/collections.py", line 4, in <module> c = collections.Counter() AttributeError: 'module' object has no attribute 'Counter'
抛出异常,明明在解释器里面可以正常导入使用啊,怎么会提示没Counter属性呢,问题就出现你的文件名与导入的模块名重名,导致程序import了这个文件,上面讲过文件名就是模块名。所以文件名不要与引用的模块名相同。
还有一个使用方法也说明下,使用as关键字设置模块别名,这样使用中就不用输入那么长的模块名了,按照上面的例子,把名字先改成collections1.py,做测试:
#!/usr/bin/python # -*- coding: utf-8 -*- import collections as cc c = cc.Counter() for i in "Hello world!": c[i] += 1 print c # python collections1.py Counter({'l': 3, 'o': 2, '!': 1, ' ': 1, 'e': 1, 'd': 1, 'H': 1, 'r': 1, 'w': 1})