evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add type hints to template result wrapper (#82575)

Browse source 
* Add type hints to template result wrapper

* Remove `ignore[call-arg]`

* Use tuple

* Alphabetise
This commit is contained in:
epenet 2022-11-24 08:41:14 +01:00 • committed by GitHub
parent 9dd1b85cbb
commit b525259878
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

12
homeassistant/helpers/template.py
View file

@ -170,10 +170,10 @@ class ResultWrapper:
render_result: str | None
def gen_result_wrapper(kls):
def gen_result_wrapper(kls: type[dict | list | set]) -> type:
"""Generate a result wrapper."""
class Wrapper(kls, ResultWrapper):
class Wrapper(kls, ResultWrapper): # type: ignore[valid-type,misc]
"""Wrapper of a kls that can store render_result."""
def __init__(self, *args: Any, render_result: str | None = None) -> None:
@ -186,7 +186,7 @@ def gen_result_wrapper(kls):
if kls is set:
return str(set(self))
return cast(str, kls.__str__(self))
return kls.__str__(self)
return self.render_result
@ -214,10 +214,8 @@ class TupleWrapper(tuple, ResultWrapper):
return self.render_result
RESULT_WRAPPERS: dict[type, type] = {
kls: gen_result_wrapper(kls) # type: ignore[no-untyped-call]
for kls in (list, dict, set)
}
_types: tuple[type[dict | list | set], ...] = (dict, list, set)
RESULT_WRAPPERS: dict[type, type] = {kls: gen_result_wrapper(kls) for kls in _types}
RESULT_WRAPPERS[tuple] = TupleWrapper