Add support for filtering pollens

Author: amaximus

README.md

@@ -25,12 +25,16 @@ Define sensor with the following configuration parameters:<br />
 | :---- | :---- | :------- | :----------- |
 | name | **Y** | `pollen_hu` | name of the sensor |
 | all_dominant | **Y** | `False` | gather all dominant pollens. By the default it will show the first dominant pollen even if there are others too. |
+| pollens | **Y** | `` | list of pollen names to filter on. |
 ---
 
 #### Example
 ```
 platform: pollen_hu
 name: 'Pollen adatok'
+pollens:
+  - Parlagfű
+  - Alternaria
 ```
 
 #### Lovelace UI

custom_components/pollen_hu/manifest.json

@@ -3,7 +3,7 @@
   "name": "Pollen Information Hungary",
   "documentation": "https://github.com/amaximus/pollen_hu",
   "issue_tracker": "https://github.com/amaximus/pollen_hu/issues",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "dependencies": [],
   "codeowners": ["@amaximus"],
   "iot_class": "cloud_polling",

custom_components/pollen_hu/sensor.py

@@ -18,6 +18,7 @@
 CONF_ALLDOMINANT = 'all_dominant'
 CONF_ATTRIBUTION = "Data provided by antsz.hu"
 CONF_NAME = 'name'
+CONF_POLLENS = 'pollens'
 
 DEFAULT_ALLDOMINANT = False
 DEFAULT_ICON = 'mdi:blur'
@@ -28,17 +29,20 @@
 PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
     vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
     vol.Optional(CONF_ALLDOMINANT, default=DEFAULT_ALLDOMINANT): cv.boolean,
+    vol.Optional(CONF_POLLENS, default=[]): vol.All(cv.ensure_list, [cv.string]),
 })
 
 async def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
     name = config.get(CONF_NAME)
     alldominant = config.get(CONF_ALLDOMINANT)
+    pollens = config.get(CONF_POLLENS)
 
     async_add_devices(
-        [PollenHUSensor(hass, name, alldominant )],update_before_add=True)
+        [PollenHUSensor(hass, name, alldominant, pollens )],update_before_add=True)
 
 async def async_get_pdata(self):
-    pjson = {}
+    pjson = {"pollens": []}
+    pjson1 = {}
 
     url = 'https://efop180.antsz.hu/polleninformaciok/'
     async with self._session.get(url) as response:
@@ -59,18 +63,29 @@ async def async_get_pdata(self):
              .replace(" \"","\",\"") \
              .replace("\",\"name","\"},{\"name") \
              .replace(">", "{\"pollens\":[{") + "\"}]}"
-        pjson = json.loads(p3)
+        pjson1 = json.loads(p3)
+        if len(self._pollens) != 0:
+            i = 0
+            for item in pjson1['pollens']:
+                if item.get('name') not in self._pollens:
+                    continue
+                else:
+                    pjson['pollens'].append({"name": item.get('name'),"value": item.get('value')})
+                    i += 1;
+        else:
+            pjson = pjson1
     return pjson
 
 class PollenHUSensor(Entity):
 
-    def __init__(self, hass, name, alldominant):
+    def __init__(self, hass, name, alldominant, pollens):
         """Initialize the sensor."""
         self._hass = hass
         self._name = name
         self._alldominant = alldominant
         self._state = None
         self._pdata = []
+        self._pollens = pollens
         self._icon = DEFAULT_ICON
         self._session = async_get_clientsession(hass)
 
@@ -105,9 +120,8 @@ def extra_state_attributes(self):
     async def async_update(self):
         dominant_value = 0
 
-        pdata = await async_get_pdata(self)
+        self._pdata = await async_get_pdata(self)
 
-        self._pdata = pdata
         if 'pollens' in self._pdata:
             for item in self._pdata['pollens']:
                 val = item.get('value')

info.md

@@ -25,12 +25,16 @@ Define sensor with the following configuration parameters:<br />
 | :---- | :---- | :------- | :----------- |
 | name | **Y** | `pollen_hu` | name of the sensor |
 | all_dominant | **Y** | `False` | gather all dominant pollens. By the default it will show the first dominant pollen even if there are others too. |
+| pollens | **Y** | `` | list of pollen names to filter on. |
 ---
 
 #### Example
 ```
 platform: pollen_hu
 name: 'Pollen adatok'
+pollens:
+  - Parlagfű
+  - Alternaria
 ```
 
 #### Lovelace UI

tracker.json

@@ -1,8 +1,8 @@
 {
   "pollen_hu": {
-    "version": "0.3.0",
-    "updated_at": "2023-06-08",
+    "version": "0.4.0",
+    "updated_at": "2023-07-10",
     "visit_repo": "https://github.com/amaximus/pollen_hu",
-    "changelog": "https://github.com/amaximus/pollen_hu/releases/0.3.0"
+    "changelog": "https://github.com/amaximus/pollen_hu/releases/0.4.0"
   }
 }