==sys 模块== ``sys`` 模块提供了许多函数和变量来处理 Python 运行时环境的不同部分. === 处理命令行参数=== 在解释器启动后, ``argv`` 列表包含了传递给脚本的所有参数, 如 [Example 1-66 #eg-1-66] 所示.
列表的第一个元素为脚本自身的名称. ====Example 1-66. 使用sys模块获得脚本的参数====[eg-1-66] ```
File: sys-argv-example-1.py import sys print "script name is", sys.argv[0] if len(sys.argv) > 1:
print "there are", len(sys.argv)-1, "arguments:"
for arg in sys.argv[1:]:
print arg
else:
print "there are no arguments!" *B*script name is sys-argv-example-1.py
there are no arguments!*b*
``` 如果是从标准输入读入脚本 (比如 "``python < sys-argv-example-1.py``"),
脚本的名称将被设置为空串. 如果把脚本作为字符串传递给python (使用 ``-c`` 选项),
脚本名会被设置为 "-c". === 处理模块 === ``path`` 列表是一个由目录名构成的列表, Python 从中查找扩展模块( Python 源模块, 编译模块,或者二进制扩展).
启动 Python 时,这个列表从根据内建规则, PYTHONPATH 环境变量的内容, 以及注册表( Windows 系统)等进行初始化.
由于它只是一个普通的列表, 你可以在程序中对它进行操作, 如 [Example 1-67 #eg-1-67] 所示. ====Example 1-67. 使用sys模块操作模块搜索路径====[eg-1-67] ```
File: sys-path-example-1.py import sys print "path has", len(sys.path), "members" # add the sample directory to the path
sys.path.insert(0, "samples")
import sample # nuke the path
sys.path = []
import random # oops! *B*path has 7 members
this is the sample module!
Traceback (innermost last):
File "sys-path-example-1.py", line 11, in ?
import random # oops!
ImportError: No module named random*b*
``` ``builtin_module_names`` 列表包含 Python 解释器中所有内建模块的名称, [Example 1-68 #eg-1-68]
给出了它的样例代码. ====Example 1-68. 使用sys模块查找内建模块====[eg-1-68] ```
File: sys-builtin-module-names-example-1.py import sys def dump(module):
print module, "=>",
if module in sys.builtin_module_names:
print "<BUILTIN>"
else:
module = _ _import_ _(module)
print module._ _file_ _ dump("os")
dump("sys")
dump("string")
dump("strop")
dump("zlib") *B*os => C:\python\lib\os.pyc
sys => <BUILTIN>
string => C:\python\lib\string.pyc
strop => <BUILTIN>
zlib => C:\python\zlib.pyd*b*
``` ``modules`` 字典包含所有加载的模块. ``import`` 语句在从磁盘导入内容之前会先检查这个字典. 正如你在 [Example 1-69 #eg-1-69] 中所见到的, Python 在处理你的脚本之前就已经导入了很多模块. ====Example 1-69. 使用sys模块查找已导入的模块====[eg-1-69] ```
File: sys-modules-example-1.py import sys print sys.modules.keys() *B*['os.path', 'os', 'exceptions', '_ _main_ _', 'ntpath', 'strop', 'nt',
'sys', '_ _builtin_ _', 'site', 'signal', 'UserDict', 'string', 'stat']*b*
``` === 处理引用记数=== ``getrefcount`` 函数 (如 [Example 1-70 #eg-1-70] 所示) 返回给定对象的引用记数 -
也就是这个对象使用次数. Python 会跟踪这个值, 当它减少为0的时候, 就销毁这个对象. ====Example 1-70. 使用sys模块获得引用记数====[eg-1-70] ```
File: sys-getrefcount-example-1.py import sys variable = 1234 print sys.getrefcount(0)
print sys.getrefcount(variable)
print sys.getrefcount(None) *B*50
3
192*b*
``` 注意这个值总是比实际的数量大, 因为该函数本身在确定这个值的时候依赖这个对象. == 检查主机平台=== [Example 1-71 #eg-1-71] 展示了 ``platform`` 变量, 它包含主机平台的名称. ====Example 1-71. 使用sys模块获得当前平台====[eg-1-71] ```
File: sys-platform-example-1.py import sys #
# emulate "import os.path" (sort of)... if sys.platform == "win32":
import ntpath
pathmodule = ntpath
elif sys.platform == "mac":
import macpath
pathmodule = macpath
else:
# assume it's a posix platform
import posixpath
pathmodule = posixpath print pathmodule
``` 典型的平台有Windows 9X/NT(显示为 ``win32`` ), 以及 Macintosh(显示为 ``mac`` ) .
对于 Unix 系统而言, platform 通常来自 "``uname -r``" 命令的输出, 例如 ``irix6`` , ``linux2`` ,
或者 ``sunos5`` (Solaris). === 跟踪程序=== ``setprofiler`` 函数允许你配置一个分析函数(profiling function).
这个函数会在每次调用某个函数或方法时被调用(明确或隐含的),
或是遇到异常的时候被调用. 让我们看看 [Example 1-72 #eg-1-72] 的代码. ====Example 1-72. 使用sys模块配置分析函数====[eg-1-72] ```
File: sys-setprofiler-example-1.py import sys def test(n):
j = 0
for i in range(n):
j = j + i
return n def profiler(frame, event, arg):
print event, frame.f_code.co_name, frame.f_lineno, "->", arg # profiler is activated on the next call, return, or exception
# 分析函数将在下次函数调用, 返回, 或异常时激活
sys.setprofile(profiler) # profile this function call
# 分析这次函数调用
test(1) # disable profiler
# 禁用分析函数
sys.setprofile(None) # don't profile this call
# 不会分析这次函数调用
test(2) *B*call test 3 -> None
return test 7 -> 1*b*
``` 基于该函数, ``profile`` 模块提供了一个完整的分析器框架. [Example 1-73 #eg-1-73] 中的 ``settrace`` 函数与此类似,
但是 ``trace`` 函数会在解释器每执行到新的一行时被调用. ====Example 1-73. 使用sys模块配置单步跟踪函数====[eg-1-73] ```
File: sys-settrace-example-1.py import sys def test(n):
j = 0
for i in range(n):
j = j + i
return n def tracer(frame, event, arg):
print event, frame.f_code.co_name, frame.f_lineno, "->", arg
return tracer # tracer is activated on the next call, return, or exception
# 跟踪器将在下次函数调用, 返回, 或异常时激活
sys.settrace(tracer) # trace this function call
# 跟踪这次函数调用
test(1) # disable tracing
# 禁用跟踪器
sys.settrace(None) # don't trace this call
# 不会跟踪这次函数调用
test(2) *B*call test 3 -> None
line test 3 -> None
line test 4 -> None
line test 5 -> None
line test 5 -> None
line test 6 -> None
line test 5 -> None
line test 7 -> None
return test 7 -> 1*b*
``` 基于该函数提供的跟踪功能, ``pdb`` 模块提供了完整的调试( debug )框架. === 处理标准输出/输入 === ``stdin`` , ``stdout`` , 以及 ``stderr`` 变量包含与标准 I/O 流对应的流对象.
如果需要更好地控制输出,而 ``print`` 不能满足你的要求, 它们就是你所需要的.
你也可以 //替换// 它们, 这时候你就可以重定向输出和输入到其它设备( device ),
或者以非标准的方式处理它们. 如 [Example 1-74 #eg-1-74] 所示. ====Example 1-74. 使用sys重定向输出====[eg-1-74] ```
File: sys-stdout-example-1.py import sys
import string class Redirect: def _ _init_ _(self, stdout):
self.stdout = stdout def write(self, s):
self.stdout.write(string.lower(s)) # redirect standard output (including the print statement)
# 重定向标准输出(包括print语句)
old_stdout = sys.stdout
sys.stdout = Redirect(sys.stdout) print "HEJA SVERIGE",
print "FRISKT HUM\303\226R" # restore standard output
# 恢复标准输出
sys.stdout = old_stdout print "M\303\205\303\205\303\205\303\205L!" *B*heja sverige friskt hum\303\266r
M\303\205\303\205\303\205\303\205L!*b*
``` 要重定向输出只要创建一个对象, 并实现它的 ``write`` 方法. (除非 C 类型的实例外：Python 使用一个叫做 ``softspace`` 的整数属性来控制输出中的空白.
如果没有这个属性, Python 将把这个属性附加到这个对象上. 你不需要在使用 Python 对象时担心,
但是在重定向到一个 C 类型时, 你应该确保该类型支持 ``softspace`` 属性.) === 退出程序=== 执行至主程序的末尾时,解释器会自动退出. 但是如果需要中途退出程序, 你可以调用 ``sys.exit`` 函数,
它带有一个可选的整数参数返回给调用它的程序. [Example 1-75 #eg-1-75] 给出了范例. ====Example 1-75. 使用sys模块退出程序====[eg-1-75] ```
File: sys-exit-example-1.py import sys print "hello" sys.exit(1) print "there" *B*hello*b*
``` 注意 ``sys.exit`` 并不是立即退出. 而是引发一个 //SystemExit// 异常.
这意味着你可以在主程序中捕获对 ``sys.exit`` 的调用, 如 [Example 1-76 #eg-1-76] 所示. ====Example 1-76. 捕获sys.exit调用====[eg-1-76] ```
File: sys-exit-example-2.py import sys print "hello" try:
sys.exit(1)
except SystemExit:
pass print "there" *B*hello
there*b*
``` 如果准备在退出前自己清理一些东西(比如删除临时文件), 你可以配置一个
"退出处理函数"(exit handler), 它将在程序退出的时候自动被调用. 如 [Example 1-77 #eg-1-77] 所示. ====Example 1-77. 另一种捕获sys.exit调用的方法====[eg-1-77] ```
File: sys-exitfunc-example-1.py import sys def exitfunc():
print "world" sys.exitfunc = exitfunc print "hello"
sys.exit(1)
print "there" # never printed # 不会被 print *B*hello
world*b*
``` 在 Python 2.0 以后, 你可以使用 ``atexit`` 模块来注册多个退出处理函数.