Home-IoT/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2024-12-22 21:44:55 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Speed up YAML by using YAML C loader when available (#5721)

Browse source
This commit is contained in:
J. Nick Koston 2023-11-10 03:17:40 -06:00 committed by GitHub
parent 0f19450ab4
commit 3b891bc146
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
esphome/yaml_util.py
View file

@ -23,6 +23,14 @@ 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
except ImportError:
from yaml import ( # type: ignore[assignment]
SafeLoader as FastestAvailableSafeLoader,
)
_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 +97,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