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

Use setdefault() in scaffold script for setting hass.data (#103338)

Browse Source
This commit is contained in:
Matthias Alphart 2023-11-04 10:51:34 +01:00 committed by GitHub
parent bb8375da72
commit 5cd27a877e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -11,7 +11,7 @@ from .const import DOMAIN
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up NEW_NAME from a config entry.""" """Set up NEW_NAME from a config entry."""
# TODO Optionally store an object for your platforms to access # TODO Optionally store an object for your platforms to access
# hass.data[DOMAIN][entry.entry_id] = ... # hass.data.setdefault(DOMAIN, {})[entry.entry_id] = ...
# TODO Optionally validate config entry options before setting up platform # TODO Optionally validate config entry options before setting up platform

6
script/scaffold/templates/config_flow_oauth2/integration/__init__.py
View File

@ -25,10 +25,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
session = config_entry_oauth2_flow.OAuth2Session(hass, entry, implementation) session = config_entry_oauth2_flow.OAuth2Session(hass, entry, implementation)
# If using a requests-based API lib # If using a requests-based API lib
hass.data[DOMAIN][entry.entry_id] = api.ConfigEntryAuth(hass, session) hass.data.setdefault(DOMAIN, {})[entry.entry_id] = api.ConfigEntryAuth(
hass, session
)
# If using an aiohttp-based API lib # If using an aiohttp-based API lib
hass.data[DOMAIN][entry.entry_id] = api.AsyncConfigEntryAuth( hass.data.setdefault(DOMAIN, {})[entry.entry_id] = api.AsyncConfigEntryAuth(
aiohttp_client.async_get_clientsession(hass), session aiohttp_client.async_get_clientsession(hass), session
) )