Question

我正在尝试创建一个替换.yml文件中单行的工具（作为文本文档打开）。我把它写入文件，但它将所有内容都写出来并写入“固定”行。我的代码如下，以及一个实例/文件的示例。文件名是重要的，无法更改。

import sys
from os import listdir

from os.path import isfile, join
onlyfiles = [ f for f in listdir() if isfile(f) and str(f)[-4:]==".yml" ] #Gets the file names, can be ANYTHING
for pos in range(1, len(onlyfiles)): #So it stops at the end of the List
    with open(onlyfiles[pos], 'r') as a: #Lets open it make sure it is good
        for line in a:
            if line == "  world: Trolling\n": # this is the target line to replace
                line = line.replace("  world: Trolling", "  world: world\n") #rework the line, though it wont write even if i change the R to W
                with open(onlyfiles[pos], 'w') as repl: #Writes to the file, but gets rid of everything else
                    repl.write(line)
                    print(line)

未修改文件的样本：

timestamps:
  login: 1373913158118
  lastteleport: 1373918169442
  logout: 1373918539235
  kits:
    trusted: 1373052268213
ipAddress: 142.165.45.129
nickname: Erin
lastlocation:
  world: Trolling
  x: -890.6999999880791
  y: 87.0
  z: -764.6999999880791
  yaw: 91.99242
  pitch: 75.449974
homes:
  e:
    world: Trolling
    x: 4653.2140183238
    y: 64.0
    z: 80.02726116652944
    yaw: 192.08363
    pitch: 66.29998

我试图用100多个文件替换“世界”的“Trolling”实例，而不涉及其他任何内容

Answer 1

在else之后，当该行与此模式不匹配时，您需要if line == " world: Trolling\n"条件。还要将您要写回的所有数据放回到某个变量中，并在关闭您正在读取的文件后最后一次写入，您无法同时读取和写入文件。 / p>

import sys
from os import listdir 
from os.path import isfile, join

onlyfiles = [ f for f in listdir() if isfile(f) and str(f)[-4:]==".yml" ]

for file in onlyfiles:
  wdata = ''
  with open(file, 'r') as a: #Lets open it make sure it is good
    for line in a: 
        if line == "  world: Trolling\n":
            line = "  world: world\n" # You already know what it should be
        print line
        wdata += line
  with open(file, 'w') as repl: 
    repl.write(wdata)

可能更容易做到

for file in onlyfiles:
  with open(file, 'r') as f:
    d = f.read().replace('\n  world: Trolling\n','\n  world: world\n')
  with open(file, 'w') as f:
    f.write(d)

Answer 2

我可能会寻求更通用的解决方案。

import yaml

def replace(doc, inval, outval):
    if isinstance(doc, dict):
        for k, v in doc.items():
            if v == inval:
                doc[k] = outval
            elif isinstance(v, dict):
                doc[k] = replace(v, inval, outval)
            elif isinstance(v, list):
                for i, item in enumerate(v[:]):
                    doc[k][i] = replace(item, inval, outval)
    return doc


doc = yaml.load(open("original.yml"))
newdoc = replace(doc, "Trolling", "world")
yaml.dump(newdoc, open("modified.yml", "w"))

替换.yml文件中的一行的程序（PYTHON）

2 个答案: