Merge pull request #11 from jalezi/municipalities

Municipalities

cepimose/__init__.py

@@ -11,6 +11,10 @@
     _vaccinations_by_age_range_90_dose1_req,
     _vaccinations_by_age_range_90_dose2_req,
     _vaccination_by_age_range_requests,
+    _vaccinations_pomurska_by_day_req,
+    _vaccinations_by_region_by_day_requests,
+    _vaccinations_municipalities_share_req,
+    _vaccinations_timestamp_req,
 )
 from .parser import (
     _parse_vaccinations_by_age,
@@ -20,6 +24,9 @@
     _parse_vaccines_supplied_by_manufacturer,
     _parse_vaccines_supplied_by_manufacturer_cum,
     _parse_vaccinations_by_age_range,
+    _parse_vaccinations_by_region_by_day,
+    _parse_vaccinations_by_municipalities_share,
+    _parse_vaccinations_timestamp,
 )
 
 from .types import (
@@ -39,6 +46,10 @@ def _get_data(req, parse_response):
     return parse_response(resp.json())
 
 
+def vaccinations_timestamp():
+    return _get_data(_vaccinations_timestamp_req, _parse_vaccinations_timestamp)
+
+
 def vaccinations_by_day() -> "list[VaccinationByDayRow]":
     return _get_data(_vaccinations_by_day_req, _parse_vaccinations_by_day)
 
@@ -98,3 +109,32 @@ def vaccinations_by_age_range():
         obj[key] = VaccinationByAgeRange(dose1=dose1, dose2=dose2)
 
     return obj
+
+
+# by individual region
+def vaccinations_pomurska_by_day() -> "list[VaccinationByDayRow]":
+    return _get_data(
+        _vaccinations_pomurska_by_day_req, _parse_vaccinations_by_region_by_day
+    )
+
+
+def vaccinations_by_region_by_day():
+    key_value = _vaccinations_by_region_by_day_requests.items()
+    obj = {}
+    for el in key_value:
+        key = el[0]
+        doses_req = el[1][0]
+
+        doses = _get_data(doses_req, _parse_vaccinations_by_region_by_day)
+        obj[key] = doses
+
+    return obj
+
+
+# PAGE 2
+# municipalities
+def vaccinations_by_municipalities_share():
+    return _get_data(
+        _vaccinations_municipalities_share_req,
+        _parse_vaccinations_by_municipalities_share,
+    )