Data_Source change for a more Aggregate approach

Author: Solracion

app/data/__init__.py

@@ -2,20 +2,16 @@
 from ..services.location.csbs import CSBSLocationService
 from ..services.location.jhu import JhuLocationService
 from ..services.location.nyt import NYTLocationService
+from ..services.location import LocationService
 
 # Mapping of services to data-sources.
-DATA_SOURCES = {
-    "jhu": JhuLocationService(),
-    "csbs": CSBSLocationService(),
-    "nyt": NYTLocationService(),
-}
+class DATA_SOURCES:
+    DATA_SOURCES = {}
 
+    def data(self)
+    self.data_Sources['jhu'] = JhuLocationService()
+    self.data_Sources['csbs'] = CSBSLocationService()
+    self.data_Sources['nyt'] = NYTLocationService()
 
-def data_source(source):
-    """
-    Retrieves the provided data-source service.
-
-    :returns: The service.
-    :rtype: LocationService
-    """
-    return DATA_SOURCES.get(source.lower())
+    def data_source(self, source : str)-> LocationService:
+         return self.dataSource.get(source.lower())

app/main.py

@@ -21,7 +21,6 @@
 # ############
 # FastAPI App
 # ############
-# this is a test
 LOGGER = logging.getLogger("api")
 
 SETTINGS = get_settings()