configparser 读写

Question

我正在使用 Python 3.2 和 configparser 模块，并遇到了一些问题。我需要读取配置文件，然后写入配置文件。我尝试了以下操作：

import configparser
data = open('data.txt', 'r+')
a = configparser.ConfigParser()
a.read_file(data)
a['example']['test'] = 'red'
a.write(data)

问题是，当我使用 r+ 打开数据时，当我写入数据时，会附加新信息；它不会覆盖旧的。

import configparser
data = open('data.txt', 'r')
a = configparser.ConfigParser()
a.read_file(data)
a['example']['test'] = 'red'
data = open('data.txt', 'w')
a.write(data)

这种方式 ^ 似乎不安全，因为用 w 打开它会清空文件。如果程序在还没来得及写入之前就崩溃了怎么办？配置文件丢失。用 w 打开之前备份是唯一的解决方案吗？

编辑：

以下也是一种可能性，但是安全吗？

a.write(open('data.txt','w'))

Answer 1

如果你真的担心这一点，你可以写入一个临时文件，然后将临时文件重命名为这个文件——如果配置写入失败，原始文件将保持不变；重命名/移动通常是原子的（尽管在 Windows 下您可能需要直接调用 MoveFileEx，而不是使用 os.rename），因此您可以确定您会具有旧内容或新内容，并且文件不会处于任何其他状态（当然，没有底层文件系统的任何严重故障）。

# ...
a['example']['test'] = 'red'

import tempfile, os
with tempfile.NamedTemporaryFile() as tmp:
    a.write(tmp)

    # or ctypes.windll.kernel32.MoveFileExW(tmp.name, 'data.txt', 2)
    # 2 = MOVEFILE_REPLACE_EXISTING 
    # I'll leave wrapping it in a cross-platform manner up to you
    os.rename(tmp.name, 'data.txt')

Answer 2

当我遇到类似的情况时，这就是我会做的。

import configparser
data = open('data.txt', 'r+')
a = configparser.ConfigParser()
a.read_file(data)
a['example']['test'] = 'red'
data.truncate(0)#<--
data.seek(0)#<--
a.write(data)

这会导致文件对象被截断为零。然后它将指针重置为文件的开头。之后，configparser 可以像正常情况一样处理空文件对象。

应该注意的是，我是在 Python 3 中这样做的。