Home-IoT/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2024-11-12 18:27:46 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Fix type annotation on extract_registry_entry_config (#3623)

Browse source 
Co-authored-by: Jesse Hills <3060199+jesserockz@users.noreply.github.com>
This commit is contained in:
Quentin Smith 2022-10-11 22:15:03 -04:00 committed by GitHub
parent a8ff0a8913
commit 19900b004b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
esphome/cpp_helpers.py
View file

@ -13,7 +13,7 @@ from esphome.const import (
# pylint: disable=unused-import
from esphome.core import coroutine, ID, CORE
from esphome.types import ConfigType
from esphome.types import ConfigType, ConfigFragmentType
from esphome.cpp_generator import add, get_variable
from esphome.cpp_types import App
from esphome.util import Registry, RegistryEntry
@ -108,8 +108,9 @@ async def setup_entity(var, config):
def extract_registry_entry_config(
registry: Registry, full_config: ConfigType
) -> RegistryEntry:
registry: Registry,
full_config: ConfigType,
) -> tuple[RegistryEntry, ConfigFragmentType]:
key, config = next((k, v) for k, v in full_config.items() if k in registry)
return registry[key], config

2
esphome/util.py
View file

@ -34,7 +34,7 @@ class RegistryEntry:
return Schema(self.raw_schema)
class Registry(dict):
class Registry(dict[str, RegistryEntry]):
def __init__(self, base_schema=None, type_id_key=None):
super().__init__()
self.base_schema = base_schema or {}