Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 5 additions & 2 deletions alert_system/etl/usgs_earthquake/transform.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,10 @@ class USGSTransformer(BaseTransformerClass):
# NOTE: This logic might change in future
def compute_people_exposed(self, metadata_list) -> Optional[int]:
for data in metadata_list:
if data["category"] == ImpactDetailsEnum.Category.PEOPLE and data["type"] == ImpactDetailsEnum.Type.AFFECTED_TOTAL:
if data["category"] == ImpactDetailsEnum.Category.PEOPLE and data["type"] in [
ImpactDetailsEnum.Type.DEATH,
ImpactDetailsEnum.Type.POTENTIALLY_AFFECTED,
]:
return data["value"]
return None

Expand All @@ -26,7 +29,7 @@ def compute_buildings_exposed(self, metadata_list) -> Optional[int]:
Compute the 'buildings_exposed' field.
"""
for data in metadata_list:
if data["category"] == "buildings" and data["type"] == "damaged":
if data["category"] == ImpactDetailsEnum.Category.BUILDINGS and data["type"] == ImpactDetailsEnum.Type.LOSS_COST:
return data["value"]
return None

Expand Down
3 changes: 3 additions & 0 deletions alert_system/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,9 @@ class Type(models.TextChoices):
HIGHEST_RISK = "highest_risk", _("Highest Risk")
AFFECTED_TOTAL = "affected_total", _("Affected Total")
AFFECTED_DIRECT = "affected_direct", _("Affected Direct")
DEATH = "death", _("Death")
LOSS_COST = "cost", _("Cost")
DAMAGED = "damaged", _("Damaged")

class Category(models.TextChoices):
PEOPLE = "people", _("People")
Expand Down
Loading