[Python] 输入与输出

1. 读取命令行选项

Python启动时，命令行选项放置在列表sys.argv中。例如：

import sys
if len(sys.argv) != 3:
    sys.stderr.write("Usage: python %s inputfile outputfile\n" % sys.argv[0])
    raise SystemExit(1)
inputfile = sys.argv[1]
outputfile = sys.argv[2]

在该程序中，sys.argv[0]包含所执行脚本的名称。对于更复杂的命令行处理，可以使用optparse模块。例如：

imprt optparse
p = optparse.OptionParser()
# 选项接受一个参数
p.add_option("-o", action="store", dest="outfile")
p.add_option("--output", action="store", dest="outfile")
# 选项设置一个布尔值标志
p.add_option("-d", action="store_true", dest="debug")
p.add_option("--debug", action="store_true", dest="debug")
# 设置选定选项的默认值
p.set_defaults(debug=False)
# 解析命令行
opts, args = p.parse_args()
# 检索选项设置
outfile = opts.outfile
debugmode = opts.debug

以上例子中添加了两类选项。第一个选项-o或--output具有一个必需参数。这是通过指定action='store'实现的。dest参数选择将在解析后存储参数值的属性名称。p.set_defaults()方法设置一个或多个选项的默认值。
使用p.parse_args()方法执行解析。该方法返回一个二元组(opts, args)，其中opts是包含已解析选项值的对象，args是命令行上未解析为选项的项目列表。选项值使用opts.dest检索。
 

2. 环境变量

可以通过字典os.environ访问环境变量，例如：

import os
path = os.environ["PATH"]
user = os.environ["USER"]
editor = os.environ["EDITOR"]

要修改环境变量，需要设置os.environ变量，例如：

os.environ["FOO"] = "BAR"

3. 文件和文件对象

内置函数open(name [, mode [, bufsize]])用于打开和创建文件对象，例如：

f = open("foo")
f = open("foo", "r")
f = open("foo", "w")

文件模式"r"表示读取，"w"表示写入，"a"表示附加。这些文件模式假定采用文本模式，可以隐式对换行字符"\n"执行转换。如果正在处理二进制数据，可以将"b"附加到文件模式后面，如"rb"或"wb"。这将禁用换行符转换。如果使用模式"U"或"rU"打开文件，将会提供通用的换行符支持，以方便阅读。
通过提供"+"字符，比如"r+"或"w+"，可以打开文件进行直接更新。打开文件进行更新时，可以同时执行输入和输出，只要所有输出操作在任何后续输入操作之前清除其数据即可。如果使用"w+"模式打开文件，其长度首先会被截断为0。
可选的bufsize参数控制文件的缓冲行为，其中0表示没有缓存，1表示进行了行缓冲，任何其他正值都表示将使用的近似缓冲区大小。
下表列表file对象支持的方法：

read()方法以字符串形式返回整个文件，除非指定了最大字符数。readline()方法返回下一行输入，包括最后的换行符。readlines()方法以字符串列表的形式返回所有输入行。readlines()方法接受一个Size参数，指定在停止读取之前要读取的近似字符数。
write()方法将一个字符串写入到文件。writelines()方法将一个字符串列表写入到文件中。write()和writelines()不会将换行字符添加到输出中，所有生成的所有输出都应该已经包含所有必要的格式。
在内部，每个文件对象都有一个文件指针，用于存储下次读取或写入操作所需的字节偏移位置。tell()方法以长整型返回文件指针的当前值。seek()方法根据给定的offset和whence中的位置规则随机访问文件的各个部分。
fileno()方法返回文件的整数文件描述符，有时用在某些库模块中的低级I/O操作中。
文件对象还拥有一些只读数据属性，例如：

4. 标准输入、输出和错误

解释器提供了3种标准文件对象，分别为标准输入、标准输出和标准错误，它们在sys模块中分别以sys.stdin、sys.stdout和sys.stderr的形式提供。stdin是与提供给解释器的输入字符流相对应的文件对象。stdout是一个文件对象，接收由print生成的输出。stderr是接收错误消息的文件。
例如，以下代码写入标准输出并从标准输入中读取一行输入：

import sys
sys.stdout.write("Enter your name:")
name = sys.stdin.readline()

另外，内置函数input(prompt)也可以从stdin读取一行文本，并可以打印一个提示符。键盘中断（通常由Ctrl+C生成）会生成KeyboardInterrupt异常，可使用异常处理程序捕获该异常。
 

5. print()函数

要打印一系列以空格分隔的值，只需将这些值提供给print()，例如：

print("The values are", x, y, z)

要禁止或更改行终止，可以使用end=ending关键字参数，例如：

print("The values are", x, y, z, end='') # 禁止换行符

要将输出重定向到一个文件，可以使用file=outfile关键字参数，例如：

print("The values are", x, y, z, file=f)

要更改项目之间的分隔字符，可以使用sep=sepchr关键字参数，例如：

print("The values are", x, y, z, spe=',')

6. 文本输出中的变量插入

通过格式化I/O与三对带引号的字符串来将变量插入字符串中，例如：

# 斜杠可以防止第一行显示为空行
form = """\
Dear %(name)s,
Please send back my %(item)s or pay me $%(amount)0.2f.
"""
print form % {'name': 'Mr.Bush', 'item': 'blender', 'amount': 50.00}

format()方法可以使以上代码更简洁，例如：

form = """\
Dear {name}s,
Please send back my {item}s or pay me $%{amount:0.2f}.
"""
print form.format {name='Mr.Bush', item='blender', amount=50.00}

7. Unicode处理

如果有一个原始字节字符串s，其中包含已编码的Unicode字符串表示，那么可以使用s.decode([encoding [, errors]])方法将其转换为合适的Unicode字符串。要将Unicode字符串u转换为已编码的字节字符串，可以使用字符串方法u.encoding([encoding [, errors]])。这两种转换字符串都需要使用一个特殊编码名称，指定如何在Unicode字符值与字节字符串中的一个8位字符序列上建立起映射关系。以下编码是最常用的：

默认编码在site模块中设置，可以使用sys.getdefaultencoding()查询。在很多情况下，默认编码是ascii，utf-8也是一种非常常见的设置。转换字符串时，如果遇到无法转换的字符，可能会引发UnicodeError异常。encode()和decode()方法的errors参数决定了如何处理编码错误。可用参数如下：

处理Unicode字符串时，无法将原始Unicode数据直接写入文件。因为Unicode字符在内部表示为多字节整数。Unicode字符串的外部表示总是根据具体的编码规则进行，该编码规则应该明确定义如何将Unicode字符表示为字节序列。内置的codecs模块包含一组函数，用于根据各种不同的数据编码方案，在字节数据与Unicode字符串之间来回切换。
处理Unicode文件最直接方式是用codecs.open(filename [, mode [, encoding [, errors]]])函数，例如：

f = codecs.open('foo.txt', 'r', 'utf-8', 'strict')
g = codecs.open('bar.txt', 'w', 'utf-8')

这两个函数创建了一个文件对象，用于读取或写入Unicode字符串。如果已经拥有一个文件对象，可以使用codecs.EncodedFile(file, inputenc [, outputenc [, errors]])函数为该对象添加一个编码包装器，例如：

f = open("foo.txt", "rb")
...
fenc = codecs.EncodeFile(f, 'utf-8')

8. 对象持久性

有时会需要将对象内容保存到一个文件或从中进行还原。执行该任务的一种方法是编写两个函数，以一种特殊格式在文件中读取和写入数据。另一种方法是用pickle和shelve模块。
pickle模块将对象序列化为一个字节流，这个字节流可以写入到文件并在以后进行还原。例如：

import pickle
obj = SomeObject()
f = open(filename, 'wb')
pickle.dump(obj, f) # 将对象保存到f上
f.close()

要还原对象，可以使用以下代码：

import pickle
f = open(filename, 'rb')
obj = pickle.load(f) # 还原对象
f.close()

可以依次发生一系列dump()操作来保存对象序列。要还原这些对象，只需使用一个类似的load()操作序列。shelve模块类似于pickle，但它将对象保存在一个类似字典的数据库中。例如：

import shelve
obj = SomeObject()
db = shelve.open("filename")
db['key'] = obj
...
obj = db['key']
db.close()

尽管由shelve创建的对象类似于一个字典，但它具有一些限制。首先，键必须是字符串。其次，shelve中存储的值必须与pickle兼容。