本文旨在提供一种更清晰、更易于维护的方式，来修改文件中特定行的数值。通过将文件读取、数据处理和文件写入分离成独立的函数，可以有效提高代码的可读性和可重用性。我们将展示如何使用 Python 将文本文件转换为易于操作的数据结构，修改数据，并将修改后的数据写回文件。
代码结构优化
原始代码存在一些问题，例如在读取文件的同时进行写入，这可能导致数据损坏。此外，代码逻辑较为复杂，可读性较差。为了解决这些问题，我们将代码分解为三个独立的函数：getBooks、decrementBookUnits 和 writeBooks。
1. getBooks(file)：读取文件并转换为数据结构
此函数负责读取文件内容，并将其转换为一个易于操作的 Python 列表，其中每个元素都是一个字典，代表一本书的信息。
def getBooks(file):
"""
读取文件内容，并将其转换为一个列表，其中每个元素都是一个字典，代表一本书的信息。
:param file: 文件名或文件路径。
:return: 一个列表，其中每个元素都是一个字典，代表一本书的信息。
"""
with open(file, 'r') as booksFile:
content = booksFile.read().splitlines()
lines=[line.split(' -- ') for line in content]
books=[dict(entry.split(':') for entry in line) for line in lines]
return books代码解释:
立即学习“Python免费学习笔记（深入）”；
- with open(file, 'r') as booksFile:: 使用 with 语句打开文件，确保文件在使用完毕后自动关闭。
- content = booksFile.read().splitlines(): 读取文件所有内容，并按行分割成一个字符串列表。
- lines=[line.split(' -- ') for line in content]: 将每一行按照 " -- " 分割成键值对列表。
- books=[dict(entry.split(':') for entry in line) for line in lines]: 将每个键值对列表转换为字典，最终生成一个包含所有书籍信息的字典列表。
2. decrementBookUnits(books, isbn)：修改书籍信息
此函数接收书籍列表和 ISBN 作为输入，并在列表中找到匹配的书籍，然后修改其 ISBN 的值。
def decrementBookUnits(books, isbn):
"""
修改书籍列表中指定 ISBN 的书籍的 ISBN 值。
:param books: 书籍列表，每个元素是一个字典。
:param isbn: 要修改的书籍的 ISBN。
:return: 修改后的书籍列表。
"""
for idx, entry in enumerate(books):
if entry['ISBN'] == str(isbn):
books[idx]['ISBN'] = '666'
return books代码解释:
立即学习“Python免费学习笔记（深入）”；
- for idx, entry in enumerate(books):: 遍历书籍列表，idx 是索引，entry 是书籍字典。
- if entry['ISBN'] == str(isbn):: 检查当前书籍的 ISBN 是否与给定的 ISBN 匹配。注意将 isbn 转换为字符串类型进行比较。
- books[idx]['ISBN'] = '666': 如果找到匹配的书籍，则将其 ISBN 的值修改为 '666'。
- return books: 返回修改后的书籍列表。
3. writeBooks(books, file)：将数据写回文件
此函数接收书籍列表和文件名作为输入，并将书籍信息以原始格式写回文件。
def writeBooks(books, file):
"""
将书籍列表以原始格式写回文件。
:param books: 书籍列表，每个元素是一个字典。
:param file: 要写入的文件名。
"""
with open(file, 'w') as booksFile:
for entry in books:
booksFile.writelines(' -- '.join(['='.join([k,v]) for k,v in entry.items()]) + '\n')代码解释:
立即学习“Python免费学习笔记（深入）”；
- with open(file, 'w') as booksFile:: 以写入模式打开文件。
- for entry in books:: 遍历书籍列表。
- ' -- '.join(['='.join([k,v]) for k,v in entry.items()]): 将书籍字典转换为原始格式的字符串。首先，['='.join([k,v]) for k,v in entry.items()] 将每个键值对转换为 "key=value" 格式的字符串，然后 ' -- '.join(...) 将这些字符串用 " -- " 连接起来。
- booksFile.writelines(... + '\n'): 将书籍信息写入文件，并在末尾添加换行符。
使用示例
# 读取书籍信息
books = getBooks('books.txt')
# 修改 ISBN 为 33 的书籍的 ISBN 值
books = decrementBookUnits(books, 33)
# 将修改后的书籍信息写回文件
writeBooks(books, 'books2.txt')注意事项
- 确保 books.txt 文件存在，并且格式正确。
- writeBooks 函数会将修改后的数据写入到 books2.txt 文件中，而不是覆盖原始文件。如果需要覆盖原始文件，请将 writeBooks 函数的第二个参数修改为 'books.txt'。
- 在实际应用中，可以根据需要修改 decrementBookUnits 函数中的修改逻辑，例如，可以将 ISBN 减 1，而不是直接设置为 '666'。
- 异常处理：在实际应用中，应该添加适当的异常处理机制，例如，处理文件不存在、文件格式错误等情况。
总结
通过将文件读取、数据处理和文件写入分离成独立的函数，可以有效提高代码的可读性和可重用性。这种模块化的设计方法使得代码更易于维护和扩展。同时，使用适当的数据结构（如字典列表）可以简化数据处理过程。
