Python黑帽编程2.6 模块

我们已经学习了如何在你的程序中定义一次函数而重用代码。如果你想要在其他程序中重用很多函数，那么你该如何编写程序呢？你可能已经猜到了，答案是使用模块。模块基本上就是一个包含了所有你定义的函数和变量的文件。为了在其他程序中重用模块，模块的文件名必须以.py为扩展名。

模块可以从其他程序导入以便利用它的功能。这也是我们使用Python标准库的方法。首先，我们将学习如何使用标准库模块。

2.6.1 调用sys模块

先上代码：

#!/usr/bin/python

import sys

print 'The command line arguments are:'

for i in sys.argv:

print i

print '\n\nThe PYTHONPATH is', sys.path, '\n'

首先，我们利用import语句导入sys模块。sys模块包含了与Python解释器和它的环境有关的函数。

当Python执行import sys语句的时候，它在sys.path变量中所列目录中寻找sys.py模块。如果找到了这个文件，这个模块的主块中的语句将被运行，然后这个模块将能够被你使用。注意，初始化过程仅在我们第一次输入模块的时候进行。

sys.argv变量是一个字符串的列表，包含了命令行参数的列表。

sys.path包含输入模块的目录名列表。我们可以观察到sys.path的第一个字符串是空的——这个空的字符串表示当前目录也是sys.path的一部分，这与PYTHONPATH环境变量是相同的。这意味着你可以直接输入位于当前目录的模块。否则，你得把你的模块放在sys.path所列的目录之一。

下面我们保存这段代码为module_sys.py，从终端使用python来运行，并输入参数“test argument”，如下图所示：

图2

2.6.2 from..import语句

上面的示例中我们通过sys.argv的方式来获取argv变量中的内容，那如果我们想直接调用argv，就可以获取变量内容或进行方法调用，该如何做呢？

可以使用

from sys import argv

语句。

还有一种更简单的方法

from sys import *

导入sys模块中所有可用的变量或方法。

一般说来，应该避免使用from..import而使用import语句，因为这样可以使你的程序更加易读，也可以避免名称的冲突。

2.6.3 name

每个模块都有一个名称，在模块中可以通过语句来找出模块的名称。前面说过，当一个模块被第一次导入的时候，这个模块的主块将被运行。假如我们只想在程序本身被使用的时候运行主块，而在它被别的模块输入的时候不运行主块，我们该怎么做呢？这可以通过模块的__name__属性完成。

#!/usr/bin/python

if __name__ == '__main__':

print 'This program is being run by itself'

else:

print 'I am being imported from another module'

每个Python模块都有它的__name__，如果它是'__main__'，这说明这个模块被用户单独运行，我们可以进行相应的恰当操作。

我们直接运行该脚本，运行结果如下：

图3

2.6.4 创建模块

创建你自己的模块是十分简单的，你一直在这样做！每个以.py结尾的文件都是一个模块。

下面我们先创建一个myModule.py的文件，内容如下：

#!/usr/bin/python

def sayhi():

print 'Hi, this is mymodule speaking.'

version = '0.1'

从上面的代码可以看到，它与我们普通的Python程序相比并没有什么特别之处。我们接下来将看看如何在我们别的Python程序中使用这个模块。

这个模块应该被放置在我们导入它的程序的同一个目录中，或者在sys.path所列目录之一。

测试代码如下：

#!/usr/bin/python

import mymodule

mymodule.sayhi()

print 'Version', mymodule.version

运行结果如下：

图4

2.6.5 dir()函数

我们可以使用内建的dir函数来列出模块定义的标识符。标识符有函数、类和变量。当为dir()提供一个模块名的时候，它返回模块定义的名称列表。如果不提供参数，它返回当前模块中定义的名称列表。

先看下面的代码：

#!/usr/bin/python

import sys

print dir(sys)

a=5

print dir()

del a

print dir()

首先，我们来看一下在输入的sys模块上使用dir。我们看到它包含一个庞大的属性列表。如下图：

图5

接下来，我们不给dir函数传递参数，默认地，它返回当前模块的属性列表。

为了观察dir的作用，我们定义一个新的变量a并且给它赋一个值，然后检验dir，我们观察到在列表中增加了以上相同的值。我们使用del语句删除当前模块中的变量/属性，这个变化再一次反映在dir的输出中。结果如下：

图6

2.5.9 小结

模块的用处在于它能为你在别的程序中重用它提供的服务和功能。Python附带的标准库就是这样一组模块的例子。我们已经学习了如何使用这些模块以及如何创造我们自己的模块。

下一节我们学校异常处理。

第2.7节《异常处理》已经在微信订阅号抢先发布，心急的同学进入订阅号（二维码在下方），从菜单“网络安全”—>”Python黑帽编程”进入即可。

由于教程仍在创作过程中，在整套教程完结前，感兴趣的同学请关注我的微信订阅号（xuanhun521，下方二维码），我会第一时间在订阅号推送图文教程和视频教程。问题讨论请加qq群：Hacking （1群）：303242737 Hacking （2群）：147098303。