Structural Pattern -- Composite

Author: louliu888

app/coordinates.py

@@ -2,6 +2,8 @@
 from ..services.location.csbs import CSBSLocationService
 from ..services.location.jhu import JhuLocationService
 from ..services.location.nyt import NYTLocationService
+from ..services.location.dataSource import dataSource
+
 
 # Mapping of services to data-sources.
 DATA_SOURCES = {
@@ -10,12 +12,3 @@
     "nyt": NYTLocationService(),
 }
 
-
-def data_source(source):
-    """
-    Retrieves the provided data-source service.
-
-    :returns: The service.
-    :rtype: LocationService
-    """
-    return DATA_SOURCES.get(source.lower())

app/data/__init__.py

@@ -0,0 +1,9 @@
+class dataSource(object):
+    def data_source(source):
+    """
+    Retrieves the provided data-source service.
+
+    :returns: The service.
+    :rtype: LocationService
+    """
+    return DATA_SOURCES.get(source.lower())