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

Ask for IoT class during scaffold (#49647)

Browse Source 
Co-authored-by: Milan Meulemans <milan.meulemans@live.be>
Co-authored-by: Franck Nijhof <git@frenck.dev>
This commit is contained in:
Paulus Schoutsen 2021-04-25 03:13:22 -07:00 committed by GitHub
parent b92f29997e
commit 9f8e683ae3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 23 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
script/scaffold/__main__.py
View File

@ -99,7 +99,7 @@ def main():
if args.develop: if args.develop:
print("Running tests") print("Running tests")
print(f"$ pytest -vvv tests/components/{info.domain}") print(f"$ pytest -vvv tests/components/{info.domain}")
subprocess.run(["pytest", "-vvv", "tests/components/{info.domain}"]) subprocess.run(["pytest", "-vvv", f"tests/components/{info.domain}"])
print() print()
docs.print_relevant_docs(args.template, info) docs.print_relevant_docs(args.template, info)

18
script/scaffold/gather_info.py
View File

@ -2,6 +2,7 @@
import json import json
from homeassistant.util import slugify from homeassistant.util import slugify
from script.hassfest.manifest import SUPPORTED_IOT_CLASSES
from .const import COMPONENT_DIR from .const import COMPONENT_DIR
from .error import ExitApp from .error import ExitApp
@ -46,6 +47,7 @@ def gather_info(arguments) -> Info:
"codeowner": "@developer", "codeowner": "@developer",
"requirement": "aiodevelop==1.2.3", "requirement": "aiodevelop==1.2.3",
"oauth2": True, "oauth2": True,
"iot_class": "local_polling",
} }
) )
else: else:
@ -86,6 +88,22 @@ def gather_new_integration(determine_auth: bool) -> Info:
] ]
], ],
}, },
"iot_class": {
"prompt": (
f"""How will your integration gather data?
Valid values are {', '.join(SUPPORTED_IOT_CLASSES)}
More info @ https://developers.home-assistant.io/docs/creating_integration_manifest#iot-class
"""
),
"validators": [
[
f"You need to pick one of {', '.join(SUPPORTED_IOT_CLASSES)}",
lambda value: value in SUPPORTED_IOT_CLASSES,
]
],
},
} }
if determine_auth: if determine_auth:

4
script/scaffold/generate.py
View File

@ -67,10 +67,10 @@ def _append(path: Path, text):
path.write_text(path.read_text() + text) path.write_text(path.read_text() + text)
def _custom_tasks(template, info) -> None: def _custom_tasks(template, info: Info) -> None:
"""Handle custom tasks for templates.""" """Handle custom tasks for templates."""
if template == "integration": if template == "integration":
changes = {"codeowners": [info.codeowner]} changes = {"codeowners": [info.codeowner], "iot_class": info.iot_class}
if info.requirement: if info.requirement:
changes["requirements"] = [info.requirement] changes["requirements"] = [info.requirement]

1
script/scaffold/model.py
View File

@ -18,6 +18,7 @@ class Info:
is_new: bool = attr.ib() is_new: bool = attr.ib()
codeowner: str = attr.ib(default=None) codeowner: str = attr.ib(default=None)
requirement: str = attr.ib(default=None) requirement: str = attr.ib(default=None)
iot_class: str = attr.ib(default=None)
authentication: str = attr.ib(default=None) authentication: str = attr.ib(default=None)
discoverable: str = attr.ib(default=None) discoverable: str = attr.ib(default=None)
oauth2: str = attr.ib(default=None) oauth2: str = attr.ib(default=None)

2
script/scaffold/templates/config_flow_discovery/integration/__init__.py
View File

@ -10,7 +10,7 @@ from .const import DOMAIN
# TODO List the platforms that you want to support. # TODO List the platforms that you want to support.
# For your initial PR, limit it to 1 platform. # For your initial PR, limit it to 1 platform.
PLATFORMS = ["light"] PLATFORMS = ["binary_sensor"]
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: