193 lines
6.4 KiB
Python
193 lines
6.4 KiB
Python
|
# coding: utf-8
|
||
|
|
||
|
from __future__ import absolute_import
|
||
|
|
||
|
from _ruamel_yaml import CParser, CEmitter # type: ignore
|
||
|
|
||
|
from .constructor import Constructor, BaseConstructor, SafeConstructor
|
||
|
from .representer import Representer, SafeRepresenter, BaseRepresenter
|
||
|
from .resolver import Resolver, BaseResolver
|
||
|
|
||
|
if False: # MYPY
|
||
|
from typing import Any, Union, Optional # NOQA
|
||
|
from .compat import StreamTextType, StreamType, VersionType # NOQA
|
||
|
|
||
|
__all__ = [
|
||
|
"CBaseLoader",
|
||
|
"CSafeLoader",
|
||
|
"CLoader",
|
||
|
"CBaseDumper",
|
||
|
"CSafeDumper",
|
||
|
"CDumper",
|
||
|
]
|
||
|
|
||
|
|
||
|
# this includes some hacks to solve the usage of resolver by lower level
|
||
|
# parts of the parser
|
||
|
|
||
|
|
||
|
class CBaseLoader(CParser, BaseConstructor, BaseResolver): # type: ignore
|
||
|
def __init__(self, stream, version=None, preserve_quotes=None):
|
||
|
# type: (StreamTextType, Optional[VersionType], Optional[bool]) -> None
|
||
|
CParser.__init__(self, stream)
|
||
|
self._parser = self._composer = self
|
||
|
BaseConstructor.__init__(self, loader=self)
|
||
|
BaseResolver.__init__(self, loadumper=self)
|
||
|
# self.descend_resolver = self._resolver.descend_resolver
|
||
|
# self.ascend_resolver = self._resolver.ascend_resolver
|
||
|
# self.resolve = self._resolver.resolve
|
||
|
|
||
|
|
||
|
class CSafeLoader(CParser, SafeConstructor, Resolver): # type: ignore
|
||
|
def __init__(self, stream, version=None, preserve_quotes=None):
|
||
|
# type: (StreamTextType, Optional[VersionType], Optional[bool]) -> None
|
||
|
CParser.__init__(self, stream)
|
||
|
self._parser = self._composer = self
|
||
|
SafeConstructor.__init__(self, loader=self)
|
||
|
Resolver.__init__(self, loadumper=self)
|
||
|
# self.descend_resolver = self._resolver.descend_resolver
|
||
|
# self.ascend_resolver = self._resolver.ascend_resolver
|
||
|
# self.resolve = self._resolver.resolve
|
||
|
|
||
|
|
||
|
class CLoader(CParser, Constructor, Resolver): # type: ignore
|
||
|
def __init__(self, stream, version=None, preserve_quotes=None):
|
||
|
# type: (StreamTextType, Optional[VersionType], Optional[bool]) -> None
|
||
|
CParser.__init__(self, stream)
|
||
|
self._parser = self._composer = self
|
||
|
Constructor.__init__(self, loader=self)
|
||
|
Resolver.__init__(self, loadumper=self)
|
||
|
# self.descend_resolver = self._resolver.descend_resolver
|
||
|
# self.ascend_resolver = self._resolver.ascend_resolver
|
||
|
# self.resolve = self._resolver.resolve
|
||
|
|
||
|
|
||
|
class CBaseDumper(CEmitter, BaseRepresenter, BaseResolver): # type: ignore
|
||
|
def __init__(
|
||
|
self,
|
||
|
stream,
|
||
|
default_style=None,
|
||
|
default_flow_style=None,
|
||
|
canonical=None,
|
||
|
indent=None,
|
||
|
width=None,
|
||
|
allow_unicode=None,
|
||
|
line_break=None,
|
||
|
encoding=None,
|
||
|
explicit_start=None,
|
||
|
explicit_end=None,
|
||
|
version=None,
|
||
|
tags=None,
|
||
|
block_seq_indent=None,
|
||
|
top_level_colon_align=None,
|
||
|
prefix_colon=None,
|
||
|
):
|
||
|
# type: (StreamType, Any, Any, Any, Optional[bool], Optional[int], Optional[int], Optional[bool], Any, Any, Optional[bool], Optional[bool], Any, Any, Any, Any, Any) -> None # NOQA
|
||
|
CEmitter.__init__(
|
||
|
self,
|
||
|
stream,
|
||
|
canonical=canonical,
|
||
|
indent=indent,
|
||
|
width=width,
|
||
|
encoding=encoding,
|
||
|
allow_unicode=allow_unicode,
|
||
|
line_break=line_break,
|
||
|
explicit_start=explicit_start,
|
||
|
explicit_end=explicit_end,
|
||
|
version=version,
|
||
|
tags=tags,
|
||
|
)
|
||
|
self._emitter = self._serializer = self._representer = self
|
||
|
BaseRepresenter.__init__(
|
||
|
self,
|
||
|
default_style=default_style,
|
||
|
default_flow_style=default_flow_style,
|
||
|
dumper=self,
|
||
|
)
|
||
|
BaseResolver.__init__(self, loadumper=self)
|
||
|
|
||
|
|
||
|
class CSafeDumper(CEmitter, SafeRepresenter, Resolver): # type: ignore
|
||
|
def __init__(
|
||
|
self,
|
||
|
stream,
|
||
|
default_style=None,
|
||
|
default_flow_style=None,
|
||
|
canonical=None,
|
||
|
indent=None,
|
||
|
width=None,
|
||
|
allow_unicode=None,
|
||
|
line_break=None,
|
||
|
encoding=None,
|
||
|
explicit_start=None,
|
||
|
explicit_end=None,
|
||
|
version=None,
|
||
|
tags=None,
|
||
|
block_seq_indent=None,
|
||
|
top_level_colon_align=None,
|
||
|
prefix_colon=None,
|
||
|
):
|
||
|
# type: (StreamType, Any, Any, Any, Optional[bool], Optional[int], Optional[int], Optional[bool], Any, Any, Optional[bool], Optional[bool], Any, Any, Any, Any, Any) -> None # NOQA
|
||
|
self._emitter = self._serializer = self._representer = self
|
||
|
CEmitter.__init__(
|
||
|
self,
|
||
|
stream,
|
||
|
canonical=canonical,
|
||
|
indent=indent,
|
||
|
width=width,
|
||
|
encoding=encoding,
|
||
|
allow_unicode=allow_unicode,
|
||
|
line_break=line_break,
|
||
|
explicit_start=explicit_start,
|
||
|
explicit_end=explicit_end,
|
||
|
version=version,
|
||
|
tags=tags,
|
||
|
)
|
||
|
self._emitter = self._serializer = self._representer = self
|
||
|
SafeRepresenter.__init__(
|
||
|
self, default_style=default_style, default_flow_style=default_flow_style
|
||
|
)
|
||
|
Resolver.__init__(self)
|
||
|
|
||
|
|
||
|
class CDumper(CEmitter, Representer, Resolver): # type: ignore
|
||
|
def __init__(
|
||
|
self,
|
||
|
stream,
|
||
|
default_style=None,
|
||
|
default_flow_style=None,
|
||
|
canonical=None,
|
||
|
indent=None,
|
||
|
width=None,
|
||
|
allow_unicode=None,
|
||
|
line_break=None,
|
||
|
encoding=None,
|
||
|
explicit_start=None,
|
||
|
explicit_end=None,
|
||
|
version=None,
|
||
|
tags=None,
|
||
|
block_seq_indent=None,
|
||
|
top_level_colon_align=None,
|
||
|
prefix_colon=None,
|
||
|
):
|
||
|
# type: (StreamType, Any, Any, Any, Optional[bool], Optional[int], Optional[int], Optional[bool], Any, Any, Optional[bool], Optional[bool], Any, Any, Any, Any, Any) -> None # NOQA
|
||
|
CEmitter.__init__(
|
||
|
self,
|
||
|
stream,
|
||
|
canonical=canonical,
|
||
|
indent=indent,
|
||
|
width=width,
|
||
|
encoding=encoding,
|
||
|
allow_unicode=allow_unicode,
|
||
|
line_break=line_break,
|
||
|
explicit_start=explicit_start,
|
||
|
explicit_end=explicit_end,
|
||
|
version=version,
|
||
|
tags=tags,
|
||
|
)
|
||
|
self._emitter = self._serializer = self._representer = self
|
||
|
Representer.__init__(
|
||
|
self, default_style=default_style, default_flow_style=default_flow_style
|
||
|
)
|
||
|
Resolver.__init__(self)
|