mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-09-08 17:33:12 -04:00
49195cb0d3
This is a better name for the module because it’s: - not just constants, there’s a method in here now - only stuff to do with populations, not other kinds of constants
99 lines
2.8 KiB
Python
99 lines
2.8 KiB
Python
from notifications_utils.serialised_model import SerialisedModelCollection
|
|
from werkzeug.utils import cached_property
|
|
|
|
from .polygons import Polygons
|
|
from .populations import CITY_OF_LONDON
|
|
from .repo import BroadcastAreasRepository
|
|
|
|
|
|
class SortableMixin:
|
|
|
|
def __repr__(self):
|
|
return f'{self.__class__.__name__}(<{self.id}>)'
|
|
|
|
def __lt__(self, other):
|
|
# Implementing __lt__ means any classes inheriting from this
|
|
# method are sortable
|
|
return self.name < other.name
|
|
|
|
|
|
class GetItemByIdMixin:
|
|
def get(self, id):
|
|
for item in self:
|
|
if item.id == id:
|
|
return item
|
|
raise KeyError(id)
|
|
|
|
|
|
class BroadcastArea(SortableMixin):
|
|
|
|
def __init__(self, row):
|
|
self.id, self.name, self._count_of_phones = row
|
|
|
|
def __eq__(self, other):
|
|
return self.id == other.id
|
|
|
|
@cached_property
|
|
def polygons(self):
|
|
return Polygons(
|
|
BroadcastAreasRepository().get_polygons_for_area(self.id)
|
|
)
|
|
|
|
@cached_property
|
|
def simple_polygons(self):
|
|
return Polygons(
|
|
BroadcastAreasRepository().get_simple_polygons_for_area(self.id)
|
|
)
|
|
|
|
@cached_property
|
|
def sub_areas(self):
|
|
return [
|
|
BroadcastArea(row)
|
|
for row in BroadcastAreasRepository().get_all_areas_for_group(self.id)
|
|
]
|
|
|
|
@property
|
|
def count_of_phones(self):
|
|
if self.id.endswith(CITY_OF_LONDON.WARDS):
|
|
return CITY_OF_LONDON.DAYTIME_POPULATION * (
|
|
self.polygons.estimated_area / CITY_OF_LONDON.AREA_SQUARE_MILES
|
|
)
|
|
if self.sub_areas:
|
|
return sum(area.count_of_phones or 0 for area in self.sub_areas)
|
|
return self._count_of_phones
|
|
|
|
|
|
class BroadcastAreaLibrary(SerialisedModelCollection, SortableMixin, GetItemByIdMixin):
|
|
|
|
model = BroadcastArea
|
|
|
|
def __init__(self, row):
|
|
id, name, name_singular, is_group = row
|
|
self.id = id
|
|
self.name = name
|
|
self.name_singular = name_singular
|
|
self.is_group = bool(is_group)
|
|
self.items = BroadcastAreasRepository().get_all_areas_for_library(self.id)
|
|
|
|
def get_examples(self):
|
|
return BroadcastAreasRepository().get_library_description(self.id)
|
|
|
|
|
|
class BroadcastAreaLibraries(SerialisedModelCollection, GetItemByIdMixin):
|
|
|
|
model = BroadcastAreaLibrary
|
|
|
|
def __init__(self):
|
|
self.items = BroadcastAreasRepository().get_libraries()
|
|
|
|
def get_areas(self, *area_ids):
|
|
# allow people to call `get_areas('a', 'b') or get_areas(['a', 'b'])`
|
|
if len(area_ids) == 1 and isinstance(area_ids[0], list):
|
|
area_ids = area_ids[0]
|
|
|
|
areas = BroadcastAreasRepository().get_areas(area_ids)
|
|
return [BroadcastArea(area) for area in areas]
|
|
|
|
|
|
broadcast_area_libraries = BroadcastAreaLibraries()
|