1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
# Copyright (C) 2023 Collabora Limited
#
# Author: Igor Ponomarev <igor.ponomarev@collabora.com>
#
# SPDX-License-Identifier: GPL-2.0-or-later
from __future__ import annotations
from contextvars import ContextVar
from io import StringIO
from typing import TYPE_CHECKING
from yaml import dump, load
if TYPE_CHECKING:
from typing import Union # For Python 3.9 compatibility
from yaml import safe_dump, safe_load
yaml_safe_dump = safe_dump
yaml_safe_load = safe_load
else:
# Handle compatibility with system without C yaml
try:
from yaml import CSafeLoader as SafeLoader
except ImportError:
from yaml import SafeLoader
try:
from yaml import CSafeDumper as SafeDumper
except ImportError:
from yaml import SafeDumper
def yaml_safe_load(data):
return load(data, Loader=SafeLoader)
def yaml_safe_dump(data, *args, **kwargs):
# Preserve key order by default
kwargs["sort_keys"] = kwargs.get("sort_keys", False)
return dump(data, *args, Dumper=SafeDumper, **kwargs)
yaml_quote_dumper: ContextVar[tuple[StringIO, SafeDumper]] = ContextVar(
"yaml_quote_dumper"
)
def _get_largest_width_possible() -> int:
# A largest value that both C and Python implementation accept
from ctypes import c_uint
return c_uint(-1).value // 2
def yaml_quote(obj: Union[str, int, float, dict, list]) -> str:
try:
stream, yaml_dumper = yaml_quote_dumper.get()
except LookupError:
stream = StringIO()
yaml_dumper = SafeDumper(
stream,
default_flow_style=True, # Output in a single line
width=_get_largest_width_possible(), # Disable automatic line breaks
default_style='"', # Use "double quoted" scalar type
explicit_start=True, # Always output "--- " YAML document start
explicit_end=False, # Do not output "\n..." at the end
)
yaml_dumper.open() # Start accepting any objects
yaml_quote_dumper.set((stream, yaml_dumper))
else:
# Clean-up existing stream
stream.seek(0)
stream.truncate(0)
yaml_dumper.represent(obj)
return stream.getvalue()[4:-1] # Skip "--- " and newline
__all__ = (
"yaml_quote",
"yaml_safe_dump",
"yaml_safe_load",
)
|
