sascha/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2025-02-16 01:48:14 +00:00
Code Issues Releases Wiki Activity

Use YAML C loader when available

Browse Source 
This is borrowed from HA core
527a3dba9c/homeassistant/util/yaml/loader.py (L14)
527a3dba9c/homeassistant/util/yaml/dumper.py (L12)
This commit is contained in:
J. Nick Koston 2023-11-09 16:57:48 -06:00
parent bc7519f645
commit 037f593363
No known key found for this signature in database
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
esphome/yaml_util.py
View File

@ -23,6 +23,18 @@ from esphome.core import (
from esphome.helpers import add_class_to_obj from esphome.helpers import add_class_to_obj
from esphome.util import OrderedDict, filter_yaml_files from esphome.util import OrderedDict, filter_yaml_files
try:
from yaml import CSafeLoader as FastestAvailableSafeLoader
from yaml import CSafeDumper as FastestAvailableSafeDumper
except ImportError:
from yaml import ( # type: ignore[assignment]
SafeLoader as FastestAvailableSafeLoader,
)
from yaml import ( # type: ignore[assignment]
SafeDumper as FastestAvailableSafeDumper,
)
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
# Mostly copied from Home Assistant because that code works fine and # Mostly copied from Home Assistant because that code works fine and
@ -89,7 +101,7 @@ def _add_data_ref(fn):
return wrapped return wrapped
class ESPHomeLoader(yaml.SafeLoader): class ESPHomeLoader(FastestAvailableSafeLoader):
"""Loader class that keeps track of line numbers.""" """Loader class that keeps track of line numbers."""
@_add_data_ref @_add_data_ref
@ -439,7 +451,7 @@ def is_secret(value):
return None return None
class ESPHomeDumper(yaml.SafeDumper): class ESPHomeDumper(FastestAvailableSafeDumper):
def represent_mapping(self, tag, mapping, flow_style=None): def represent_mapping(self, tag, mapping, flow_style=None):
value = [] value = []
node = yaml.MappingNode(tag, value, flow_style=flow_style) node = yaml.MappingNode(tag, value, flow_style=flow_style)