使用 Python 编辑文本文件

Question

每当我的 IP 地址发生变化时，我都需要更新一个文本文件，然后从 shell 运行一些命令。

1. 创建变量 LASTKNOWN = "212.171.135.53" 这是我们编写此脚本时的 IP 地址。

2. 获取当前 IP 地址。每天都会变化。

3. 为新 IP 创建变量 CURRENT。

4. 比较（作为字符串）CURRENT 和 LASTKNOWN

5. 如果相同，则退出()

6. 如果它们不同，

   A.将包含 LASTKNOWN IP 地址的旧配置文件（/etc/ipf.conf）“复制”到 /tmp B. 将 /tmp/ipf.conf 文件中的 LASTKNOWN 替换为 CURRENT。
   C. 使用子进程“mv /tmp/ipf.conf /etc/ipf.conf”
   D. 使用子进程执行，“ipf -Fa -f /etc/ipf.conf”
   E. 使用子进程执行，“ipnat -CF -f /etc/ipnat.conf”

7. 退出()

我知道如何执行第 1 步到第 6 步。我陷入了“文件编辑”部分，A -> C。我不知道要使用哪个模块或是否应该在适当的位置编辑文件。有很多方法可以做到这一点，我无法决定最好的方法。我想我想要最保守的。

我知道如何使用子流程，因此您无需对此发表评论。

我不想替换整行；只是一个特定的虚线四边形。

谢谢！

Answer 1

另一种简单地编辑文件的方法是使用fileinput 模块：

import fileinput, sys
for line in fileinput.input(["test.txt"], inplace=True):
    line = line.replace("car", "truck")
    # sys.stdout is redirected to the file
    sys.stdout.write(line)

Answer 2

将 /etc/ipf.conf 中的 LASTKNOWN 替换为 CURRENT

一次性全部更换

filename = "/etc/ipf.conf"
text = open(filename).read()
open(filename, "w").write(text.replace(LASTKNOWN, CURRENT))

逐行替换

from __future__ import with_statement
from contextlib import nested

in_filename, outfilename = "/etc/ipf.conf", "/tmp/ipf.conf"
with nested(open(in_filename), open(outfilename, "w")) as in_, out:
     for line in in_:
         out.write(line.replace(LASTKNOWN, CURRENT))
os.rename(outfilename, in_filename)

注意：“/tmp/ipf.conf”应替换为tempfile.NamedTemporaryFile()或类似
注意：代码未经测试。

Answer 3

fileinput 模块的 API 非常丑陋，我为这项任务找到了漂亮的模块 - in_place，Python 3 的示例：

import in_place

with in_place.InPlace('data.txt') as file:
    for line in file:
        line = line.replace('test', 'testZ')
        file.write(line)

与文件输入的主要区别：

• 不是劫持 sys.stdout，而是返回一个新的文件句柄以供写入。
• 文件句柄支持所有标准 I/O 方法，而不仅仅是 readline()。

例如 - fileinput 只能逐行编辑，in_pace 允许将整个文件读取到内存（如果它不大）并修改它。

Answer 4

您正在尝试“原子地”更新文件的内容，并且在该主题上发生了许多令人愉快的激烈争吵。但一般模式是：

1) 将新文件写入临时文件，并确保刷新并关闭。

2) 使用操作系统的工具将临时文件自动重命名为旧文件。

现在，您根本无法在 Windows 中以原子方式重命名文件，但听起来您无论如何都在一个类似 unix 的系统上。你使用os.rename()自动重命名。

Answer 5

可能最简单的方法是使用 f=open(filename, mode) 打开文件。然后，使用 f.readlines() 读取所有行（这将返回代表程序行的字符串列表）。

然后您可以搜索这些字符串以找到地址并将其替换为新地址（使用标准字符串替换、正则表达式或任何您想要的）。

最后，您可以使用 f.writelines(lines) 将行写回文件，它可以方便地取回行列表。

注意：这不是一种有效的方法，它只是最简单的方法。请

示例代码：

f = open(filename, "r")
lines = f.readlines()

# Assume that change_ip is a function that takes a string and returns a new one with the ip changed): example below
ret_lines = [change_ip(lines) for line in lines]
new_file = open(new_filename, "w")
new_file.writelines(lines)

def change_ip(str):
   ''' Gets a string, returns a new string where the ip is changed '''
   # Add implementation, something like: return str.replace(old_ip, new_ip) or something similair.

Answer 6

在检查了 cmets 和您放在 pastebin 上的代码后，这是一个可行的解决方案。首先，文件 /tmp/iiiipf.conf 包含：

Simply a test file 175.48.204.168

And two times 175.48.204.168 on this line 175.48.204.168

Done.

运行代码后，文件/tmp/iiiipf.conf包含：

Simply a test file 10.73.144.112

And two times 10.73.144.112 on this line 10.73.144.112

Done.

这是经过测试的工作代码，我的东西合并到你的 pastebin 代码中：

import socket
import fileinput
import subprocess
import string
import re

CURRENT = socket.getaddrinfo(socket.gethostname(), None)[0][4][0]
LASTKNOWN = '175.48.204.168'

if CURRENT == LASTKNOWN:
    print 'Nevermind.'
    subprocess.sys.exit()

else:

    cf = open("/tmp/iiiipf.conf", "r")
    lns = cf.readlines()
    # close it so that we can open for writing later
    cf.close()

    # assumes LASTKNOWN and CURRENT are strings with dotted notation IP addresses
    lns = "".join(lns)
    lns = re.sub(LASTKNOWN, CURRENT, lns)  # This replaces all occurences of LASTKNOWN with CURRENT

    cf = open("/tmp/iiiipf.conf", "w")
    cf.write(lns)
    cf.close()

即使在配置文件中多次使用 IP 地址，这段代码也能满足您的需求。它还会在注释行中更改它。

此方法不需要复制到 /tmp，并且在重启防火墙和 NAT 时使用更少的子进程调用。