1

我正在尝试制作一个工具来替换 .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”实例，而不涉及其他任何内容

4

2 回答 2

2

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

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)
于 2013-07-22T00:36:00.033 回答
0

I would probably go for a more general solution. 

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"))
于 2013-07-22T04:08:02.627 回答