replace yaml parsed/dumper w/ simple stdlib impl

This commit is contained in:
AcerecA 2026-01-02 12:46:49 +01:00
parent aa93e8c712
commit a4efe1fd08
2 changed files with 76 additions and 10 deletions

View File

@ -7,6 +7,5 @@ requires-python = ">=3.12"
dependencies = [ dependencies = [
"pydantic-settings~=2.12", "pydantic-settings~=2.12",
"pynvim~=0.6.0", "pynvim~=0.6.0",
"ruamel.yaml~=0.18.17",
"requests>=2.32.5", "requests>=2.32.5",
] ]

View File

@ -1,12 +1,11 @@
from io import StringIO from json import loads
from ruamel.yaml import YAML from re import sub
from textwrap import dedent
from . import hints from . import hints
yaml = YAML(typ="safe")
def dump_scalar(entry: hints.JSONDataScalar) -> str:
def dump_scalar(entry: JSONDataScalar) -> str:
match entry: match entry:
case None: case None:
return "null\n" return "null\n"
@ -18,7 +17,7 @@ def dump_scalar(entry: JSONDataScalar) -> str:
return f"{entry:s}\n" return f"{entry:s}\n"
def dump(obj: dict[str, JSONDataScalar | list[JSONDataScalar]]) -> str: def dump(obj: dict[str, hints.JSONDataScalar | list[hints.JSONDataScalar]]) -> str:
ret = "" ret = ""
for key, value in obj.items(): for key, value in obj.items():
ret += key ret += key
@ -35,6 +34,74 @@ def dump(obj: dict[str, JSONDataScalar | list[JSONDataScalar]]) -> str:
return ret return ret
def load(content: str) -> JSONData: def load(content: str) -> hints.JSONData:
with StringIO(content) as sio: """parse subset of yaml into ``JSONData``
return yaml.load(sio) # pyright: ignore[reportAny, reportUnknownMemberType]
>>> from pprint import pprint
>>> yaml = '''
... key1: str
... key2: "str"
... key3: 1
... key4: 2.3
... key5: 2e12
... key6: null
... key7: true
... key8: false
... key9:
... - 1
... - 23.2
... key10:
... - str
... - "str"
... '''
>>> yaml_parsed = load(yaml)
>>> compare = {
... 'key1': 'str',
... 'key2': 'str',
... 'key3': 1,
... 'key4': 2.3,
... 'key5': 2e12,
... 'key6': None,
... 'key7': True,
... 'key8': False,
... 'key9': [1, 23.2],
... 'key10': ['str', 'str'],
... }
>>> {k: yaml_parsed.get(k, None) for k in compare if compare[k] != yaml_parsed[k]}
{}
"""
ret = {}
key = None
value = None
for line in sub(r":[\s\n]*", ":\n ", dedent(content).strip()).splitlines():
if line.endswith(":"):
if key:
ret[key] = value
value = None
key = line.removesuffix(":")
elif line.startswith(" -"):
value = value or []
try:
parsed = loads(line.removeprefix(" -").strip())
except:
parsed = loads('"' + line.removeprefix(" -").strip() + '"')
value.append(parsed)
else:
try:
value = loads(line.strip())
except:
value = loads('"' + line.strip() + '"')
if key:
ret[key] = value
value = None
return ret
if __name__ == "__main__":
import doctest
doctest.testmod()