diff --git a/app/location/__init__.py b/app/location/__init__.py index 1da5e9e5..147b6aa0 100644 --- a/app/location/__init__.py +++ b/app/location/__init__.py @@ -4,6 +4,19 @@ from ..utils.populations import country_population + +class Locations: + def __init__(self,locations: list[Location]): + self.locations = locations + + def add_service(self,location): + self.locations.append(location) + + def get_location(self): + return locations + + + # pylint: disable=redefined-builtin,invalid-name class Location: # pylint: disable=too-many-instance-attributes """ diff --git a/app/services/location/__init__.py b/app/services/location/__init__.py index 6d292b54..981dc770 100644 --- a/app/services/location/__init__.py +++ b/app/services/location/__init__.py @@ -2,6 +2,16 @@ from abc import ABC, abstractmethod +class LocationServices: + def __init__(self,location_services: list[LocationService]): + self.location_services = location_services + + def add_service(self,location_service): + self.location_services.append(location_service) + + def get_service(self): + return location_services + class LocationService(ABC): """ Service for retrieving locations.