Add context to login flow

homeassistant/auth/__init__.py

@@ -215,7 +215,7 @@ async def _async_create_login_flow(self, handler, *, context, data):
         """Create a login flow."""
         auth_provider = self._providers[handler]
 
-        return await auth_provider.async_credential_flow()
+        return await auth_provider.async_credential_flow(context)
 
     async def _async_finish_login_flow(self, result):
         """Result of a credential login flow."""

homeassistant/auth/providers/__init__.py

@@ -123,7 +123,7 @@ def async_create_credentials(self, data):
 
     # Implement by extending class
 
-    async def async_credential_flow(self):
+    async def async_credential_flow(self, context):
         """Return the data flow for logging in with auth provider."""
         raise NotImplementedError
 

homeassistant/auth/providers/homeassistant.py

@@ -158,7 +158,7 @@ async def async_initialize(self):
         self.data = Data(self.hass)
         await self.data.async_load()
 
-    async def async_credential_flow(self):
+    async def async_credential_flow(self, context):
         """Return a flow to login."""
         return LoginFlow(self)
 

homeassistant/auth/providers/insecure_example.py

@@ -31,7 +31,7 @@ class InvalidAuthError(HomeAssistantError):
 class ExampleAuthProvider(AuthProvider):
     """Example auth provider based on hardcoded usernames and passwords."""
 
-    async def async_credential_flow(self):
+    async def async_credential_flow(self, context):
         """Return a flow to login."""
         return LoginFlow(self)
 

homeassistant/auth/providers/legacy_api_password.py

@@ -36,7 +36,7 @@ class LegacyApiPasswordAuthProvider(AuthProvider):
 
     DEFAULT_TITLE = 'Legacy API Password'
 
-    async def async_credential_flow(self):
+    async def async_credential_flow(self, context):
         """Return a flow to login."""
         return LoginFlow(self)
 

homeassistant/components/auth/login_flow.py

@@ -152,7 +152,7 @@ async def post(self, request, data):
             handler = data['handler']
 
         try:
-            result = await self._flow_mgr.async_init(handler)
+            result = await self._flow_mgr.async_init(handler, context={})
         except data_entry_flow.UnknownHandler:
             return self.json_message('Invalid handler specified', 404)
         except data_entry_flow.UnknownStep:

homeassistant/data_entry_flow.py

@@ -47,6 +47,7 @@ def async_progress(self) -> List[Dict]:
             'flow_id': flow.flow_id,
             'handler': flow.handler,
             'source': flow.source,
+            'context': flow.context,
         } for flow in self._progress.values()]
 
     async def async_init(self, handler: Callable, *, context: Dict = None,
@@ -57,6 +58,7 @@ async def async_init(self, handler: Callable, *, context: Dict = None,
         flow.hass = self.hass
         flow.handler = handler
         flow.flow_id = uuid.uuid4().hex
+        flow.context = context
         self._progress[flow.flow_id] = flow
 
         return await self._async_handle_step(flow, flow.init_step, data)
@@ -122,11 +124,12 @@ class FlowHandler:
     flow_id = None
     hass = None
     handler = None
-    source = None
     cur_step = None
+    context = None
 
     # Set by _async_create_flow callback
     init_step = 'init'
+    source = None
 
     # Set by developer
     VERSION = 1

tests/components/config/test_config_entries.py

@@ -295,7 +295,8 @@ def async_step_account(self, user_input=None):
         {
             'flow_id': form['flow_id'],
             'handler': 'test',
-            'source': 'hassio'
+            'source': 'hassio',
+            'context': {'source': 'hassio'}
         }
     ]