From 64fbc5eeb6bb69bf60d976321cac4b2a0d6b150f Mon Sep 17 00:00:00 2001 From: Briar Date: Mon, 14 Oct 2024 14:42:33 -0400 Subject: [PATCH] Remove filter for food security --- src/hapi/pipelines/app/pipelines.py | 1 + src/hapi/pipelines/database/food_security.py | 6 ++++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/hapi/pipelines/app/pipelines.py b/src/hapi/pipelines/app/pipelines.py index 1db93560..7305e035 100644 --- a/src/hapi/pipelines/app/pipelines.py +++ b/src/hapi/pipelines/app/pipelines.py @@ -246,6 +246,7 @@ def output_food_security(self): admins=self.admins, adminone=self.adminone, admintwo=self.admintwo, + countryiso3s=self.countries, configuration=self.configuration, ) food_security.populate() diff --git a/src/hapi/pipelines/database/food_security.py b/src/hapi/pipelines/database/food_security.py index e3cad0ce..74b59670 100644 --- a/src/hapi/pipelines/database/food_security.py +++ b/src/hapi/pipelines/database/food_security.py @@ -2,7 +2,7 @@ from dataclasses import dataclass from logging import getLogger -from typing import Dict, Optional, Set +from typing import Dict, List, Optional, Set from hapi_schema.db_food_security import DBFoodSecurity from hdx.api.configuration import Configuration @@ -38,6 +38,7 @@ def __init__( admins: admins.Admins, adminone: AdminLevel, admintwo: AdminLevel, + countryiso3s: List[str], configuration: Configuration, ): super().__init__(session) @@ -45,6 +46,7 @@ def __init__( self._admins = admins self._adminone = adminone self._admintwo = admintwo + self._countryiso3s = countryiso3s self._configuration = configuration self._country_status = {} @@ -334,7 +336,7 @@ def populate(self) -> None: if "#" in row["Date of analysis"]: # ignore HXL row continue countryiso3 = row["Country"] - if countryiso3 not in self._configuration["HAPI_countries"]: + if countryiso3 not in self._countryiso3s: continue provider_admin1_name = get_provider_name(row, "Level 1") provider_admin2_name = get_provider_name(row, "Area")