《Python:编程:从入门到实践》学习笔记_第10章 文件和异常

第10章 文件和异常

异常是Python创建的特殊对象，用于管理程序运行时出现的错误；你还将学习模块json，它让你能够保存用户数据，以免在程序停止运行后丢失。

---

10.1 从文件中读取数据

文本文件可存储的数据量多得难以置信,每当需要分析或修改存储在文件中的信息时，读取文件都很有用，对数据分析应用程序来说尤其如此。例如，你可以编写一个这样的程序：读取一个文本文件的内容，重新设置这些数据的格式并将其写入文件，让浏览器能够显示这些内容。

要使用文本文件中的信息，首先需要将信息读取到内存中。为此，你可以一次性读取文件的全部内容，也可以以每次一行的方式逐步读取。

10.1.1 读取整个文件

要读取文件，需要一个包含几行文本的文件。

pi_digits.txt（文件名） 
3.1415926535 
 8979323846 
 2643383279

下面的程序打开并读取这个文件，再将其内容显示到屏幕上：

with open('pi_digits.txt') as file_object: 
 	contents = file_object.read() 
 	print(contents)

要以任何方式使用文件——哪怕仅仅是打印其内容，都得先打开文件，这样才能访问它。函数open() 接受一个参数：要打开的文件的名称。Python在当前执行的文件所在的目录中查找指定的文件。

关键字with在不再需要访问文件后将其关闭。在这个程序中，注意到我们调用了open()，但没有调用close()；你也可以调用open()和close()来打开和关闭文件，但这样做时，如果程序存在bug，导致close()语句未执行，文件将不会关闭。这看似微不足道，但未妥善地关闭文件可能会导致数据丢失或受损。如果在程序中过早地调用close()，你会发现需要使用文件时它已关闭（无法访问），这会导致更多的错误。并非在任何情况下都能轻松确定关闭文件的恰当时机，但通过使用前面所示的结构，可让Python去确定：你只管打开文件，并在需要时使用它，Python自会在合适的时候自动将其关闭。

有了表示pi_digits.txt的文件对象后，我们使用方法read() 读取这个文件的全部内容，并将其作为一个长长的字符串存储在变量contents中并打印。

3.1415926535 
 8979323846 
 2643383279

相比于原始文件，该输出唯一不同的地方是末尾多了一个空行。为何会多出这个空行呢？因为read()到达文件末尾时返回一个空字符串，而将这个空字符串显示出来时就是一个空行。要删除多出来的空行，可在print语句中使用rstrip()：

with open('pi_digits.txt') as file_object: 
 contents = file_object.read() 
 print(contents.rstrip()) 

现在，输出与原始文件的内容完全相同。

10.1.2 文件路径

当你将类似pi_digits.txt这样的简单文件名传递给函数open()时，Python将在当前执行的文件（即.py程序文件）所在的目录中查找文件。

根据你组织文件的方式，有时可能要打开不在程序文件所属目录中的文件。例如，你可能将程序文件存储在了文件夹python_work中，而在文件夹python_work中，有一个名为text_files的文件夹，用于存储程序文件操作的文本文件。虽然文件夹text_files包含在文件夹python_work中，但仅向open()传递位于该文件夹中的文件的名称也不可行，因为Python只在文件夹python_work中查找，而不会在其子文件夹text_files中查找。要让Python打开不与程序文件位于同一个目录中的文件，需要提供文件路径，它让Python到系统的特定位置去查找。由于文件夹text_files位于文件夹python_work中，因此可使用相对文件路径来打开该文件夹中的文件。相对文件路径让Python到指定的位置去查找，而该位置是相对于当前运行的程序所在目录的。在Linux和OS X中，你可以这样编写代码：

with open('text_files/filename.txt') as file_object: 

这行代码让Python到文件夹python_work下的文件夹text_files中去查找指定的.txt文件。在Windows系统中，在文件路径中使用反斜杠（\） 而不是斜杠（/）：

with open('text_files\filename.txt') as file_object:

你还可以将文件在计算机中的准确位置告诉Python，这样就不用关心当前运行的程序存储在什么地方了。这称为绝对文件路径。在相对路径行不通时，可使用绝对路径。例如，如果text_files并不在文件夹python_work中，而在文件夹other_files中，则向open()传递路径’text_files/ filename.txt’行不通，因为Python只在文件夹python_work中查找该位置。

在Linux和OS X中，绝对路径类似于下面这样：

file_path = '/home/ehmatthes/other_files/text_files/filename.txt' 
with open(file_path) as file_object: 

而在Windows系统中，它们类似于下面这样：

file_path = 'C:\Users\ehmatthes\other_files\text_files\filename.txt' 
with open(file_path) as file_object: 

注：Windows系统有时能够正确地解读文件路径中的斜杠。如果你使用的是Windows系统，且
结果不符合预期，请确保在文件路径中使用的是反斜杠。

10.1.3 逐行读取

读取文件时，常常需要检查其中的每一行：你可能要在文件中查找特定的信息，或者要以某种方式修改文件中的文本。

要以每次一行的方式检查文件，可对文件对象使用for循环：

filename = 'pi_digits.txt' 
with open(filename) as file_object: 
	for line in file_object: 
		print(line) 

我们打印每一行时，发现空白行更多了：

3.1415926535 

 8979323846 
 
 2643383279 
 

因为在这个文件中，每行的末尾都有一个看不见的换行符，而print语句也会加上一个换行符，因此每行末尾都有两个换行符：一个来自文件，另一个来自print语句。要消除这些多余的空白行，可在print语句中使用rstrip()：

filename = 'pi_digits.txt' 

with open(filename) as file_object: 
	for line in file_object: 
		print(line.rstrip()) 

现在，输出又与文件内容完全相同了：

3.1415926535 
 8979323846 
 2643383279 

10.1.4 创建一个包含文件各行内容的列表

使用关键字with时，open()返回的文件对象只在with代码块内可用。如果要在with代码块外访问文件的内容，可在with代码块内将文件的各行存储在一个列表中，并在with代码块外使用该列表：你可以立即处理文件的各个部分，也可推迟到程序后面再处理。

filename = 'pi_digits.txt' 

with open(filename) as file_object: 
	lines = file_object.readlines() 

for line in lines: 
	print(line.rstrip()) 

10.1.5 使用文件的内容

将文件读取到内存中后，就可以以任何方式使用这些数据了。

filename = 'pi_digits.txt' 

with open(filename) as file_object: 
	lines = file_object.readlines() 

pi_string = '' 
for line in lines: 
		pi_string += line.rstrip() 
 
print(pi_string) 
print(len(pi_string)) 

我们打印这个字符串及其长度：

3.1415926535 8979323846 2643383279 
36

在变量pi_string存储的字符串中，包含原来位于每行左边的空格，为删除这些空格，可使用strip() 而不是rstrip()：

filename = 'pi_30_digits.txt' 

with open(filename) as file_object: 
	lines = file_object.readlines() 

pi_string = '' 
for line in lines: 
	pi_string += line.strip() 
 
print(pi_string) 
print(len(pi_string)) 

这样，我们就获得了一个这样的字符串：它包含精确到30位小数的圆周率值。

3.141592653589793238462643383279 
32

注： 读取文本文件时，Python将其中的所有文本都解读为字符串。如果你读取的是数字，并要将其作为数值使用，就必须使用函数int()将其转换为整数，或使用函数float()将其转换为浮点数。

10.1.6 包含一百万位的大型文件

前面我们分析的都是一个只有三行的文本文件，但这些代码示例也可处理大得多的文件。如果我们有一个文本文件，其中包含精确到小数点后1 000 000位而不是30位的圆周率值，也可创建一个包含所有这些数字的字符串。为此，我们无需对前面的程序做任何修改，只需将这个文件传递给它即可。

对于你可处理的数据量，Python没有任何限制；只要系统的内存足够多，你想处理多少数据都可以。

10.1.7 圆周率值中包含你的生日吗

我一直想知道自己的生日是否包含在圆周率值中。下面来扩展刚才编写的程序，以确定某个人的生日是否包含在圆周率值的前1 000 000位中。

filename = 'pi_million_digits.txt' 

with open(filename) as file_object: 
 	lines = file_object.readlines() 

pi_string = '' 
for line in lines: 
	pi_string += line.rstrip() 
	
birthday = input("Enter your birthday, in the form mmddyy: ") 
if birthday in pi_string: 
 	print("Your birthday appears in the first million digits of pi!") 
else: 
 	print("Your birthday does not appear in the first million digits of pi.") 

运行一下这个程序：

Enter your birthdate, in the form mmddyy: 120372 
Your birthday appears in the first million digits of pi! 

---

10.2 写入文件

保存数据的最简单的方式之一是将其写入到文件中。通过将输出写入文件，即便关闭包含程序输出的终端窗口，这些输出也依然存在：你可以在程序结束运行后查看这些输出，可与别人分享输出文件，还可编写程序来将这些输出读取到内存中并进行处理。

10.2.1 写入空文件

要将文本写入文件，你在调用open()时需要提供另一个实参，告诉Python你要写入打开的文件。

filename = 'programming.txt' 

with open(filename, 'w') as file_object: 
	file_object.write("I love programming.") 

第二个实参（‘w’）告诉Python，我们要以写入模式打开这个文件。打开文件时，可指定读取模式（‘r’）、写入模式（‘w’）、附加模式（‘a’） 或让你能够 读取和写入文件的模式（‘r+’）。如果你省略了模式实参，Python将以默认的只读模式打开文件。

如果你要写入的文件不存在，函数open()将自动创建它。然而，以写入（‘w’）模式打开文件时千万要小心，因为如果指定的文件已经存在，Python将在返回文件对象前清空该文件。

这个程序没有终端输出，但如果你打开文件programming.txt，将看到其中包含如下一行内容：

I love programming.

注：Python只能将字符串写入文本文件。要将数值数据存储到文本文件中，必须先使用函数str()将其转换为字符串格式。

10.2.2 写入多行

函数write()不会在你写入的文本末尾添加换行符。

filename = 'programming.txt' 

with open(filename, 'w') as file_object: 
	file_object.write("I love programming.") 
 	file_object.write("I love creating new games.") 

如果你打开programming.txt，将发现两行内容挤在一起：

I love programming.I love creating new games. 

要让每个字符串都单独占一行，需要在write()语句中包含换行符：

filename = 'programming.txt' 

with open(filename, 'w') as file_object: 
 	file_object.write("I love programming.\n") 
 	file_object.write("I love creating new games.\n") 

现在，输出出现在不同行中：

I love programming. 
I love creating new games. 

10.2.3 附加到文件

如果你要给文件添加内容，而不是覆盖原有的内容，可以附加模式打开文件。你以附加模式打开文件时，Python不会在返回文件对象前清空文件，而你写入到文件的行都将添加到文件末尾。如果指定的文件不存在，Python将为你创建一个空文件。

filename = 'programming.txt' 

with open(filename, 'a') as file_object: 
	file_object.write("I also love finding meaning in large datasets.\n") 
 	file_object.write("I love creating apps that can run in a browser.\n") 

我们写入了两行，它们被添加到文件programming.txt末尾：

I love programming. 
I love creating new games.
I also love finding meaning in large datasets. 
I love creating apps that can run in a browser. 

---

10.3 异常

Python使用被称为异常的特殊对象来管理程序执行期间发生的错误。每当发生让Python不知所措的错误时，它都会创建一个异常对象。如果你编写了处理该异常的代码，程序将继续运行；如果你未对异常进行处理，程序将停止，并显示一个traceback，其中包含有关异常的报告。

异常是使用try-except代码块处理的。try-except代码块让Python执行指定的操作，同时告诉Python发生异常时怎么办。使用了try-except代码块时，即便出现异常，程序也将继续运行：显示你编写的友好的错误消息，而不是令用户迷惑的traceback。

10.3.1 处理 ZeroDivisionError 异常

print(5/0) 

显然，Python无法这样做，因此你将看到一个traceback：

Traceback (most recent call last): 
  File "division.py", line 1, in <module> 
    print(5/0) 
ZeroDivisionError: division by zero

10.3.2 使用 try-except 代码块

当你认为可能发生了错误时，可编写一个try-except代码块来处理可能引发的异常。你让Python尝试运行一些代码，并告诉它如果这些代码引发了指定的异常，该怎么办。

try: 
 	print(5/0) 
except ZeroDivisionError: 
 	print("You can't divide by zero!") 

我们将导致错误的代码行print(5/0)放在了一个try代码块中。如果try代码块中的代码运行起来没有问题，Python将跳过except代码块；如果try代码块中的代码导致了错误，Python将查找这样的except代码块，并运行其中的代码，即其中指定的错误与引发的错误相同。

这样，用户看到的是一条友好的错误消息，而不是traceback：

You can't divide by zero!

10.3.3 使用异常避免崩溃

发生错误时，如果程序还有工作没有完成，妥善地处理错误就尤其重要。这种情况经常会出现在要求用户提供输入的程序中；如果程序能够妥善地处理无效输入，就能再提示用户提供有效输入，而不至于崩溃。

print("Give me two numbers, and I'll divide them.") 
print("Enter 'q' to quit.") 

while True: 
	first_number = input("\nFirst number: ") 
 	if first_number == 'q': 
 		break 
	second_number = input("Second number: ") 
 	if second_number == 'q': 
 		break 
	answer = int(first_number) / int(second_number) 
 	print(answer) 

这个程序没有采取任何处理错误的措施，因此让它执行除数为0的除法运算时，它将崩溃：

Give me two numbers, and I'll divide them. 
Enter 'q' to quit. 

First number: 5 
Second number: 0 
Traceback (most recent call last): 
  File "division.py", line 9, in <module> 
    answer = int(first_number) / int(second_number) 
ZeroDivisionError: division by zero 

10.3.4 else 代码块

依赖于try代码块成功执行的代码都应放到else代码块中：

print("Give me two numbers, and I'll divide them.") 
print("Enter 'q' to quit.") 

while True: 
 	first_number = input("\nFirst number: ") 
 	if first_number == 'q': 
 		break 
 	second_number = input("Second number: ") 
	try: 
 		answer = int(first_number) / int(second_number) 
 	except ZeroDivisionError: 
 		print("You can't divide by 0!") 
	else: 
 		print(answer)

如果try代码块因除零错误而失败，我们就打印一条友好的消息，告诉用户如何避免这种错误。程序将继续运行，用户根本看不到traceback：

Give me two numbers, and I'll divide them. 
Enter 'q' to quit. 

First number: 5 
Second number: 0 
You can't divide by 0! 

First number: 5 
Second number: 2 
2.5 

First number: q

try-except-else代码块的工作原理大致如下：Python尝试执行try代码块中的代码；只有可能引发异常的代码才需要放在try语句中。有时候，有一些仅在try代码块成功执行时才需要运行的代码；这些代码应放在else代码块中。except代码块告诉Python，如果它尝试运行try代码块中的代码时引发了指定的异常，该怎么办。

10.3.5 处理 FileNotFoundError 异常

使用文件时，一种常见的问题是找不到文件：你要查找的文件可能在其他地方、文件名可能不正确或者这个文件根本就不存在。对于所有这些情形，都可使用try-except代码块以直观的方式进行处理。

filename = 'alice.txt' 

with open(filename) as f_obj: 
 	contents = f_obj.read() 

Python无法读取不存在的文件，因此它引发一个异常：

Traceback (most recent call last): 
  File "alice.py", line 3, in <module> 
    with open(filename) as f_obj: 
FileNotFoundError: [Errno 2] No such file or directory: 'alice.txt'

这个错误是函数open()导致的，因此要处理这个错误，必须将try语句放在包含open()的代码行之前：

filename = 'alice.txt' 

try: 
	with open(filename) as f_obj: 
 	contents = f_obj.read() 
except FileNotFoundError: 
 	msg = "Sorry, the file " + filename + " does not exist." 
 	print(msg) 

在这个示例中，try代码块引发FileNotFoundError异常，因此Python找出与该错误匹配的except代码块，并运行其中的代码。最终的结果是显示一条友好的错误消息，而不是traceback：

Sorry, the file alice.txt does not exist. 

10.3.6 分析文本

下面来提取童话Alice in Wonderland的文本，并尝试计算它包含多少个单词。

>>> title = "Alice in Wonderland" 
>>> title.split() 
['Alice', 'in', 'Wonderland'] 

方法split() 以空格为分隔符将字符串分拆成多个部分，并将这些部分都存储到一个列表中。结果是一个包含字符串中所有单词的列表，虽然有些单词可能包含标点。

filename = 'alice.txt' 

try: 
 	with open(filename) as f_obj: 
 		contents = f_obj.read() 
except FileNotFoundError: 
 	msg = "Sorry, the file " + filename + " does not exist." 
 	print(msg) 
else: 
 	# 计算文件大致包含多少个单词
	words = contents.split() 
	num_words = len(words) 
	print("The file " + filename + " has about " + str(num_words) + " words.")

我们打印一条消息，指出文件包含多少个单词。这些代码都放在else代码块中，因为仅当try代码块成功执行时才执行它们。输出指出了文件alice.txt包含多少个单词：

The file alice.txt has about 29461 words. 

10.3.7 使用多个文件

def count_words(filename): 
	"""计算一个文件大致包含多少个单词""" 
 	try: 
 		with open(filename) as f_obj: 
 			contents = f_obj.read() 
 	except FileNotFoundError: 
 		msg = "Sorry, the file " + filename + " does not exist." 
 		print(msg) 
 	else: 
 		# 计算文件大致包含多少个单词
 		words = contents.split() 
 		num_words = len(words) 
 		print("The file " + filename + " has about " + str(num_words) + 
 			  " words.") 

filename = 'alice.txt' 
count_words(filename)

现在可以编写一个简单的循环，计算要分析的任何文本包含多少个单词了。

def count_words(filename): 
	--snip-- 

filenames = ['alice.txt', 'siddhartha.txt', 'moby_dick.txt', 'little_women.txt'] 
for filename in filenames: 
	count_words(filename) 

文件siddhartha.txt不存在，但这丝毫不影响这个程序处理其他文件：

The file alice.txt has about 29461 words. 
Sorry, the file siddhartha.txt does not exist. 
The file moby_dick.txt has about 215136 words. 
The file little_women.txt has about 189079 words. 

在这个示例中，使用try-except代码块提供了两个重要的优点：避免让用户看到traceback；让程序能够继续分析能够找到的其他文件。如果不捕获因找不到siddhartha.txt而引发的FileNotFoundError异常，用户将看到完整的traceback，而程序将在尝试分析Siddhartha后停止运行——根本不分析Moby Dick和Little Women。

10.3.8 失败时一声不吭

在前一个示例中，我们告诉用户有一个文件找不到。但并非每次捕获到异常时都需要告诉用户，有时候你希望程序在发生异常时一声不吭，就像什么都没有发生一样继续运行。要让程序在失败时一声不吭，可像通常那样编写try代码块，但在except代码块中明确地告诉Python什么都不要做。Python有一个pass语句，可在代码块中使用它来让Python什么都不要做：

def count_words(filename): 
 	"""计算一个文件大致包含多少个单词""" 
 	try: 
 		--snip-- 
 	except FileNotFoundError: 
		pass 
 	else: 
 		--snip-- 

filenames = ['alice.txt', 'siddhartha.txt', 'moby_dick.txt', 'little_women.txt'] 
for filename in filenames: 
	count_words(filename)

用户将看到存在的每个文件包含多少个单词，但没有任何迹象表明有一个文件未找到：

The file alice.txt has about 29461 words. 
The file moby_dick.txt has about 215136 words. 
The file little_women.txt has about 189079 words.

pass语句还充当了占位符，它提醒你在程序的某个地方什么都没有做，并且以后也许要在这里做些什么。例如，在这个程序中，我们可能决定将找不到的文件的名称写入到文件missing_files.txt中。用户看不到这个文件，但我们可以读取这个文件，进而处理所有文件找不到的问题。

10.3.9 决定报告哪些错误

Python的错误处理结构让你能够细致地控制与用户分享错误信息的程度，要分享多少信息由你决定。

编写得很好且经过详尽测试的代码不容易出现内部错误，如语法或逻辑错误，但只要程序依赖于外部因素，如用户输入、存在指定的文件、有网络链接，就有可能出现异常。凭借经验可判断该在程序的什么地方包含异常处理块，以及出现错误时该向用户提供多少相关的信息。

---

10.4 存储数据

很多程序都要求用户输入某种信息，程序都把用户提供的信息存储在列表和字典等数据结构中。用户关闭程序时，你几乎总是要保存他们提供的信息；一种简单的方式是使用模块json来存储数据。

模块json让你能够将简单的Python数据结构转储到文件中，并在程序再次运行时加载该文件中的数据。你还可以使用json在Python程序之间分享数据。更重要的是，JSON数据格式并非Python专用的，这让你能够将以JSON格式存储的数据与使用其他编程语言的人分享。
注：JSON（JavaScript Object Notation）格式最初是为JavaScript开发的，但随后成了一种常见格式，被包括Python在内的众多语言采用。

10.4.1 使用 json.dump()和 json.load()

我们来编写一个存储一组数字的简短程序，再编写一个将这些数字读取到内存中的程序。第一个程序将使用json.dump() 来存储这组数字，而第二个程序将使用json.load()。

函数json.dump()接受两个实参：要存储的数据以及可用于存储数据的文件对象。

import json 

numbers = [2, 3, 5, 7, 11, 13] 
filename = 'numbers.json' 
with open(filename, 'w') as f_obj: 
	json.dump(numbers, f_obj) 

这个程序没有输出，但我们可以打开文件numbers.json，看看其内容。数据的存储格式与Python中一样：

[2, 3, 5, 7, 11, 13] 

下面再编写一个程序，使用json.load()将这个列表读取到内存中：

import json 

filename = 'numbers.json' 
with open(filename) as f_obj: 
	numbers = json.load(f_obj) 
 
print(numbers)

我们打印恢复的数字列表，看看它是否与number_writer.py中创建的数字列表相同：

[2, 3, 5, 7, 11, 13] 

这是一种在程序之间共享数据的简单方式。

10.4.2 保存和读取用户生成的数据

对于用户生成的数据，使用json保存它们大有裨益，因为如果不以某种方式进行存储，等程序停止运行时用户的信息将丢失。

import json 

username = input("What is your name? ") 

filename = 'username.json' 
with open(filename, 'w') as f_obj: 
	json.dump(username, f_obj) 
	print("We'll remember you when you come back, " + username + "!") 

我们打印一条消息，指出我们存储了他输入的信息：

What is your name? Eric 
We'll remember you when you come back, Eric! 

现在再编写一个程序，向其名字被存储的用户发出问候：

import json 

filename = 'username.json' 

with open(filename) as f_obj: 
	username = json.load(f_obj) 
	print("Welcome back, " + username + "!") 

恢复用户名后，我们就可以欢迎用户回来了：

Welcome back, Eric!

我们将这两个程序合并到一个程序中。

import json 

# 如果以前存储了用户名，就加载它
# 否则，就提示用户输入用户名并存储它
filename = 'username.json' 
try: 
	with open(filename) as f_obj: 
		username = json.load(f_obj) 
except FileNotFoundError: 
	username = input("What is your name? ") 
	with open(filename, 'w') as f_obj: 
 		json.dump(username, f_obj) 
 		print("We'll remember you when you come back, " + username + "!") 
else: 
 	print("Welcome back, " + username + "!")

无论执行的是except代码块还是else代码块，都将显示用户名和合适的问候语。如果这个程序是首次运行，输出将如下：

What is your name? Eric 
We'll remember you when you come back, Eric! 

否则，输出将如下：

Welcome back, Eric!

10.4.3 重构

你经常会遇到这样的情况：代码能够正确地运行，但可做进一步的改进——将代码划分为一系列完成具体工作的函数。这样的过程被称为重构。

import json
 
def greet_user(): 
	"""问候用户，并指出其名字""" 
 	filename = 'username.json' 
 	try: 
 		with open(filename) as f_obj: 
 			username = json.load(f_obj) 
 	except FileNotFoundError: 
 		username = input("What is your name? ") 
 		with open(filename, 'w') as f_obj: 
 			json.dump(username, f_obj) 
 			print("We'll remember you when you come back, " + username + "!") 
 	else: 
 		print("Welcome back, " + username + "!") 

greet_user() 

下面来重构greet_user()，让它不执行这么多任务。

import json 

def get_stored_username(): 
	"""如果存储了用户名，就获取它""" 
 	filename = 'username.json' 
 	try: 
 		with open(filename) as f_obj: 
 			username = json.load(f_obj) 
 	except FileNotFoundError: 
		return None 
 	else: 
 		return username 

def greet_user(): 
 	"""问候用户，并指出其名字""" 
 	username = get_stored_username() 
	if username: 
 		print("Welcome back, " + username + "!") 
 	else: 
 		username = input("What is your name? ") 
 		filename = 'username.json' 
 		with open(filename, 'w') as f_obj: 
 			json.dump(username, f_obj) 
 			print("We'll remember you when you come back, " + username + "!") 

greet_user()

我们还需将greet_user()中的另一个代码块提取出来：将没有存储用户名时提示用户输入的代码放在一个独立的函数中：

import json 

def get_stored_username(): 
	"""如果存储了用户名，就获取它""" 
 	--snip-- 

def get_new_username(): 
 	"""提示用户输入用户名""" 
 	username = input("What is your name? ") 
 	filename = 'username.json' 
 	with open(filename, 'w') as f_obj: 
 		json.dump(username, f_obj) 
 	return username 

def greet_user(): 
 	"""问候用户，并指出其名字""" 
 	username = get_stored_username() 
 	if username: 
 		print("Welcome back, " + username + "!") 
 	else: 
 		username = get_new_username() 
 		print("We'll remember you when you come back, " + username + "!") 

greet_user()

---

10.5 小结

在本章中，你学习了：如何使用文件；如何一次性读取整个文件，以及如何以每次一行的方式读取文件的内容；如何写入文件，以及如何将文本附加到文件末尾；什么是异常以及如何处理程序可能引发的异常；如何存储Python数据结构，以保存用户提供的信息，避免用户每次运行程序时都需要重新提供。