chore: Pass code style

Author: NaqGuug

scim2_models/base.py

@@ -116,7 +116,7 @@ class _SCIMClassInfo(NamedTuple):
     complex_fields: frozenset[str] = frozenset()
     """Field names whose root type is a ``ComplexAttribute`` subclass."""
 
-    extensions: frozenset[set] = frozenset()
+    extensions: frozenset[str] = frozenset()
     """Field names whose root type is a ``Extension`` subclass."""
 
 
@@ -288,7 +288,7 @@ def __pydantic_on_complete__(cls) -> None:
 
             # Is extension
             if (
-                main_schema
+                extension_cls is not None
                 and isclass(root_type)
                 and issubclass(root_type, extension_cls)
             ):
@@ -465,7 +465,6 @@ def _apply_replace_constraints(self, original: Self) -> None:
 
         Recursively applies to nested single-valued complex attributes.
         """
-
         for field_name in type(self).model_fields:
             mutability = type(self).get_field_annotation(field_name, Mutability)
             original_val = getattr(original, field_name)
@@ -485,15 +484,16 @@ def _apply_replace_constraints(self, original: Self) -> None:
                         attribute=field_name, mutability="immutable"
                     )
 
-        complex_and_extensions = self.__scim_info__.complex_fields.union(self.__scim_info__.extensions)
+        complex_and_extensions = self.__scim_info__.complex_fields.union(
+            self.__scim_info__.extensions
+        )
         for complex_attr in complex_and_extensions:
             if not type(self).get_field_multiplicity(complex_attr):
                 original_sub = getattr(original, complex_attr)
                 replacement_sub = getattr(self, complex_attr)
                 if original_sub is not None and replacement_sub is not None:
                     replacement_sub._apply_replace_constraints(original_sub)
 
-
     def get_attribute_urn(self, field_name: str) -> str:
         """Build the full URN of the attribute.
 
@@ -542,7 +542,11 @@ def scim_serializer(
 
         # Delete empty extensions
         for extension_field in self.__scim_info__.extensions:
-            key = self.__scim_info__.attribute_urns[extension_field] if info.by_alias else extension_field
+            key = (
+                self.__scim_info__.attribute_urns[extension_field]
+                if info.by_alias
+                else extension_field
+            )
             if key in serialized and serialized[key] is None:
                 del serialized[key]
 

scim2_models/resources/resource.py

@@ -112,7 +112,7 @@ def from_schema(cls, schema: "Schema") -> type["Extension"]:
 
 def _extension_serializer(
     value: Any, handler: SerializerFunctionWrapHandler, info: SerializationInfo
-) -> dict[str, Any] | None:
+) -> Any:
     """Exclude the Resource attributes from the extension dump.
 
     For instance, attributes 'meta', 'id' or 'schemas' should not be

tests/test_model_serialization.py

@@ -7,8 +7,8 @@
 from scim2_models.annotations import Returned
 from scim2_models.attributes import ComplexAttribute
 from scim2_models.context import Context
-from scim2_models.resources.resource import Resource
 from scim2_models.resources.resource import Extension
+from scim2_models.resources.resource import Resource
 
 
 class SubRetModel(ComplexAttribute):
@@ -155,7 +155,7 @@ def test_dump_default(mut_resource):
 
 
 def test_dump_extension(mut_resource_extension):
-    """Test dumps with extension"""
+    """Test dumps with extension."""
     assert mut_resource_extension.model_dump() == {
         "schemas": ["urn:org:example:MutResource", "urn:org:extensions:MutExtension"],
         "id": "id",
@@ -168,7 +168,7 @@ def test_dump_extension(mut_resource_extension):
             "readWrite": "y",
             "immutable": "y",
             "writeOnly": "y",
-        }
+        },
     }
 
     assert mut_resource_extension.model_dump(scim_ctx=None) == {
@@ -186,7 +186,7 @@ def test_dump_extension(mut_resource_extension):
             "read_write": "y",
             "immutable": "y",
             "write_only": "y",
-        }
+        },
     }
 
     assert mut_resource_extension.model_dump(scim_ctx=None, exclude_none=True) == {
@@ -202,7 +202,7 @@ def test_dump_extension(mut_resource_extension):
             "read_write": "y",
             "immutable": "y",
             "write_only": "y",
-        }
+        },
     }
 
     assert mut_resource_extension.model_dump(by_alias=False) == {
@@ -217,7 +217,7 @@ def test_dump_extension(mut_resource_extension):
             "read_write": "y",
             "immutable": "y",
             "write_only": "y",
-        }
+        },
     }
 
     assert mut_resource_extension.model_dump(scim_ctx=None, by_alias=True) == {
@@ -235,12 +235,12 @@ def test_dump_extension(mut_resource_extension):
             "readWrite": "y",
             "immutable": "y",
             "writeOnly": "y",
-        }
+        },
     }
 
 
 def test_dump_empty_extension(mut_resource_extension_empty):
-    """Test dumps with empty extension"""
+    """Test dumps with empty extension."""
     assert mut_resource_extension_empty.model_dump() == {
         "schemas": ["urn:org:example:MutResource", "urn:org:extensions:MutExtension"],
         "id": "id",
@@ -265,17 +265,19 @@ def test_dump_empty_extension(mut_resource_extension_empty):
             "read_write": None,
             "immutable": None,
             "write_only": None,
-        }
+        },
     }
 
-    assert mut_resource_extension_empty.model_dump(scim_ctx=None, exclude_none=True) == {
+    assert mut_resource_extension_empty.model_dump(
+        scim_ctx=None, exclude_none=True
+    ) == {
         "schemas": ["urn:org:example:MutResource", "urn:org:extensions:MutExtension"],
         "id": "id",
         "read_only": "x",
         "read_write": "x",
         "immutable": "x",
         "write_only": "x",
-        "MutExtension": {"schemas": ["urn:org:extensions:MutExtension"]}
+        "MutExtension": {"schemas": ["urn:org:extensions:MutExtension"]},
     }
 
     assert mut_resource_extension_empty.model_dump(by_alias=False) == {
@@ -302,7 +304,7 @@ def test_dump_empty_extension(mut_resource_extension_empty):
             "readWrite": None,
             "immutable": None,
             "writeOnly": None,
-        }
+        },
     }