Allow User-Defined Connection/Edge Types

Author: gravy-jones-locker

src/strawberry_sqlalchemy_mapper/mapper.py

@@ -151,6 +151,8 @@ def __init__(
         extra_sqlalchemy_type_to_strawberry_type_map: Optional[
             Mapping[Type[TypeEngine], Type[Any]]
         ] = None,
+        edge_type: Type = None,
+        connection_type: Type = None,
     ) -> None:
         if model_to_type_name is None:
             model_to_type_name = self._default_model_to_type_name
@@ -172,6 +174,9 @@ def __init__(
         self._related_type_models = set()
         self._related_interface_models = set()
 
+        self.edge_type = edge_type
+        self.connection_type = connection_type
+
     @staticmethod
     def _default_model_to_type_name(model: Type[BaseModelType]) -> str:
         return model.__name__
@@ -211,6 +216,8 @@ def _edge_type_for(self, type_name: str) -> Type[Any]:
         Get or create a corresponding Edge model for the given type
         (to support future pagination)
         """
+        if self.edge_type is not None:
+            return self.edge_type
         edge_name = f"{type_name}Edge"
         if edge_name not in self.edge_types:
             self.edge_types[edge_name] = edge_type = strawberry.type(
@@ -229,6 +236,8 @@ def _connection_type_for(self, type_name: str) -> Type[Any]:
         Get or create a corresponding Connection model for the given type
         (to support future pagination)
         """
+        if self.connection_type is not None:
+            return self.connection_type[ForwardRef(type_name)]
         connection_name = f"{type_name}Connection"
         if connection_name not in self.connection_types:
             self.connection_types[connection_name] = connection_type = strawberry.type(