jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-26 22:57:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Enforce flow-handler result type hint for step_* (#72834)

Browse Source
This commit is contained in:
epenet 2022-06-28 10:08:36 +02:00 committed by GitHub
parent 35df012b6e
commit c19a8ef8e0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pylint/plugins/hass_enforce_type_hints.py
View File

@ -37,6 +37,8 @@ class TypeHintMatch:
self.function_name == node.name self.function_name == node.name
or self.has_async_counterpart or self.has_async_counterpart
and node.name == f"async_{self.function_name}" and node.name == f"async_{self.function_name}"
or self.function_name.endswith("*")
and node.name.startswith(self.function_name[:-1])
) )
@ -370,6 +372,16 @@ _FUNCTION_MATCH: dict[str, list[TypeHintMatch]] = {
_CLASS_MATCH: dict[str, list[ClassTypeHintMatch]] = { _CLASS_MATCH: dict[str, list[ClassTypeHintMatch]] = {
"config_flow": [ "config_flow": [
ClassTypeHintMatch(
base_class="FlowHandler",
matches=[
TypeHintMatch(
function_name="async_step_*",
arg_types={},
return_type="FlowResult",
),
],
),
ClassTypeHintMatch( ClassTypeHintMatch(
base_class="ConfigFlow", base_class="ConfigFlow",
matches=[ matches=[