strawberry-graphql 0.222.0__py3-none-any.whl → 0.224.0__py3-none-any.whl

strawberry/channels/handlers/http_handler.py

@@ -2,6 +2,7 @@
 
 A consumer to provide a graphql endpoint, and optionally graphiql.
 """
+
 from __future__ import annotations
 
 import dataclasses

strawberry/experimental/__init__.py

@@ -1,6 +1,6 @@
 try:
     from . import pydantic
-except ImportError:
+except ModuleNotFoundError:
     pass
 else:
     __all__ = ["pydantic"]

strawberry/experimental/pydantic/_compat.py

@@ -1,10 +1,16 @@
 import dataclasses
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Type
+from decimal import Decimal
+from functools import cached_property
+from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Type
+from uuid import UUID
 
+import pydantic
 from pydantic import BaseModel
 from pydantic.version import VERSION as PYDANTIC_VERSION
 
+from strawberry.experimental.pydantic.exceptions import UnsupportedTypeError
+
 if TYPE_CHECKING:
     from pydantic.fields import FieldInfo
 
@@ -24,21 +30,101 @@ class CompatModelField:
     allow_none: bool
     has_alias: bool
     description: Optional[str]
+    _missing_type: Any
+    is_v1: bool
 
+    @property
+    def has_default_factory(self) -> bool:
+        return self.default_factory is not self._missing_type
 
-if IS_PYDANTIC_V2:
-    from typing_extensions import get_args, get_origin
+    @property
+    def has_default(self) -> bool:
+        return self.default is not self._missing_type
 
-    from pydantic._internal._typing_extra import is_new_type
-    from pydantic._internal._utils import lenient_issubclass, smart_deepcopy
-    from pydantic_core import PydanticUndefined
 
-    PYDANTIC_MISSING_TYPE = PydanticUndefined
+ATTR_TO_TYPE_MAP = {
+    "NoneStr": Optional[str],
+    "NoneBytes": Optional[bytes],
+    "StrBytes": None,
+    "NoneStrBytes": None,
+    "StrictStr": str,
+    "ConstrainedBytes": bytes,
+    "conbytes": bytes,
+    "ConstrainedStr": str,
+    "constr": str,
+    "EmailStr": str,
+    "PyObject": None,
+    "ConstrainedInt": int,
+    "conint": int,
+    "PositiveInt": int,
+    "NegativeInt": int,
+    "ConstrainedFloat": float,
+    "confloat": float,
+    "PositiveFloat": float,
+    "NegativeFloat": float,
+    "ConstrainedDecimal": Decimal,
+    "condecimal": Decimal,
+    "UUID1": UUID,
+    "UUID3": UUID,
+    "UUID4": UUID,
+    "UUID5": UUID,
+    "FilePath": None,
+    "DirectoryPath": None,
+    "Json": None,
+    "JsonWrapper": None,
+    "SecretStr": str,
+    "SecretBytes": bytes,
+    "StrictBool": bool,
+    "StrictInt": int,
+    "StrictFloat": float,
+    "PaymentCardNumber": None,
+    "ByteSize": None,
+    "AnyUrl": str,
+    "AnyHttpUrl": str,
+    "HttpUrl": str,
+    "PostgresDsn": str,
+    "RedisDsn": str,
+}
+
+ATTR_TO_TYPE_MAP_Pydantic_V2 = {
+    "EmailStr": str,
+    "SecretStr": str,
+    "SecretBytes": bytes,
+    "AnyUrl": str,
+}
+
+ATTR_TO_TYPE_MAP_Pydantic_Core_V2 = {
+    "MultiHostUrl": str,
+}
 
-    def new_type_supertype(type_: Any) -> Any:
-        return type_.__supertype__
 
-    def get_model_fields(model: Type[BaseModel]) -> Dict[str, CompatModelField]:
+def get_fields_map_for_v2() -> Dict[Any, Any]:
+    import pydantic_core
+
+    fields_map = {
+        getattr(pydantic, field_name): type
+        for field_name, type in ATTR_TO_TYPE_MAP_Pydantic_V2.items()
+        if hasattr(pydantic, field_name)
+    }
+    fields_map.update(
+        {
+            getattr(pydantic_core, field_name): type
+            for field_name, type in ATTR_TO_TYPE_MAP_Pydantic_Core_V2.items()
+            if hasattr(pydantic_core, field_name)
+        }
+    )
+
+    return fields_map
+
+
+class PydanticV2Compat:
+    @property
+    def PYDANTIC_MISSING_TYPE(self) -> Any:
+        from pydantic_core import PydanticUndefined
+
+        return PydanticUndefined
+
+    def get_model_fields(self, model: Type[BaseModel]) -> Dict[str, CompatModelField]:
         field_info: dict[str, FieldInfo] = model.model_fields
         new_fields = {}
         # Convert it into CompatModelField
@@ -55,24 +141,34 @@ if IS_PYDANTIC_V2:
                 allow_none=False,
                 has_alias=field is not None,
                 description=field.description,
+                _missing_type=self.PYDANTIC_MISSING_TYPE,
+                is_v1=False,
             )
         return new_fields
 
-else:
-    from pydantic.typing import (  # type: ignore[no-redef]
-        get_args,
-        get_origin,
-        is_new_type,
-        new_type_supertype,
-    )
-    from pydantic.utils import (  # type: ignore[no-redef]
-        lenient_issubclass,
-        smart_deepcopy,
-    )
+    @cached_property
+    def fields_map(self) -> Dict[Any, Any]:
+        return get_fields_map_for_v2()
 
-    PYDANTIC_MISSING_TYPE = dataclasses.MISSING  # type: ignore[assignment]
+    def get_basic_type(self, type_: Any) -> Type[Any]:
+        if type_ in self.fields_map:
+            type_ = self.fields_map[type_]
 
-    def get_model_fields(model: Type[BaseModel]) -> Dict[str, CompatModelField]:
+            if type_ is None:
+                raise UnsupportedTypeError()
+
+        if is_new_type(type_):
+            return new_type_supertype(type_)
+
+        return type_
+
+
+class PydanticV1Compat:
+    @property
+    def PYDANTIC_MISSING_TYPE(self) -> Any:
+        return dataclasses.MISSING
+
+    def get_model_fields(self, model: Type[BaseModel]) -> Dict[str, CompatModelField]:
         new_fields = {}
         # Convert it into CompatModelField
         for name, field in model.__fields__.items():  # type: ignore[attr-defined]
@@ -87,17 +183,104 @@ else:
                 allow_none=field.allow_none,
                 has_alias=field.has_alias,
                 description=field.field_info.description,
+                _missing_type=self.PYDANTIC_MISSING_TYPE,
+                is_v1=True,
             )
         return new_fields
 
+    @cached_property
+    def fields_map(self) -> Dict[Any, Any]:
+        if IS_PYDANTIC_V2:
+            return {
+                getattr(pydantic.v1, field_name): type
+                for field_name, type in ATTR_TO_TYPE_MAP.items()
+                if hasattr(pydantic.v1, field_name)
+            }
+
+        return {
+            getattr(pydantic, field_name): type
+            for field_name, type in ATTR_TO_TYPE_MAP.items()
+            if hasattr(pydantic, field_name)
+        }
+
+    def get_basic_type(self, type_: Any) -> Type[Any]:
+        if IS_PYDANTIC_V1:
+            ConstrainedInt = pydantic.ConstrainedInt
+            ConstrainedFloat = pydantic.ConstrainedFloat
+            ConstrainedStr = pydantic.ConstrainedStr
+            ConstrainedList = pydantic.ConstrainedList
+        else:
+            ConstrainedInt = pydantic.v1.ConstrainedInt
+            ConstrainedFloat = pydantic.v1.ConstrainedFloat
+            ConstrainedStr = pydantic.v1.ConstrainedStr
+            ConstrainedList = pydantic.v1.ConstrainedList
+
+        if lenient_issubclass(type_, ConstrainedInt):
+            return int
+        if lenient_issubclass(type_, ConstrainedFloat):
+            return float
+        if lenient_issubclass(type_, ConstrainedStr):
+            return str
+        if lenient_issubclass(type_, ConstrainedList):
+            return List[self.get_basic_type(type_.item_type)]  # type: ignore
+
+        if type_ in self.fields_map:
+            type_ = self.fields_map[type_]
+
+            if type_ is None:
+                raise UnsupportedTypeError()
+
+        if is_new_type(type_):
+            return new_type_supertype(type_)
+
+        return type_
+
+
+class PydanticCompat:
+    def __init__(self, is_v2: bool):
+        if is_v2:
+            self._compat = PydanticV2Compat()
+        else:
+            self._compat = PydanticV1Compat()  # type: ignore[assignment]
+
+    @classmethod
+    def from_model(cls, model: Type[BaseModel]) -> "PydanticCompat":
+        if hasattr(model, "model_fields"):
+            return cls(is_v2=True)
+
+        return cls(is_v2=False)
+
+    def __getattr__(self, name: str) -> Any:
+        return getattr(self._compat, name)
+
+
+if IS_PYDANTIC_V2:
+    from typing_extensions import get_args, get_origin
+
+    from pydantic._internal._typing_extra import is_new_type
+    from pydantic._internal._utils import lenient_issubclass, smart_deepcopy
+
+    def new_type_supertype(type_: Any) -> Any:
+        return type_.__supertype__
+else:
+    from pydantic.typing import (  # type: ignore[no-redef]
+        get_args,
+        get_origin,
+        is_new_type,
+        new_type_supertype,
+    )
+    from pydantic.utils import (  # type: ignore[no-redef]
+        lenient_issubclass,
+        smart_deepcopy,
+    )
+
 
 __all__ = [
-    "smart_deepcopy",
+    "PydanticCompat",
+    "is_new_type",
     "lenient_issubclass",
-    "get_args",
     "get_origin",
-    "is_new_type",
+    "get_args",
     "new_type_supertype",
-    "get_model_fields",
-    "PYDANTIC_MISSING_TYPE",
+    "smart_deepcopy",
 ]

strawberry/experimental/pydantic/error_type.py

@@ -19,7 +19,7 @@ from pydantic import BaseModel
 from strawberry.auto import StrawberryAuto
 from strawberry.experimental.pydantic._compat import (
     CompatModelField,
-    get_model_fields,
+    PydanticCompat,
     lenient_issubclass,
 )
 from strawberry.experimental.pydantic.utils import (
@@ -72,7 +72,8 @@ def error_type(
     all_fields: bool = False,
 ) -> Callable[..., Type]:
     def wrap(cls: Type) -> Type:
-        model_fields = get_model_fields(model)
+        compat = PydanticCompat.from_model(model)
+        model_fields = compat.get_model_fields(model)
         fields_set = set(fields) if fields else set()
 
         if fields:

strawberry/experimental/pydantic/fields.py

@@ -1,23 +1,17 @@
 import builtins
-from decimal import Decimal
-from typing import Any, Dict, List, Optional, Type, Union
+from typing import Any, Union
 from typing_extensions import Annotated
-from uuid import UUID
 
-import pydantic
 from pydantic import BaseModel
 
 from strawberry.experimental.pydantic._compat import (
-    IS_PYDANTIC_V1,
+    PydanticCompat,
     get_args,
     get_origin,
-    is_new_type,
     lenient_issubclass,
-    new_type_supertype,
 )
 from strawberry.experimental.pydantic.exceptions import (
     UnregisteredTypeException,
-    UnsupportedTypeError,
 )
 from strawberry.types.types import StrawberryObjectDefinition
 
@@ -44,115 +38,6 @@ except ImportError:
         raise
 
 
-ATTR_TO_TYPE_MAP = {
-    "NoneStr": Optional[str],
-    "NoneBytes": Optional[bytes],
-    "StrBytes": None,
-    "NoneStrBytes": None,
-    "StrictStr": str,
-    "ConstrainedBytes": bytes,
-    "conbytes": bytes,
-    "ConstrainedStr": str,
-    "constr": str,
-    "EmailStr": str,
-    "PyObject": None,
-    "ConstrainedInt": int,
-    "conint": int,
-    "PositiveInt": int,
-    "NegativeInt": int,
-    "ConstrainedFloat": float,
-    "confloat": float,
-    "PositiveFloat": float,
-    "NegativeFloat": float,
-    "ConstrainedDecimal": Decimal,
-    "condecimal": Decimal,
-    "UUID1": UUID,
-    "UUID3": UUID,
-    "UUID4": UUID,
-    "UUID5": UUID,
-    "FilePath": None,
-    "DirectoryPath": None,
-    "Json": None,
-    "JsonWrapper": None,
-    "SecretStr": str,
-    "SecretBytes": bytes,
-    "StrictBool": bool,
-    "StrictInt": int,
-    "StrictFloat": float,
-    "PaymentCardNumber": None,
-    "ByteSize": None,
-    "AnyUrl": str,
-    "AnyHttpUrl": str,
-    "HttpUrl": str,
-    "PostgresDsn": str,
-    "RedisDsn": str,
-}
-
-ATTR_TO_TYPE_MAP_Pydantic_V2 = {
-    "EmailStr": str,
-    "SecretStr": str,
-    "SecretBytes": bytes,
-    "AnyUrl": str,
-}
-
-ATTR_TO_TYPE_MAP_Pydantic_Core_V2 = {
-    "MultiHostUrl": str,
-}
-
-
-def get_fields_map_for_v2() -> Dict[Any, Any]:
-    import pydantic_core
-
-    fields_map = {
-        getattr(pydantic, field_name): type
-        for field_name, type in ATTR_TO_TYPE_MAP_Pydantic_V2.items()
-        if hasattr(pydantic, field_name)
-    }
-    fields_map.update(
-        {
-            getattr(pydantic_core, field_name): type
-            for field_name, type in ATTR_TO_TYPE_MAP_Pydantic_Core_V2.items()
-            if hasattr(pydantic_core, field_name)
-        }
-    )
-
-    return fields_map
-
-
-FIELDS_MAP = (
-    {
-        getattr(pydantic, field_name): type
-        for field_name, type in ATTR_TO_TYPE_MAP.items()
-        if hasattr(pydantic, field_name)
-    }
-    if IS_PYDANTIC_V1
-    else get_fields_map_for_v2()
-)
-
-
-def get_basic_type(type_: Any) -> Type[Any]:
-    if IS_PYDANTIC_V1:
-        # only pydantic v1 has these
-        if lenient_issubclass(type_, pydantic.ConstrainedInt):
-            return int
-        if lenient_issubclass(type_, pydantic.ConstrainedFloat):
-            return float
-        if lenient_issubclass(type_, pydantic.ConstrainedStr):
-            return str
-        if lenient_issubclass(type_, pydantic.ConstrainedList):
-            return List[get_basic_type(type_.item_type)]  # type: ignore
-
-    if type_ in FIELDS_MAP:
-        type_ = FIELDS_MAP.get(type_)
-        if type_ is None:
-            raise UnsupportedTypeError()
-
-    if is_new_type(type_):
-        return new_type_supertype(type_)
-
-    return type_
-
-
 def replace_pydantic_types(type_: Any, is_input: bool) -> Any:
     if lenient_issubclass(type_, BaseModel):
         attr = "_strawberry_input_type" if is_input else "_strawberry_type"
@@ -163,17 +48,21 @@ def replace_pydantic_types(type_: Any, is_input: bool) -> Any:
     return type_
 
 
-def replace_types_recursively(type_: Any, is_input: bool) -> Any:
+def replace_types_recursively(
+    type_: Any, is_input: bool, compat: PydanticCompat
+) -> Any:
     """Runs the conversions recursively into the arguments of generic types if any"""
-    basic_type = get_basic_type(type_)
+    basic_type = compat.get_basic_type(type_)
     replaced_type = replace_pydantic_types(basic_type, is_input)
 
     origin = get_origin(type_)
+
     if not origin or not hasattr(type_, "__args__"):
         return replaced_type
 
     converted = tuple(
-        replace_types_recursively(t, is_input=is_input) for t in get_args(replaced_type)
+        replace_types_recursively(t, is_input=is_input, compat=compat)
+        for t in get_args(replaced_type)
     )
 
     if isinstance(replaced_type, TypingGenericAlias):

strawberry/experimental/pydantic/object_type.py

@@ -19,9 +19,8 @@ from typing import (
 from strawberry.annotation import StrawberryAnnotation
 from strawberry.auto import StrawberryAuto
 from strawberry.experimental.pydantic._compat import (
-    IS_PYDANTIC_V1,
     CompatModelField,
-    get_model_fields,
+    PydanticCompat,
 )
 from strawberry.experimental.pydantic.conversion import (
     convert_pydantic_model_to_strawberry_class,
@@ -44,10 +43,12 @@ if TYPE_CHECKING:
     from graphql import GraphQLResolveInfo
 
 
-def get_type_for_field(field: CompatModelField, is_input: bool):  # noqa: ANN201
+def get_type_for_field(field: CompatModelField, is_input: bool, compat: PydanticCompat):  # noqa: ANN201
     outer_type = field.outer_type_
-    replaced_type = replace_types_recursively(outer_type, is_input)
-    if IS_PYDANTIC_V1:
+
+    replaced_type = replace_types_recursively(outer_type, is_input, compat=compat)
+
+    if field.is_v1:
         # only pydantic v1 has this Optional logic
         should_add_optional: bool = field.allow_none
         if should_add_optional:
@@ -62,9 +63,10 @@ def _build_dataclass_creation_fields(
     existing_fields: Dict[str, StrawberryField],
     auto_fields_set: Set[str],
     use_pydantic_alias: bool,
+    compat: PydanticCompat,
 ) -> DataclassCreationFields:
     field_type = (
-        get_type_for_field(field, is_input)
+        get_type_for_field(field, is_input, compat=compat)
         if field.name in auto_fields_set
         else existing_fields[field.name].type
     )
@@ -129,7 +131,8 @@ def type(
     use_pydantic_alias: bool = True,
 ) -> Callable[..., Type[StrawberryTypeFromPydantic[PydanticModel]]]:
     def wrap(cls: Any) -> Type[StrawberryTypeFromPydantic[PydanticModel]]:
-        model_fields = get_model_fields(model)
+        compat = PydanticCompat.from_model(model)
+        model_fields = compat.get_model_fields(model)
         original_fields_set = set(fields) if fields else set()
 
         if fields:
@@ -182,7 +185,12 @@ def type(
 
         all_model_fields: List[DataclassCreationFields] = [
             _build_dataclass_creation_fields(
-                field, is_input, extra_fields_dict, auto_fields_set, use_pydantic_alias
+                field,
+                is_input,
+                extra_fields_dict,
+                auto_fields_set,
+                use_pydantic_alias,
+                compat=compat,
             )
             for field_name, field in model_fields.items()
             if field_name in fields_set

strawberry/experimental/pydantic/utils.py

@@ -14,9 +14,8 @@ from typing import (
 )
 
 from strawberry.experimental.pydantic._compat import (
-    PYDANTIC_MISSING_TYPE,
     CompatModelField,
-    get_model_fields,
+    PydanticCompat,
     smart_deepcopy,
 )
 from strawberry.experimental.pydantic.exceptions import (
@@ -83,12 +82,8 @@ def get_default_factory_for_field(
     Returns optionally a NoArgAnyCallable representing a default_factory parameter
     """
     # replace dataclasses.MISSING with our own UNSET to make comparisons easier
-    default_factory = (
-        field.default_factory
-        if field.default_factory is not PYDANTIC_MISSING_TYPE
-        else UNSET
-    )
-    default = field.default if field.default is not PYDANTIC_MISSING_TYPE else UNSET
+    default_factory = field.default_factory if field.has_default_factory else UNSET
+    default = field.default if field.has_default else UNSET
 
     has_factory = default_factory is not None and default_factory is not UNSET
     has_default = default is not None and default is not UNSET
@@ -126,8 +121,9 @@ def get_default_factory_for_field(
 def ensure_all_auto_fields_in_pydantic(
     model: Type[BaseModel], auto_fields: Set[str], cls_name: str
 ) -> None:
+    compat = PydanticCompat.from_model(model)
     # Raise error if user defined a strawberry.auto field not present in the model
-    non_existing_fields = list(auto_fields - get_model_fields(model).keys())
+    non_existing_fields = list(auto_fields - compat.get_model_fields(model).keys())
 
     if non_existing_fields:
         raise AutoFieldsNotInBaseModelError(

strawberry/printer/printer.py

@@ -74,7 +74,9 @@ def _serialize_dataclasses(value: Dict[_T, object]) -> Dict[_T, object]:
 
 
 @overload
-def _serialize_dataclasses(value: Union[List[object], Tuple[object]]) -> List[object]:
+def _serialize_dataclasses(
+    value: Union[List[object], Tuple[object]],
+) -> List[object]:
     ...
 
 

strawberry/schema_codegen/__init__.py

@@ -2,6 +2,9 @@ from __future__ import annotations
 
 import dataclasses
 import keyword
+from collections import defaultdict
+from typing import TYPE_CHECKING, List, Tuple, Union
+from typing_extensions import Protocol, TypeAlias
 
 import libcst as cst
 from graphql import (
@@ -19,14 +22,28 @@ from graphql import (
     OperationType,
     ScalarTypeDefinitionNode,
     SchemaDefinitionNode,
+    SchemaExtensionNode,
     StringValueNode,
     TypeNode,
     UnionTypeDefinitionNode,
     parse,
 )
+from graphql.language.ast import (
+    BooleanValueNode,
+    ConstValueNode,
+    ListValueNode,
+)
 
 from strawberry.utils.str_converters import to_snake_case
 
+if TYPE_CHECKING:
+    from graphql.language.ast import ConstDirectiveNode
+
+
+class HasDirectives(Protocol):
+    directives: Tuple[ConstDirectiveNode, ...]
+
+
 _SCALAR_MAP = {
     "Int": cst.Name("int"),
     "Float": cst.Name("float"),
@@ -48,6 +65,48 @@ _SCALAR_MAP = {
 }
 
 
+@dataclasses.dataclass(frozen=True)
+class Import:
+    module: str | None
+    imports: tuple[str]
+
+    def module_path_to_cst(self, module_path: str) -> cst.Name | cst.Attribute:
+        parts = module_path.split(".")
+
+        module_name: cst.Name | cst.Attribute = cst.Name(parts[0])
+
+        for part in parts[1:]:
+            module_name = cst.Attribute(value=module_name, attr=cst.Name(part))
+
+        return module_name
+
+    def to_cst(self) -> cst.Import | cst.ImportFrom:
+        if self.module is None:
+            return cst.Import(
+                names=[cst.ImportAlias(name=cst.Name(name)) for name in self.imports]
+            )
+
+        return cst.ImportFrom(
+            module=self.module_path_to_cst(self.module),
+            names=[cst.ImportAlias(name=cst.Name(name)) for name in self.imports],
+        )
+
+
+def _is_federation_link_directive(directive: ConstDirectiveNode) -> bool:
+    if directive.name.value != "link":
+        return False
+
+    return next(
+        (
+            argument.value.value
+            for argument in directive.arguments
+            if argument.name.value == "url"
+            if isinstance(argument.value, StringValueNode)
+        ),
+        "",
+    ).startswith("https://specs.apollo.dev/federation")
+
+
 def _get_field_type(
     field_type: TypeNode, was_non_nullable: bool = False
 ) -> cst.BaseExpression:
@@ -85,7 +144,19 @@ def _get_field_type(
     )
 
 
-def _get_argument(name: str, value: str) -> cst.Arg:
+def _sanitize_argument(value: ArgumentValue) -> cst.SimpleString | cst.Name | cst.List:
+    if isinstance(value, bool):
+        return cst.Name(value=str(value))
+
+    if isinstance(value, list):
+        return cst.List(
+            elements=[
+                cst.Element(value=_sanitize_argument(item))
+                for item in value
+                if item is not None
+            ],
+        )
+
     if "\n" in value:
         argument_value = cst.SimpleString(f'"""\n{value}\n"""')
     elif '"' in value:
@@ -93,6 +164,12 @@ def _get_argument(name: str, value: str) -> cst.Arg:
     else:
         argument_value = cst.SimpleString(f'"{value}"')
 
+    return argument_value
+
+
+def _get_argument(name: str, value: ArgumentValue) -> cst.Arg:
+    argument_value = _sanitize_argument(value)
+
     return cst.Arg(
         value=argument_value,
         keyword=cst.Name(name),
@@ -100,7 +177,27 @@ def _get_argument(name: str, value: str) -> cst.Arg:
     )
 
 
-def _get_field_value(description: str | None, alias: str | None) -> cst.Call | None:
+# TODO: this might be removed now
+def _get_argument_list(name: str, values: list[ArgumentValue]) -> cst.Arg:
+    value = cst.List(
+        elements=[cst.Element(value=_sanitize_argument(value)) for value in values],
+    )
+
+    return cst.Arg(
+        value=value,
+        keyword=cst.Name(name),
+        equal=cst.AssignEqual(cst.SimpleWhitespace(""), cst.SimpleWhitespace("")),
+    )
+
+
+def _get_field_value(
+    field: FieldDefinitionNode | InputValueDefinitionNode,
+    alias: str | None,
+    is_apollo_federation: bool,
+    imports: set[Import],
+) -> cst.Call | None:
+    description = field.description.value if field.description else None
+
     args = list(
         filter(
             None,
@@ -111,6 +208,24 @@ def _get_field_value(description: str | None, alias: str | None) -> cst.Call | N
         )
     )
 
+    directives = _get_directives(field)
+
+    apollo_federation_args = _get_federation_arguments(directives, imports)
+
+    if is_apollo_federation and apollo_federation_args:
+        args.extend(apollo_federation_args)
+
+        return cst.Call(
+            func=cst.Attribute(
+                value=cst.Attribute(
+                    value=cst.Name("strawberry"),
+                    attr=cst.Name("federation"),
+                ),
+                attr=cst.Name("field"),
+            ),
+            args=args,
+        )
+
     if args:
         return cst.Call(
             func=cst.Attribute(
@@ -125,6 +240,8 @@ def _get_field_value(description: str | None, alias: str | None) -> cst.Call | N
 
 def _get_field(
     field: FieldDefinitionNode | InputValueDefinitionNode,
+    is_apollo_federation: bool,
+    imports: set[Import],
 ) -> cst.SimpleStatementLine:
     name = to_snake_case(field.name.value)
     alias: str | None = None
@@ -141,19 +258,131 @@ def _get_field(
                     _get_field_type(field.type),
                 ),
                 value=_get_field_value(
-                    description=field.description.value if field.description else None,
-                    alias=alias if alias != name else None,
+                    field,
+                    alias=alias,
+                    is_apollo_federation=is_apollo_federation,
+                    imports=imports,
                 ),
             )
         ]
     )
 
 
+ArgumentValue: TypeAlias = Union[str, bool, List["ArgumentValue"]]
+
+
+def _get_argument_value(argument_value: ConstValueNode) -> ArgumentValue:
+    if isinstance(argument_value, StringValueNode):
+        return argument_value.value
+    elif isinstance(argument_value, EnumValueDefinitionNode):
+        return argument_value.name.value
+    elif isinstance(argument_value, ListValueNode):
+        return [_get_argument_value(arg) for arg in argument_value.values]
+    elif isinstance(argument_value, BooleanValueNode):
+        return argument_value.value
+    else:
+        raise NotImplementedError(f"Unknown argument value {argument_value}")
+
+
+def _get_directives(
+    definition: HasDirectives,
+) -> dict[str, list[dict[str, ArgumentValue]]]:
+    directives: dict[str, list[dict[str, ArgumentValue]]] = defaultdict(list)
+
+    for directive in definition.directives:
+        directive_name = directive.name.value
+
+        directives[directive_name].append(
+            {
+                argument.name.value: _get_argument_value(argument.value)
+                for argument in directive.arguments
+            }
+        )
+
+    return directives
+
+
+def _get_federation_arguments(
+    directives: dict[str, list[dict[str, ArgumentValue]]],
+    imports: set[Import],
+) -> list[cst.Arg]:
+    def append_arg_from_directive(
+        directive: str,
+        argument_name: str,
+        keyword_name: str | None = None,
+        flatten: bool = True,
+    ):
+        keyword_name = keyword_name or directive
+
+        if directive in directives:
+            values = [item[argument_name] for item in directives[directive]]
+
+            if flatten:
+                arguments.append(_get_argument(keyword_name, values))
+            else:
+                arguments.extend(_get_argument(keyword_name, value) for value in values)
+
+    arguments: list[cst.Arg] = []
+
+    append_arg_from_directive("key", "fields", "keys")
+    append_arg_from_directive("requires", "fields")
+    append_arg_from_directive("provides", "fields")
+    append_arg_from_directive(
+        "requiresScopes", "scopes", "requires_scopes", flatten=False
+    )
+    append_arg_from_directive("policy", "policies", "policy", flatten=False)
+    append_arg_from_directive("tag", "name", "tags")
+
+    boolean_keys = (
+        "shareable",
+        "inaccessible",
+        "external",
+        "authenticated",
+    )
+
+    arguments.extend(
+        _get_argument(key, True) for key in boolean_keys if directives.get(key, False)
+    )
+
+    if overrides := directives.get("override"):
+        override = overrides[0]
+
+        if "label" not in override:
+            arguments.append(_get_argument("override", override["from"]))
+        else:
+            imports.add(
+                Import(
+                    module="strawberry.federation.schema_directives",
+                    imports=("Override",),
+                )
+            )
+
+            arguments.append(
+                cst.Arg(
+                    keyword=cst.Name("override"),
+                    value=cst.Call(
+                        func=cst.Name("Override"),
+                        args=[
+                            _get_argument("override_from", override["from"]),
+                            _get_argument("label", override["label"]),
+                        ],
+                    ),
+                    equal=cst.AssignEqual(
+                        cst.SimpleWhitespace(""), cst.SimpleWhitespace("")
+                    ),
+                )
+            )
+
+    return arguments
+
+
 def _get_strawberry_decorator(
     definition: ObjectTypeDefinitionNode
     | ObjectTypeExtensionNode
     | InterfaceTypeDefinitionNode
     | InputObjectTypeDefinitionNode,
+    is_apollo_federation: bool,
+    imports: set[Import],
 ) -> cst.Decorator:
     type_ = {
         ObjectTypeDefinitionNode: "type",
@@ -168,15 +397,36 @@ def _get_strawberry_decorator(
         else None
     )
 
+    directives = _get_directives(definition)
+
     decorator: cst.BaseExpression = cst.Attribute(
         value=cst.Name("strawberry"),
         attr=cst.Name(type_),
     )
 
+    arguments: list[cst.Arg] = []
+
     if description is not None:
+        arguments.append(_get_argument("description", description.value))
+
+    federation_arguments = _get_federation_arguments(directives, imports)
+
+    # and has any directive that is a federation directive
+    if is_apollo_federation and federation_arguments:
+        decorator = cst.Attribute(
+            value=cst.Attribute(
+                value=cst.Name("strawberry"),
+                attr=cst.Name("federation"),
+            ),
+            attr=cst.Name(type_),
+        )
+
+        arguments.extend(federation_arguments)
+
+    if arguments:
         decorator = cst.Call(
             func=decorator,
-            args=[_get_argument("description", description.value)],
+            args=arguments,
         )
 
     return cst.Decorator(
@@ -189,8 +439,10 @@ def _get_class_definition(
     | ObjectTypeExtensionNode
     | InterfaceTypeDefinitionNode
     | InputObjectTypeDefinitionNode,
+    is_apollo_federation: bool,
+    imports: set[Import],
 ) -> cst.ClassDef:
-    decorator = _get_strawberry_decorator(definition)
+    decorator = _get_strawberry_decorator(definition, is_apollo_federation, imports)
 
     bases = (
         [cst.Arg(cst.Name(interface.name.value)) for interface in definition.interfaces]
@@ -204,7 +456,12 @@ def _get_class_definition(
     return cst.ClassDef(
         name=cst.Name(definition.name.value),
         bases=bases,
-        body=cst.IndentedBlock(body=[_get_field(field) for field in definition.fields]),
+        body=cst.IndentedBlock(
+            body=[
+                _get_field(field, is_apollo_federation, imports)
+                for field in definition.fields
+            ]
+        ),
         decorators=[decorator],
     )
 
@@ -243,6 +500,7 @@ def _get_schema_definition(
     root_query_name: str | None,
     root_mutation_name: str | None,
     root_subscription_name: str | None,
+    is_apollo_federation: bool,
 ) -> cst.SimpleStatementLine | None:
     if not any([root_query_name, root_mutation_name, root_subscription_name]):
         return None
@@ -265,39 +523,45 @@ def _get_schema_definition(
     if root_subscription_name:
         args.append(_get_arg("subscription", root_subscription_name))
 
+    schema_call = cst.Call(
+        func=cst.Attribute(
+            value=cst.Name("strawberry"),
+            attr=cst.Name("Schema"),
+        ),
+        args=args,
+    )
+
+    if is_apollo_federation:
+        args.append(
+            cst.Arg(
+                keyword=cst.Name("enable_federation_2"),
+                value=cst.Name("True"),
+                equal=cst.AssignEqual(
+                    cst.SimpleWhitespace(""), cst.SimpleWhitespace("")
+                ),
+            )
+        )
+        schema_call = cst.Call(
+            func=cst.Attribute(
+                value=cst.Attribute(
+                    value=cst.Name(value="strawberry"),
+                    attr=cst.Name(value="federation"),
+                ),
+                attr=cst.Name(value="Schema"),
+            ),
+            args=args,
+        )
+
     return cst.SimpleStatementLine(
         body=[
             cst.Assign(
                 targets=[cst.AssignTarget(cst.Name("schema"))],
-                value=cst.Call(
-                    func=cst.Attribute(
-                        value=cst.Name("strawberry"),
-                        attr=cst.Name("Schema"),
-                    ),
-                    args=args,
-                ),
+                value=schema_call,
             )
         ]
     )
 
 
-@dataclasses.dataclass(frozen=True)
-class Import:
-    module: str | None
-    imports: tuple[str]
-
-    def to_cst(self) -> cst.Import | cst.ImportFrom:
-        if self.module is None:
-            return cst.Import(
-                names=[cst.ImportAlias(name=cst.Name(name)) for name in self.imports]
-            )
-
-        return cst.ImportFrom(
-            module=cst.Name(self.module),
-            names=[cst.ImportAlias(name=cst.Name(name)) for name in self.imports],
-        )
-
-
 def _get_union_definition(definition: UnionTypeDefinitionNode) -> cst.Assign:
     name = definition.name.value
 
@@ -430,6 +694,12 @@ def codegen(schema: str) -> str:
 
     object_types: dict[str, cst.ClassDef] = {}
 
+    # when we encounter a extend schema @link ..., we check if is an apollo federation schema
+    # and we use this variable to keep track of it, but at the moment the assumption is that
+    # the schema extension is always done at the top, this might not be the case all the
+    # time
+    is_apollo_federation = False
+
     for definition in document.definitions:
         if isinstance(
             definition,
@@ -440,7 +710,9 @@ def codegen(schema: str) -> str:
                 ObjectTypeExtensionNode,
             ),
         ):
-            class_definition = _get_class_definition(definition)
+            class_definition = _get_class_definition(
+                definition, is_apollo_federation, imports
+            )
 
             object_types[definition.name.value] = class_definition
 
@@ -478,6 +750,11 @@ def codegen(schema: str) -> str:
                 definitions.append(cst.EmptyLine())
                 definitions.append(scalar_definition)
                 definitions.append(cst.EmptyLine())
+        elif isinstance(definition, SchemaExtensionNode):
+            is_apollo_federation = any(
+                _is_federation_link_directive(directive)
+                for directive in definition.directives
+            )
         else:
             raise NotImplementedError(f"Unknown definition {definition}")
 
@@ -496,6 +773,7 @@ def codegen(schema: str) -> str:
         root_query_name=root_query_name,
         root_mutation_name=root_mutation_name,
         root_subscription_name=root_subscription_name,
+        is_apollo_federation=is_apollo_federation,
     )
 
     if schema_definition:

strawberry/types/nodes.py

@@ -9,6 +9,7 @@ If a node has only one useful value, it's value is inlined.
 If a list of nodes have unique names, it's transformed into a mapping.
 Note Python dicts maintain ordering (for all supported versions).
 """
+
 from __future__ import annotations
 
 import dataclasses

{strawberry_graphql-0.222.0.dist-info → strawberry_graphql-0.224.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: strawberry-graphql
-Version: 0.222.0
+Version: 0.224.0
 Summary: A library for creating GraphQL APIs
 Home-page: https://strawberry.rocks/
 License: MIT

{strawberry_graphql-0.222.0.dist-info → strawberry_graphql-0.224.0.dist-info}/RECORD

@@ -22,7 +22,7 @@ strawberry/channels/__init__.py,sha256=9oRdAT7uIYETF-23gZ5NteOXSjwkUtwRmwu3YCFv1
 strawberry/channels/handlers/base.py,sha256=tgXLjKCHrP-8yOJV-qo9iOMJUgDe2wW6rGs_nDZFi80,7869
 strawberry/channels/handlers/graphql_transport_ws_handler.py,sha256=UKfxRoo1pEv1CBHhkhSTkbztJlfU-OoksFwyLktj8xc,1945
 strawberry/channels/handlers/graphql_ws_handler.py,sha256=PHRkwnXt3tY4E0XBVHh4hpTyFBt9jQfTreBUmEo9vhI,2497
-strawberry/channels/handlers/http_handler.py,sha256=9pW978XaeF-aFWM9WMaSHCOWmcWoIJCNkW8X3lKJcws,9560
+strawberry/channels/handlers/http_handler.py,sha256=c1Jpf7L4b7LvSaOolP2e9TWcEM5IqHXhScK0kZrhSg8,9561
 strawberry/channels/handlers/ws_handler.py,sha256=sHL44eay4tNoKzkrRn3WewSYH-3ZSJzxJpmBJ-aTkeM,4650
 strawberry/channels/router.py,sha256=dyOBbSF8nFiygP0zz6MM14mhkvFQAEbbLBXzcpubSHM,1927
 strawberry/channels/testing.py,sha256=IWj1CuIS3vOo2f2fw0W-0GCz-YSs7QSAAscC6suqtiI,5668
@@ -81,16 +81,16 @@ strawberry/exceptions/syntax.py,sha256=OCz2Ai1Yn0jonRZXCDI2h30tPNsa8L5wMU8vcqIO4
 strawberry/exceptions/unresolved_field_type.py,sha256=voYLO7kj_8DW0r3jvwn82HOqFIWk_3jvoPCciXFv5JA,1857
 strawberry/exceptions/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 strawberry/exceptions/utils/source_finder.py,sha256=AcuevLgWwOSOR5zBDJbVyXMGRUxsTzrr07XUB0UqNiI,20044
-strawberry/experimental/__init__.py,sha256=n6Ytr6wBgXt47jLY8rEXQs7G4gZEkm9BCLL0Q8opWfw,94
+strawberry/experimental/__init__.py,sha256=2HP5XtxL8ZKsPp4EDRAbMCqiP7p2V4Cca278JUGxnt0,102
 strawberry/experimental/pydantic/__init__.py,sha256=jlsYH1j_9W4ieRpUKgt5zQPERDL7nc1ZBNQ3dAhJs8w,241
-strawberry/experimental/pydantic/_compat.py,sha256=ofbkbF6wSlBedCKj3qprAgcCfWlXdGL5iUZo355uAvA,3137
+strawberry/experimental/pydantic/_compat.py,sha256=sBWil9Ggf-OqfK35XB-0-rx9MS66QFVEqptPkF3sjC0,8139
 strawberry/experimental/pydantic/conversion.py,sha256=_YMz4cYCwLUazkT2oCQ4B1gusksnXDHdFm1DldIrU7Q,4213
 strawberry/experimental/pydantic/conversion_types.py,sha256=psVRmKE5fVcBAUT0Z2vCOLAXHddat2A0jJ-0SB1XDn8,963
-strawberry/experimental/pydantic/error_type.py,sha256=Fy7m2E03a86XsF76UoAUeroiJLnvVxBwKKEt6RcMfKs,4311
+strawberry/experimental/pydantic/error_type.py,sha256=s8v0weIVM-9c0e3zqySaaa3pGZan7RvKTGuTtXLGPmE,4366
 strawberry/experimental/pydantic/exceptions.py,sha256=GJt22DTolA24A0ZHcHsy-fcRkBnavgUZYOo0MBIVcjs,1427
-strawberry/experimental/pydantic/fields.py,sha256=m7LJEQ39wko4NRIjYPxhUbwNyZcubrbZmyBumG34CJM,5356
-strawberry/experimental/pydantic/object_type.py,sha256=kb1gwHw7vQCshk84a7DMRBp_9zDntrAK3bbI58vjqf0,12160
-strawberry/experimental/pydantic/utils.py,sha256=vh9Mz33b5QhomuCMVygZ2mp2jrBOxVejIRK0XdfpygI,3828
+strawberry/experimental/pydantic/fields.py,sha256=QP5vYb8vKf7vOC69cjUPGOdlO_j4kZRdxalWgeqJ5Hk,2588
+strawberry/experimental/pydantic/object_type.py,sha256=gZ_-SJoH4ZG6CIVa2BBInzW9TIXeTw_0-q74a-x1EuI,12372
+strawberry/experimental/pydantic/utils.py,sha256=9m_oKyz0K1rTmh1ZrYtbFfyUNsiV11RiEb7bLrZuOXs,3770
 strawberry/ext/LICENSE,sha256=_oY0TZg0b_sW0--0T44aMTpy2e2zF1Kiyn8E1qDiivo,1249
 strawberry/ext/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 strawberry/ext/dataclasses/LICENSE,sha256=WZgm35K_3NJwLqxpEHJJi7CWxVrwTumEz5D3Dtd7WnA,13925
@@ -165,7 +165,7 @@ strawberry/parent.py,sha256=mCnJcLBQKwtokXcGxkm5HqGg7Wkst97rn61ViuaotrM,829
 strawberry/permission.py,sha256=Rm_dft2AGZCSgBRQuxWEfncgWj88nmSdL-5mJ6TvHeA,5917
 strawberry/printer/__init__.py,sha256=DmepjmgtkdF5RxK_7yC6qUyRWn56U-9qeZMbkztYB9w,62
 strawberry/printer/ast_from_value.py,sha256=MFIX2V51d9ocRvD0Njemjk8YIzKh2BB1g2iUcX6a3d8,4946
-strawberry/printer/printer.py,sha256=x6qGNchRAb75yBxOJV8jzHYSE42uaJns_PC9ulKVsYI,17454
+strawberry/printer/printer.py,sha256=PXuYfAJxMU92aRETeSixhJjk-XUDu2ZFAFwkogzTu_Y,17461
 strawberry/private.py,sha256=fS5IbVUajzr-6NcR41yt4ukXDPAMzwe1WPNdRJP5wWk,534
 strawberry/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 strawberry/quart/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -194,7 +194,7 @@ strawberry/schema/types/__init__.py,sha256=oHO3COWhL3L1KLYCJNY1XFf5xt2GGtHiMC-Ua
 strawberry/schema/types/base_scalars.py,sha256=Z_BmgwLicNexLipGyw6MmZ7OBnkGJU3ySgaY9SwBWrw,1837
 strawberry/schema/types/concrete_type.py,sha256=HB30G1hMUuuvjAvfSe6ADS35iI_T_wKO-EprVOWTMSs,746
 strawberry/schema/types/scalar.py,sha256=SVJ8HiKncCvOw2xwABI5xYaHcC7KkGHG-tx2WDtSoCA,2802
-strawberry/schema_codegen/__init__.py,sha256=U-ABa02BAfCN6zEGlc2LJEju18ErtGjbf0S8BhJXc6w,15592
+strawberry/schema_codegen/__init__.py,sha256=2TO50tEZXQkH9yLUEBCZXEd0tUuzZ_XcY5dcKaf26ps,24040
 strawberry/schema_directive.py,sha256=GxiOedFB-RJAflpQNUZv00C5Z6gavR-AYdsvoCA_0jc,1963
 strawberry/starlite/__init__.py,sha256=v209swT8H9MljVL-npvANhEO1zz3__PSfxb_Ix-NoeE,134
 strawberry/starlite/controller.py,sha256=x6Mm3r36cRfzo6hz9B4AYWbVh2QlYtndYcXFOr_3THM,11860
@@ -223,7 +223,7 @@ strawberry/types/fields/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
 strawberry/types/fields/resolver.py,sha256=Lwy2XVKnTbDyibk9pPsWsunlm22akXjy-gKCg22Tp1A,14130
 strawberry/types/graphql.py,sha256=3SWZEsa0Zy1eVW6vy75BnB7t9_lJVi6TBV3_1j3RNBs,687
 strawberry/types/info.py,sha256=b1ZWW_wUop6XrGNcGHKBQeUYjlX-y8u3s2Wm_XhKPYI,3412
-strawberry/types/nodes.py,sha256=2ZVa1HOFgHZ96QTfz3QEr1fufi6Sz9hAzcZxsN7KtGE,5026
+strawberry/types/nodes.py,sha256=5tTYmxGpVDshbydicHTTBWEiUe8A7p7mdiaSV8Ry80Y,5027
 strawberry/types/type_resolver.py,sha256=F0z_geS4VEun8EhD571LaTgI8ypjCeLfp910gF0Q3MY,6280
 strawberry/types/types.py,sha256=t5MOV4xiutPL2Ka02u3Z2V3eqy2R2JPYQxsqCnY2Aqk,7139
 strawberry/union.py,sha256=5DOzsmsrI4tU4aXO7BBEN-570ABQ9_VkRDvUJUrIofY,9926
@@ -241,8 +241,8 @@ strawberry/utils/logging.py,sha256=flS7hV0JiIOEdXcrIjda4WyIWix86cpHHFNJL8gl1y4,7
 strawberry/utils/operation.py,sha256=Um-tBCPl3_bVFN2Ph7o1mnrxfxBes4HFCj6T0x4kZxE,1135
 strawberry/utils/str_converters.py,sha256=avIgPVLg98vZH9mA2lhzVdyyjqzLsK2NdBw9mJQ02Xk,813
 strawberry/utils/typing.py,sha256=Qxz1LwyVsNGV7LQW1dFsaUbsswj5LHBOdKLMom5eyEA,13491
-strawberry_graphql-0.222.0.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
-strawberry_graphql-0.222.0.dist-info/METADATA,sha256=yc6q0kAtPFBIqo0Whk1t6bxehAmcOKFfnd4-4lcAIcs,7740
-strawberry_graphql-0.222.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-strawberry_graphql-0.222.0.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
-strawberry_graphql-0.222.0.dist-info/RECORD,,
+strawberry_graphql-0.224.0.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
+strawberry_graphql-0.224.0.dist-info/METADATA,sha256=lahprMtPJhhknOqREgJb0Yc-JujPzgHJXA5loc8tOi8,7740
+strawberry_graphql-0.224.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+strawberry_graphql-0.224.0.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
+strawberry_graphql-0.224.0.dist-info/RECORD,,