Python 学习笔记 (8)—— sys模块

主要介绍用的比较多的sys的模块命令包括:sys.argv,sys.platform,sys.getdefaultencoding,sys.setdefaultencoding(),sys.getfilesystemencoding(),sys.exit(n),sys.path,sys.modules.keys(),sys.stdin,sys.stdout,sys.stderr 等。

sys.argv    获取参数
参数获取从0开始,而不是1,0为命令本身
#!/usr/bin/python
import sys
print "I the first arg :", sys.argv[1]
print "I the second arg :", sys.argv[2]
print "All of the args are :"
for i in sys.argv:
        print i

执行结果:
[root@node1 python]# python arg.py hello python
I the first arg : hello
I the second arg : python
All of the args are :
arg.py
hello
python

sys.platform   获取当前操作系统平台

具体的返回值如下:
>>> import sys
>>> sys.platform
‘linux2‘

操作系统 返回值
Linux (2.x and 3.x)
‘linux2‘
Windows
 ‘win32‘ 
Windows/Cygwin
 ‘cygwin‘
Mac OS X
‘darwin‘ 
OS/2
 ‘os2‘
OS/2 EMX
‘os2emx‘ 
RiscOS
‘riscos‘
AtheOS
‘atheos‘

 

实例:
根据不同操作系统判断该使用什么命令,例如在在linux 下用“clear”而在windows下用“cls”

#!/usr/bin/python
import sys
ostype = sys.platform
if ostype == "linux" or ostype == "linux2":
        cmd = "clear"
else:
        cmd = "cls"
print "The clear command is :", cmd

[root@node1 python]# python os.py 
The clear command is : clear

sys.exit(n)        设置退出返回值(0表示正常退出,其他非0整数表示不正常,可以通过异常捕获)

 

#!/usr/bin/python
import sys
def exitFunc(value):
        ‘‘‘Clear function‘‘‘
        print value
        sys.exit()

print "hello"

try:
        sys.exit(1)
except SystemExit,value:
        exitFunc(value)
print "Ok"

[root@node1 python]# python esc.py 
hello
1

sys.path      返回模块的搜索路径,初始化时使用PYTHONPATH环境变量的值
在通过命令"import module_name"时,系统将在以下路径中查找导入的模块,其中第一个为空,表示当前目录
>>> import sys
>>> sys.path
[‘‘, ‘/usr/lib64/python26.zip‘, ‘/usr/lib64/python2.6‘, ‘/usr/lib64/python2.6/plat-linux2‘, ‘/usr/lib64/python2.6/lib-tk‘, 
‘/usr/lib64/python2.6/lib-old‘, ‘/usr/lib64/python2.6/lib-dynload‘, ‘/usr/lib64/python2.6/site-packages‘, 
‘/usr/lib64/python2.6/site-packages/gtk-2.0‘, ‘/usr/lib/python2.6/site-packages‘]

 

sys.modules.keys()    使用sys模块查找已导入的模块
>>> import sys
>>> print sys.modules.keys()
[‘copy_reg‘, ‘encodings‘, ‘site‘, ‘__builtin__‘, ‘__main__‘, ‘encodings.encodings‘, ‘abc‘, ‘posixpath‘, 
‘errno‘, ‘encodings.codecs‘, ‘_abcoll‘, ‘types‘, ‘_codecs‘, ‘_warnings‘, ‘genericpath‘, ‘stat‘, ‘zipimport‘,
 ‘encodings.__builtin__‘, ‘warnings‘, ‘UserDict‘, ‘encodings.utf_8‘, ‘sys‘, ‘codecs‘, ‘readline‘, ‘os.path‘,
 ‘signal‘, ‘linecache‘, ‘posix‘, ‘encodings.aliases‘, ‘exceptions‘, ‘os‘]

sys.getdefaultencoding()    获取系统当前编码,一般默认为ascii。


>>> import sys
>>> sys.getdefaultencoding()
‘ascii‘

sys.setdefaultencoding()    设置系统默认编码


执行dir(sys)时不会看到这个方法,在解释器中执行不通过,可以先执行reload(sys),再执行 setdefaultencoding(‘utf8‘),此时将系统默认编码设置为utf8。
>>> reload(sys)
<module ‘sys‘ (built-in)>
>>> sys.setdefaultencoding(‘utf8‘)

 

sys.getfilesystemencoding()    获取文件系统使用编码方式
>>> sys.getfilesystemencoding()
‘UTF-8‘

 

sys.stdin,sys.stdout,sys.stderr
标准输入和标准错误 (通常缩写为 stdout 和 stderr) 是内建在每一个 UNIX 系统中的管道。 
当你 print 某些东西时,结果前往 stdout 管道;
当你的程序崩溃并打印出调试信息 (例如 Python 中的 traceback (错误跟踪)) 的时候,信息前往 stderr 管道 
stdout 是一个类文件对象;调用它的 write 函数可以打印出你给定的任何字符串。
实际上,这就是 print 函数真正做的事情;它在你打印的字符串后面加上一个硬回车,然后调用 sys.stdout.write 函数。

>>> for i in range(3):
...     print‘Dive in‘

Dive in
Dive in
Dive in
>>> import sys
>>> for i in range(3):
...     sys.stdout.write(‘Dive in‘)

Dive inDive inDive in
>>> for i in range(3):
...     sys.stderr.write(‘Dive in‘)

Dive inDive inDive in


在最简单的例子中,stdout 和 stderr 把它们的输出发送到相同的地方
和 stdout 一样,stderr 并不为你添加硬回车;如果需要,要自己加上。
stdout 和 stderr 都是类文件对象,但是它们都是只写的。
它们都没有 read 方法,只有 write 方法。然而,它们仍然是类文件对象,因此你可以将其它任何 (类) 文件对象赋值给它们来重定向其输出。

 

使用sys重定向输出
print ‘Dive in‘                 # 标准输出
saveout = sys.stdout            # 在重定向前保存stdout,这样的话之后你还可以将其设回正常
fsock = open(‘out.log‘, ‘w‘)    # 打开一个新文件用于写入。如果文件不存在,将会被创建。如果文件存在,将被覆盖。
sys.stdout = fsock              # 所有后续的输出都会被重定向到刚才打开的新文件上。

print  ‘This message will be logged instead of displayed‘    # 这样只会将输出结果“打印”到日志文件中;屏幕上不会看到输出
sys.stdout = saveout            # 在我们将 stdout 搞乱之前,让我们把它设回原来的方式。     
fsock.close()                   # 关闭日志文件。

重定向错误信息
fsock = open(‘error.log‘, ‘w‘)                   # 打开你要存储调试信息的日志文件。 
sys.stderr = fsock                               # 将新打开的日志文件的文件对象赋值给stderr以重定向标准错误。
raise Exception, ‘this error will be logged‘     # 引发一个异常,没有在屏幕上打印出任何东西,所有正常的跟踪信息已经写进error.log

 

还要注意你既没有显式关闭日志文件,也没有将 stderr 设回最初的值。
这样挺好,因为一旦程序崩溃 (由于引发的异常),Python 将替我们清理并关闭文件打印到 stderr
向标准错误写入错误信息是很常见的,所以有一种较快的语法可以立刻导出信息

>>> print ‘entering function‘
entering function
>>> import sys
>>> print >> sys.stderr, ‘entering function‘

entering function 
print 语句的快捷语法可以用于写入任何打开的文件 (或者是类文件对象)。 
在这里,你可以将单个print语句重定向到stderr而且不用影响后面的print语句。

 

简明教程中的实例
import sys
def readfile(filename):
        ‘‘‘Print a file to the standard output.‘‘‘
        f = file(filename)
        while True:
                line = f.readline()
                if len(line) == 0:
                        break
                print line,
        f.close()
if len(sys.argv) < 2:
        print ‘No action specified.‘
        sys.exit()
if sys.argv[1].startswith(‘--‘):
        option = sys.argv[1][2:]
        if option == ‘version‘:
                print "Version 1.2"
        elif option == ‘help‘:
                print ‘‘‘\
This program prints file to the standard output.
Any number of files can be specified.
Options include:
        --version : Prints the version number
        --help    : Display this help‘‘‘
        else:
                print ‘Unknown option.‘
else:
        for filename in sys.argv[1:]:
                readfile(filename)

 

执行结果:
[root@node1 python]# python cat.py 
No action specified.

[root@node1 python]# python cat.py --help
This program prints file to the standard output.
Any number of files can be specified.
Options include:
    --version : Prints the version number
    --help    : Display this help

 

[root@node1 python]# python cat.py --version
Version 1.2

 

[root@node1 python]# python cat.py --t
Unknown option.

 

[root@node1 python]# python cat.py /tmp/test.txt 
hello girl!
hello boy!
hello man!

 

工作原理:
定义readfile 函数逐行读取文件,执行脚本后,判断参数。
若参数长度小于2(命令本身占一个长度,也就是说,命令+参数),输出No action specified.
若参数是以“--”开头(通过sys.startswith 获取开头),则从该参数的第3个字符起截取,比如--help,得到的就是help
若取得的值为"help" 则显示帮助信息,若取得的值为"version",则显示版本信息
若没有匹配到任何选项,则显示“Unknown option”
若直接加文件名,则打印出文件内容

 

Python 学习笔记 (8)—— sys模块

上一篇:Python的类实例方法,类方法,类静态方法


下一篇:python 根据中文构造url的方法