Limit shown fractions

When configuring component, limit shown fractions to fractions relevant to your municipality
eyesoft · Sep 22, 2022 · e5aaa7c · e5aaa7c
1 parent 7c9ae69
commit e5aaa7c
Showing 1 changed file with 34 additions and 3 deletions.
diff --git a/custom_components/min_renovasjon/config_flow.py b/custom_components/min_renovasjon/config_flow.py
@@ -28,7 +28,10 @@
     APP_CUSTOMERS_URL,
     CONST_APP_KEY_VALUE,
     KOMTEK_API_BASE_URL,
-    CONST_URL_FRAKSJONER
+    CONST_URL_FRAKSJONER,
+    CONST_URL_TOMMEKALENDER,
+    CONST_KOMMUNE_NUMMER,
+    CONST_APP_KEY
 )
 
 _LOGGER = logging.getLogger(__name__)
@@ -189,7 +192,7 @@ async def async_step_init(self, user_input=None):
         if user_input is not None:
             if "date_format" not in user_input:
                 user_input["date_format"] = "None"
-           
+
             self.options.update(user_input)
             return self.async_create_entry(title=DOMAIN, data=self.options)
 
@@ -198,11 +201,20 @@ async def async_step_init(self, user_input=None):
         date_format = options.get(CONF_DATE_FORMAT, DEFAULT_DATE_FORMAT)
 
         municipality_code = self.config_entry.data.get(CONF_COUNTY_ID, "")
+        street_name = self.config_entry.data.get(CONF_STREET_NAME, "")
+        street_code = self.config_entry.data.get(CONF_STREET_CODE, "")
+        house_no = self.config_entry.data.get(CONF_HOUSE_NO, "")
+
         fraction_list = await self._get_fractions(municipality_code)
+        calendar = await self._get_calendar(municipality_code, street_name, street_code, house_no)
         fractions = {}
 
         for fraction in fraction_list:
-            fractions[str(fraction["Id"])] = fraction["Navn"]
+            if calendar is not None:
+                if [item for item in calendar if item["FraksjonId"] == fraction["Id"]]:
+                    fractions[str(fraction["Id"])] = fraction["Navn"]
+            else:
+                fractions[str(fraction["Id"])] = fraction["Navn"]
 
         return self.async_show_form(
             step_id="init",
@@ -223,5 +235,24 @@ async def _get_fractions(self, municipality_code) -> Dict:
                 response = await resp.read()
                 if resp.ok:
                     return json.loads(response.decode("UTF-8"))
+                else:
+                    _LOGGER.error("_get_fractions returned: %s", resp)
+                    return None
+        return None
 
+    async def _get_calendar(self, municipality_code, street_name, street_code, house_no) -> Dict:
+        header = {CONST_KOMMUNE_NUMMER: municipality_code, CONST_APP_KEY: CONST_APP_KEY_VALUE}
+        url = CONST_URL_TOMMEKALENDER
+        url = url.replace('[gatenavn]', street_name)
+        url = url.replace('[gatekode]', street_code)
+        url = url.replace('[husnr]', house_no)
+
+        async with aiohttp.ClientSession(headers=header) as session:
+            async with session.get(url) as resp:
+                response = await resp.read()
+                if resp.ok:
+                    return json.loads(response.decode("UTF-8"))
+                else:
+                    _LOGGER.error("_get_calendar returned: %s", resp)
+                    return None
         return None