Refactor relationship mapping in QueryBuilder and related classes

src/masoniteorm/query/QueryBuilder.py

@@ -1968,12 +1968,7 @@ def _register_relationships_to_model(
         return self
 
     def _map_related(self, related_result, related):
-        if related.__class__.__name__ == 'MorphTo':
-            return related_result
-        elif related.__class__.__name__ in ['HasOneThrough', 'HasManyThrough']:
-            return related_result.group_by(related.local_key)
-
-        return related_result
+        return related.map_related(related_result)
 
     def all(self, selects=[], query=False):
         """Returns all records from the table.

src/masoniteorm/relationships/BaseRelationship.py

@@ -193,3 +193,6 @@ def query_has(self, current_query_builder, method="where_exists"):
         )
 
         return related_builder
+
+    def map_related(self, related_result):
+        return related_result

src/masoniteorm/relationships/HasOneThrough.py

@@ -267,3 +267,6 @@ def get_with_count_query(self, current_builder, callback):
         )
 
         return return_query
+
+    def map_related(self, related_result):
+        return related_result.group_by(related.local_key)

src/masoniteorm/relationships/MorphTo.py

@@ -133,3 +133,6 @@ def attach_related(self, current_model, related_record):
         raise NotImplementedError(
             "MorphTo relationship does not implement the attach_related method"
         )
+
+    def map_related(self, related_result):
+        return related_result