Structural pattern: Composite pattern

Author: Vimarsh Patel

app/location/csbs.py

@@ -1,6 +1,6 @@
 """app.locations.csbs.py"""
 from . import Location
-
+from . import serialize
 
 class CSBSLocation(Location):
     """
@@ -25,19 +25,7 @@ def __init__(self, id, state, county, coordinates, last_updated, confirmed, deat
         self.state = state
         self.county = county
 
-    def serialize(self, timelines=False):  # pylint: disable=arguments-differ,unused-argument
-        """
-        Serializes the location into a dict.
-
-        :returns: The serialized location.
-        :rtype: dict
-        """
-        serialized = super().serialize()
+   
+        
 
-        # Update with new fields.
-        serialized.update(
-            {"state": self.state, "county": self.county,}
-        )
 
-        # Return the serialized location.
-        return serialized

app/location/nyt.py

@@ -1,5 +1,6 @@
 """app.locations.nyt.py"""
 from . import TimelinedLocation
+from . import serialize
 
 
 class NYTLocation(TimelinedLocation):
@@ -14,19 +15,4 @@ def __init__(self, id, state, county, coordinates, last_updated, timelines):
         self.state = state
         self.county = county
 
-    def serialize(self, timelines=False):  # pylint: disable=arguments-differ,unused-argument
-        """
-        Serializes the location into a dict.
-
-        :returns: The serialized location.
-        :rtype: dict
-        """
-        serialized = super().serialize(timelines)
-
-        # Update with new fields.
-        serialized.update(
-            {"state": self.state, "county": self.county,}
-        )
-
-        # Return the serialized location.
-        return serialized
+ 

app/location/serialize.py

@@ -0,0 +1,19 @@
+
+class serialize():
+    def serialize(self, timelines=False):  # pylint: disable=arguments-differ,unused-argument
+        """
+        Serializes the location into a dict.
+
+        :returns: The serialized location.
+        :rtype: dict
+        """
+        serialized = super().serialize(timelines)
+
+        # Update with new fields.
+        serialized.update(
+            {"state": self.state, "county": self.county,}
+        )
+
+        # Return the serialized location.
+        return serialized
+