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

Fix sync oath2 scaffold template (#46219)

Browse source
This commit is contained in:
Martin Hjelmare 2021-02-08 15:00:17 +01:00 • committed by GitHub
parent 6f446cf627
commit 568180632e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

2
script/scaffold/templates/config_flow_oauth2/integration/__init__.py
View file

@ -65,7 +65,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
session = config_entry_oauth2_flow.OAuth2Session(hass, entry, implementation)
# If using a requests-based API lib
hass.data[DOMAIN][entry.entry_id] = api.ConfigEntryAuth(hass, entry, session)
hass.data[DOMAIN][entry.entry_id] = api.ConfigEntryAuth(hass, session)
# If using an aiohttp-based API lib
hass.data[DOMAIN][entry.entry_id] = api.AsyncConfigEntryAuth(

10
script/scaffold/templates/config_flow_oauth2/integration/api.py
View file

@ -4,7 +4,7 @@ from asyncio import run_coroutine_threadsafe
from aiohttp import ClientSession
import my_pypi_package
from homeassistant import config_entries, core
from homeassistant import core
from homeassistant.helpers import config_entry_oauth2_flow
# TODO the following two API examples are based on our suggested best practices
@ -18,15 +18,11 @@ class ConfigEntryAuth(my_pypi_package.AbstractAuth):
def __init__(
self,
hass: core.HomeAssistant,
config_entry: config_entries.ConfigEntry,
implementation: config_entry_oauth2_flow.AbstractOAuth2Implementation,
oauth_session: config_entry_oauth2_flow.OAuth2Session,
):
"""Initialize NEW_NAME Auth."""
self.hass = hass
self.config_entry = config_entry
self.session = config_entry_oauth2_flow.OAuth2Session(
hass, config_entry, implementation
)
self.session = oauth_session
super().__init__(self.session.token)
def refresh_tokens(self) -> str: