Python yaml генерирует несколько значений в кавычках - PullRequest
Новая
       21

Python yaml генерирует несколько значений в кавычках

0 голосов
/ 20 сентября 2018

Я использую модуль yaml в своем скрипте Python для создания файла YAML.Ниже приведен пример: 

import yaml
class MyDumper(yaml.Dumper):

    def increase_indent(self, flow=False, indentless=False):
        return super(MyDumper, self).increase_indent(flow, False)

foo = {
    'instance_type': 'test',
    'hostname': "\"testhost\"",
    'name': 'foo',
    'my_list': [
        {'foo': 'test', 'bar': 'test2'},
        {'foo': 'test3', 'bar': 'test4'}],
    'hello': 'world',
}

print yaml.dump(foo, Dumper=MyDumper, default_flow_style=False)

Вывод: 

hello: world
hostname: '"testhost"'
instance_type: test
my_list:
  - bar: test2
    foo: test
  - bar: test4
    foo: test3
name: foo

В приведенном выше выводе имя хоста имеет одинарные и двойные кавычки, я хочу только двойные кавычки.

Ожидаетсявывод: 

hello: world
hostname: "testhost"
instance_type: test
my_list:
  - bar: test2
    foo: test
  - bar: test4
    foo: test3
name: foo

Ответы [ 3 ]

0 голосов
/ 20 сентября 2018

Вы не можете форсировать кавычки в YAML, цитируя части ваших данных, как вы.Поскольку кавычки заставляют дампер применять к скаляру кавычки (то есть больше не могут использовать простые скаляры, как для других строковых значений в вашем файле yaml).

Вам необходимо создать тип, который будет сбрасываться с кавычками.Проще всего это сделать с помощью ruamel.yaml (отказ от ответственности: я являюсь автором этой улучшенной версии PyYAML, поддерживающей YAML 1.2, поддерживающей сохранение комментариев и цитат и т. Д.), 

import sys
import ruamel.yaml
from ruamel.yaml.scalarstring import DoubleQuotedScalarString as dq


yaml = ruamel.yaml.YAML()
yaml.indent(sequence=4, offset=2)

foo = {
    'instance_type': 'test',
    'hostname': dq("testhost"),
    'name': 'foo',
    'my_list': [
        {'foo': 'test', 'bar': 'test2'},
        {'foo': 'test3', 'bar': 'test4'}],
    'hello': 'world',
}


yaml.dump(foo, sys.stdout)

, которая дает: 

instance_type: test
hostname: "testhost"
name: foo
my_list:
  - foo: test
    bar: test2
  - foo: test3
    bar: test4
hello: world

Вы также можете легко загрузить и вывести этот вывод, генерируя точно такой же вывод: 

from ruamel.yaml.compat import StringIO

buf = StringIO()
yaml.dump(foo, buf)

yaml.preserve_quotes = True
data = yaml.load(buf.getvalue())
yaml.dump(data, sys.stdout)
0 голосов
/ 20 сентября 2018

Если вы настаиваете на том, чтобы делать это через PyYAML, вы можете объявить свой собственный тип принудительной цитаты и добавить его представитель: 

import yaml

class MyDumper(yaml.Dumper):  # your force-indent dumper

    def increase_indent(self, flow=False, indentless=False):
        return super(MyDumper, self).increase_indent(flow, False)

class QuotedString(str):  # just subclass the built-in str
    pass

def quoted_scalar(dumper, data):  # a representer to force quotations on scalars
    return dumper.represent_scalar('tag:yaml.org,2002:str', data, style='"')

# add the QuotedString custom type with a forced quotation representer to your dumper
MyDumper.add_representer(QuotedString, quoted_scalar)

foo = {
    'instance_type': 'test',
    'hostname': QuotedString('testhost'),  # here's the 'magic'
    'name': 'foo',
    'my_list': [
        {'foo': 'test', 'bar': 'test2'},
        {'foo': 'test3', 'bar': 'test4'}],
    'hello': 'world',
}

print(yaml.dump(foo, Dumper=MyDumper, default_flow_style=False))

, который даст вам: 

hello: world
hostname: "testhost"
instance_type: test
my_list:
  - bar: test2
    foo: test
  - bar: test4
    foo: test3
name: foo

Отказ от ответственности: Учитывая выбор, я также предпочитаю модуль Anthon ruamel.yaml для своих потребностей в YAML.

0 голосов
/ 20 сентября 2018

Вы получаете двойные кавычки, потому что это то, что есть у ваших входных данных.Эта строка: 

'hostname': "\"testhost\"",

говорит о том, что вы хотите, чтобы hosthame имел в качестве значения строку из 10 символов, начинающуюся и заканчивающуюся ", и это то, что вы видите в yaml.Эта строка с двойными кавычками "\"testhost\"" и версией yaml '"testhost"' представляют собой два разных представления исходного кода одних и тех же данных.Вам нужно только двойные кавычки вокруг строки в yaml, если вы хотите вставить в нее специальные символы (например, \n для новой строки).Но yaml.dump() позаботится об этом за вас.

...