Python修改文件内容的几种实现方式与注意事项

基于字符串处理的文件修改

对于结构简单的配置文件，可以通过逐行读取并替换特定字符串的方式实现修改。这种方法逻辑直观，适用于不需要复杂模式匹配的场景。

data = ''
with open('zhai.conf', 'r+') as f:
  for line in f.readlines():
    if(line.find('Server') == 0):
      line = 'Server=%s' % ('192.168.1.1',) + '\n'
    data += line
with open('zhai.conf', 'r+') as f:
  f.writelines(data)

上述代码首先读取所有行，若行首包含“Server”，则将其替换为指定IP。随后重新打开文件并写入修改后的内容。需注意，此操作会清空原文件内容，若中途发生异常，可能导致数据丢失。

使用正则表达式进行替换

当需要匹配更复杂的模式时，re模块提供了更强大的功能。通过re.search判断是否匹配，并使用re.sub进行替换，可以处理多行文本中的特定内容。

#!/usr/local/bin/python
#coding:gbk
import re
old_file='/tmp/test'
fopen=open(old_file,'r')
w_str=""
for line in fopen:
  if re.search('hello',line):
    line=re.sub('hello','world',line)
    w_str+=line
  else:
    w_str+=line
print w_str
wopen=open(file_name,'w')
wopen.write(w_str)
fopen.close()
wopen.close()

该脚本遍历文件每一行，若包含“hello”则替换为“world”，并将结果写入新字符串。最后将完整内容写回文件。这种方式适合需要保留大部分原文，仅修改特定匹配项的场景。

封装为可复用脚本

为了便于日常使用，可以将文件替换逻辑封装为命令行脚本。以下示例支持通过参数指定旧文本、新文本及文件名，并可选是否生成备份文件。

#!/usr/bin/env python
#_*_ coding:utf-8 _*_
import sys,os
if len(sys.argv)<4 or len(sys.argv)>5:
  sys.exit('There needs four or five parameters')
elif len(sys.argv)==4:
  print 'usage:./file_replace.py old_text new_text filename'
else:
  print 'usage:./file_replace.py old_text new_text filename --bak'
old_text,new_text=sys.argv[1],sys.argv[2]
file_name=sys.argv[3]
f=file(file_name,'rb')
new_file=file('.%s.bak' % file_name,'wb')#文件名以.开头的文件是隐藏文件
for line in f.xreadlines():#f.xreadlines()返回一个文件迭代器，每次只从文件（硬盘）中读一行
  new_file.write(line.replace(old_text,new_text))
f.close()
new_file.close()
if '--bak' in sys.argv: #'--bak'表示要求对原文件备份
  os.rename(file_name,'%s.bak' % file_name)  #unchanged
  os.rename('.%s.bak' % file_name,file_name)  #changed
else:
  os.rename(file_name,'wahaha.txt')#此处也可以将原文件删除，以便下一语句能够正常执行
  os.rename('.%s.bak' % file_name,file_name)

脚本通过sys.argv接收参数。若指定--bak，则先重命名原文件为备份，再将处理后的隐藏备份文件重命名为原文件名。若不指定备份，则直接覆盖原文件。此方法通过临时文件避免了直接写入时的数据损坏风险，但需注意os.rename在不同操作系统下的行为差异。

单行命令与系统命令调用

对于简单的替换需求，可以使用Python的一行代码结合re模块完成：

open('file2', 'w').write(re.sub(r'world', 'python', open('file1').read()))

此外，Python提供了多种调用系统命令的方式，如os.system、os.popen以及更推荐的subprocess模块。

使用sed命令

在Linux环境下，sed是修改文件的常用工具。可以通过Python调用sed命令实现原地修改：

sed -i "/^Server/ c\Server=192.168.0.1" zhai.conf  #-i表示在原文修改
sed -ibak "/^Server/c\Server=192.168.0.1" zhai.conf  #会生成备份文件zhai.confbak

注意-i参数用于原地编辑，而-ibak会在修改前生成备份文件。

使用subprocess模块

subprocess模块提供了更灵活的进程控制能力，可以获取命令的输出和状态：

handle = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE)
print handle.communicate()[0]

通过Popen创建进程，并使用communicate方法获取标准输出。这种方式比os.system更安全，且能捕获命令执行结果。

总结与风险提示

修改文件内容时，直接覆盖写入存在数据丢失风险。建议采用以下策略：

• 使用临时文件保存修改结果，确认无误后再替换原文件。
• 对于重要配置文件，务必先备份原文件。
• 使用subprocess模块调用系统命令，避免使用已废弃的os.popen等函数。
• 注意编码问题，确保读写文件时使用正确的编码格式。

以上就是Python实现修改文件内容的方法分析的详细内容，更多关于Python文件操作与系统命令调用的资料请关注本站其它相关文章！