add location factory

Author: lilywuyanru

app/data/__init__.py

@@ -1,15 +1,5 @@
 """app.data"""
-from ..services.location.csbs import CSBSLocationService
-from ..services.location.jhu import JhuLocationService
-from ..services.location.nyt import NYTLocationService
-
-# Mapping of services to data-sources.
-DATA_SOURCES = {
-    "jhu": JhuLocationService(),
-    "csbs": CSBSLocationService(),
-    "nyt": NYTLocationService(),
-}
-
+from ..services.location.location_factory import LocationFactory
 
 def data_source(source):
     """
@@ -18,4 +8,4 @@ def data_source(source):
     :returns: The service.
     :rtype: LocationService
     """
-    return DATA_SOURCES.get(source.lower())
+    return LocationFactory.get_location(source).

app/services/location/location_factory.py

@@ -0,0 +1,12 @@
+"""app.services.location.location_factory.py"""
+
+class LocationFactory:
+    def get_location(source):
+        if source == 'csbs':
+            return CSBSLocationService()
+
+        elif source == 'nyt'
+            return NYTLocationService()
+
+        elif source == 'jhu':
+            return JhuLocationService()