You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
90 lines
3.2 KiB
90 lines
3.2 KiB
# Copyright (c) 2006-2019 Andrey Golovigin
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining
|
|
# a copy of this software and associated documentation files (the
|
|
# "Software"), to deal in the Software without restriction, including
|
|
# without limitation the rights to use, copy, modify, merge, publish,
|
|
# distribute, sublicense, and/or sell copies of the Software, and to
|
|
# permit persons to whom the Software is furnished to do so, subject to
|
|
# the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be
|
|
# included in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
|
# IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
|
# CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
|
# TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
|
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
from collections import OrderedDict
|
|
|
|
import yaml
|
|
from pybtex.database.output import BaseWriter
|
|
|
|
|
|
class OrderedDictSafeDumper(yaml.SafeDumper):
|
|
"""
|
|
SafeDumper that dumps OrderedDicts preserving the order.
|
|
"""
|
|
def represent_odict(self, data):
|
|
return self.represent_mapping(u'tag:yaml.org,2002:map', data.items())
|
|
|
|
OrderedDictSafeDumper.add_representer(
|
|
OrderedDict, OrderedDictSafeDumper.represent_odict
|
|
)
|
|
|
|
|
|
class Writer(BaseWriter):
|
|
"""Outputs YAML markup"""
|
|
|
|
def _to_dict(self, bib_data):
|
|
def process_person_roles(entry):
|
|
for role, persons in entry.persons.items():
|
|
yield role, list(process_persons(persons))
|
|
|
|
def process_person(person):
|
|
for type in ('first', 'middle', 'prelast', 'last', 'lineage'):
|
|
name = person.get_part_as_text(type)
|
|
if name:
|
|
yield type, name
|
|
|
|
def process_persons(persons):
|
|
for person in persons:
|
|
yield OrderedDict(process_person(person))
|
|
|
|
def process_entries(bib_data):
|
|
for key, entry in bib_data.items():
|
|
fields = OrderedDict([('type', entry.original_type)])
|
|
fields.update(entry.fields)
|
|
fields.update(process_person_roles(entry))
|
|
yield key, fields
|
|
|
|
data = {'entries': OrderedDict(process_entries(bib_data.entries))}
|
|
if bib_data.preamble:
|
|
data['preamble'] = bib_data.preamble
|
|
return data
|
|
|
|
def _dump(self, bib_data, encoding=None, stream=None):
|
|
return yaml.dump(
|
|
bib_data,
|
|
stream,
|
|
encoding=encoding,
|
|
allow_unicode=True,
|
|
default_flow_style=False,
|
|
indent=4,
|
|
Dumper=OrderedDictSafeDumper,
|
|
)
|
|
|
|
def write_stream(self, bib_data, stream):
|
|
return self._dump(self._to_dict(bib_data), encoding='UTF-8', stream=stream)
|
|
|
|
def to_string(self, bib_data):
|
|
return self._dump(self._to_dict(bib_data), encoding=None)
|
|
|
|
def to_bytes(self, bib_data):
|
|
return self._dump(self._to_dict(bib_data), encoding='UTF-8')
|
|
|