如何解析和附加到flexget yaml配置文件?

时间:2014-04-09 19:07:42

标签: python linux configuration yaml

我正在尝试解析并从flexget配置文件中添加/删除节目。

我最终计划在节目名称旁边创建一个带有复选框的小网页并添加/删除它们,但首先我需要在我拥有此代码的时刻完成yaml解析:

#!/usr/bin/env python
import yaml
with open("shows.yml") as f:
        doc = yaml.safe_load(f)

shows = doc['series']['shows']
sets = doc['series']['settings']
shows.append('new show')
# this makes the list a little neater while printed to the file
showsa = []
for a in test:
  showsa.append(a)

testa = {'series': { 'shows': showsa, 'settings': sets }}

with open("showsnew.yml", 'w') as g:
        g.write( yaml.dump(testa, default_flow_style=False) )

这会打开旧配置向列表中添加一个新节目,然后将其打印到文件中,它几乎是完美的,除了输出的配置有点混乱:

**What I should get is:**                    **But instead I get:**
shows:                                         shows:
  settings:                                      settings:
    shows:                                         shows:
      setting1: value                                setting1: value
      setting2:                                      setting2:
        - value                                      - value
        - value                                      - value
  shows:                                         shows:
    - show1                                      - show1
    - show2                                      - show2
    - new show                                   - new show

虽然它不是一个巨大的差异(只是' - '后面有2个空格的行),我认为它可能会在稍后阶段弄乱配置。

我知道它很简单,可以手动添加和删除一个节目,但是当我使用数据库获取其他节目信息时,能够将节目添加到flexget和数据库以及其他一些任务中会很方便。

有谁知道我做错了什么小事?或者这只是python中的pyYaml如何工作?

目前我正在考虑对文件进行解析并为每行以“ - ”符号添加2个空格的可能性。或者将每个节目逐行写入文件(或手工字符串),并在开始时使用正确的间距。

但我认为必须有更好的方法。

2 个答案:

答案 0 :(得分:1)

我现在偶然发现了同样的问题 我想出了这个:

def writeYaml(data, filename):

  sections = data.keys()
  dump = yaml.dump(data, default_flow_style=False)

  # set 2 indent spaces in every line
  out = dump.replace('\n', '\n  ')
  # re-set indentation for sections only
  for i in sections:
      out = out.replace('\n  %s' % i, '\n%s' % i)

  with open(filename, 'w') as yaml_file:
      yaml_file.write(out)

答案 1 :(得分:0)

我继续并实现了一个简单的函数来解析给出的文件,以便为受影响的行添加额外的2个空格,因为它需要不到一秒的时间才能完成并且是不可察觉的。