我正在尝试使用 fileinput
模块的 inplace filtering feature就地重写输入文件。
需要将编码(读取和写入)设置为 latin-1
并尝试将 openhook=fileinput.hook_encoded('latin-1')
传递给 fileinput.input
但因错误而受阻
ValueError: FileInput cannot use an opening hook in inplace mode
经过仔细检查,我发现 fileinput
文档清楚地说明了这一点:您不能同时使用 inplace 和 openhook
我该如何解决这个问题?
最佳答案
据我所知,fileinput
模块无法解决这个问题。您可以通过组合使用 codecs
模块、os.rename()
和 os.remove()
来完成相同的任务:
import os
import codecs
input_name = 'some_file.txt'
tmp_name = 'tmp.txt'
with codecs.open(input_name, 'r', encoding='latin-1') as fi, \
codecs.open(tmp_name, 'w', encoding='latin-1') as fo:
for line in fi:
new_line = do_processing(line) # do your line processing here
fo.write(new_line)
os.remove(input_name) # remove original
os.rename(tmp_name, input_name) # rename temp to original name
如果你想改变它,你也可以选择为输出文件指定一个新的编码,或者如果你不想要它,在打开输出文件时将它保留为 latin-1
它改变。
我知道这不是您正在寻找的就地修改,但它会完成您尝试执行的任务并且非常灵活。
关于python - 在 fileinput 模块中结合就地过滤和编码设置,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/25203040/