• 欢迎访问蜷缩的蜗牛博客 蜷缩的蜗牛
  • 微信搜索: 蜷缩的蜗牛 | 联系站长 kbsonlong@qq.com
  • 如果您觉得本站非常有看点,那么赶紧使用Ctrl+D 收藏吧

第九章 Python自定义模块及导入方法

Python 蜷缩的蜗牛 8个月前 (02-08) 60次浏览 已收录

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']

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

本文转载自 第九章 Python 自定义模块及导入方法


蜷缩的蜗牛 , 版权所有丨如未注明 , 均为原创丨 转载请注明第九章 Python 自定义模块及导入方法
喜欢 (0)
[]
分享 (0)