From a530ba8e8e34dda1c3be63e768315860f1f8f901 Mon Sep 17 00:00:00 2001
From: vingerha <44190435+vingerha@users.noreply.github.com>
Date: Sat, 23 Mar 2024 17:38:33 +0100
Subject: [PATCH] Adjust configflow for local stops realtime

---
 custom_components/gtfs2/config_flow.py | 44 +++++++++++++++++---------
 1 file changed, 29 insertions(+), 15 deletions(-)

diff --git a/custom_components/gtfs2/config_flow.py b/custom_components/gtfs2/config_flow.py
index 8070203..306a252 100644
--- a/custom_components/gtfs2/config_flow.py
+++ b/custom_components/gtfs2/config_flow.py
@@ -437,18 +437,32 @@ async def async_step_real_time(
             self._user_inputs.update(user_input)
             _LOGGER.debug(f"UserInput Realtime: {self._user_inputs}")
             return self.async_create_entry(title="", data=self._user_inputs)
-
-        return self.async_show_form(
-            step_id="real_time",
-            data_schema=vol.Schema(
-                {
-                    vol.Required(CONF_TRIP_UPDATE_URL, default=self.config_entry.options.get(CONF_TRIP_UPDATE_URL)): str,
-                    vol.Optional(CONF_VEHICLE_POSITION_URL, default=self.config_entry.options.get(CONF_VEHICLE_POSITION_URL,"")): str,
-                    vol.Optional(CONF_ALERTS_URL, default=self.config_entry.options.get(CONF_ALERTS_URL,"")): str,
-                    vol.Optional(CONF_API_KEY, default=self.config_entry.options.get(CONF_API_KEY,"")): str,
-                    vol.Optional(CONF_X_API_KEY,default=self.config_entry.options.get(CONF_X_API_KEY,"")): str,
-                    vol.Required(CONF_API_KEY_LOCATION, default=self.config_entry.options.get(CONF_API_KEY_LOCATION,DEFAULT_API_KEY_LOCATION)) : selector.SelectSelector(selector.SelectSelectorConfig(options=ATTR_API_KEY_LOCATIONS, translation_key="api_key_location")),
-                },
-            ),
-            errors=errors,
-        )  
+        
+        if self.config_entry.data.get(CONF_DEVICE_TRACKER_ID, None):
+            return self.async_show_form(
+                step_id="real_time",
+                data_schema=vol.Schema(
+                    {
+                        vol.Required(CONF_TRIP_UPDATE_URL, default=self.config_entry.options.get(CONF_TRIP_UPDATE_URL)): str,
+                        vol.Optional(CONF_API_KEY, default=self.config_entry.options.get(CONF_API_KEY,"")): str,
+                        vol.Optional(CONF_X_API_KEY,default=self.config_entry.options.get(CONF_X_API_KEY,"")): str,
+                        vol.Required(CONF_API_KEY_LOCATION, default=self.config_entry.options.get(CONF_API_KEY_LOCATION,DEFAULT_API_KEY_LOCATION)) : selector.SelectSelector(selector.SelectSelectorConfig(options=ATTR_API_KEY_LOCATIONS, translation_key="api_key_location")),
+                    },
+                ),
+                errors=errors,
+            )  
+        else:
+            return self.async_show_form(
+                step_id="real_time",
+                data_schema=vol.Schema(
+                    {
+                        vol.Required(CONF_TRIP_UPDATE_URL, default=self.config_entry.options.get(CONF_TRIP_UPDATE_URL)): str,
+                        vol.Optional(CONF_VEHICLE_POSITION_URL, default=self.config_entry.options.get(CONF_VEHICLE_POSITION_URL,"")): str,
+                        vol.Optional(CONF_ALERTS_URL, default=self.config_entry.options.get(CONF_ALERTS_URL,"")): str,
+                        vol.Optional(CONF_API_KEY, default=self.config_entry.options.get(CONF_API_KEY,"")): str,
+                        vol.Optional(CONF_X_API_KEY,default=self.config_entry.options.get(CONF_X_API_KEY,"")): str,
+                        vol.Required(CONF_API_KEY_LOCATION, default=self.config_entry.options.get(CONF_API_KEY_LOCATION,DEFAULT_API_KEY_LOCATION)) : selector.SelectSelector(selector.SelectSelectorConfig(options=ATTR_API_KEY_LOCATIONS, translation_key="api_key_location")),
+                    },
+                ),
+                errors=errors,
+            )