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。