python标准库介绍——10 sys 模块详解

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378

==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`` 模块来注册多个退出处理函数.