Merge pull request #9 from gravy-jones-locker/federation_compatibility

Added Federation Type Option

Author: TimDumol

.pre-commit-config.yaml

@@ -2,7 +2,7 @@ exclude: '^docs/conf.py'
 
 repos:
 - repo: https://github.com/pre-commit/pre-commit-hooks
-  rev: v4.1.0
+  rev: v4.3.0
   hooks:
   - id: trailing-whitespace
   - id: check-added-large-files
@@ -28,7 +28,7 @@ repos:
 #       --remove-unused-variables,
 #     ]
 - repo: https://github.com/hadialqattan/pycln
-  rev: v1.2.4
+  rev: v2.1.1
   hooks:
     - id: pycln
       args: [--config=setup.cfg]
@@ -39,7 +39,7 @@ repos:
   - id: isort
 
 - repo: https://github.com/psf/black
-  rev: stable
+  rev: 22.8.0
   hooks:
   - id: black
     language_version: python3
@@ -52,7 +52,7 @@ repos:
 #     additional_dependencies: [black]
 
 - repo: https://github.com/PyCQA/flake8
-  rev: 4.0.1
+  rev: 5.0.4
   hooks:
   - id: flake8
   ## You can add flake8 plugins via `additional_dependencies`:

src/strawberry_sqlalchemy_mapper/mapper.py

@@ -413,14 +413,12 @@ async def resolve(self, info: Info):
                         return []
                     else:
                         return None
-		if isinstance(info.context, dict):
+                if isinstance(info.context, dict):
                     loader = info.context["sqlalchemy_loader"]
                 else:
                     loader = info.context.sqlalchemy_loader
-                related_objects = (
-                    await loader
-                    .loader_for(relationship)
-                    .load(relationship_key)
+                related_objects = await loader.loader_for(relationship).load(
+                    relationship_key
                 )
             return related_objects
 
@@ -537,7 +535,10 @@ def _handle_columns(
                 )
 
     def type(
-        self, model: Type[BaseModelType], make_interface=False
+        self,
+        model: Type[BaseModelType],
+        make_interface=False,
+        use_federation=False,
     ) -> Callable[[Type[object]], Any]:
         """
         Decorate a type with this to register it as a strawberry type
@@ -662,10 +663,11 @@ def convert(type_: Any) -> Any:
 
             if make_interface:
                 mapped_type = strawberry.interface(type_)
-                self.mapped_interfaces[type_.__name__] = mapped_type
+            elif use_federation:
+                mapped_type = strawberry.federation.type(type_)
             else:
                 mapped_type = strawberry.type(type_)
-                self.mapped_types[type_.__name__] = mapped_type
+            self.mapped_types[type_.__name__] = mapped_type
             setattr(mapped_type, _GENERATED_FIELD_KEYS_KEY, generated_field_keys)
             setattr(mapped_type, _ORIGINAL_TYPE_KEY, type_)
             return mapped_type