Resolve TODO: Replace EnumDefinition with StrawberryEnum

RELEASE.md

@@ -0,0 +1,3 @@
+Release type: patch
+
+Resolve TODOs about EnumDefinition by renaming it to StrawberryEnum

strawberry/annotation.py

@@ -27,7 +27,7 @@
     get_object_definition,
     has_object_definition,
 )
-from strawberry.types.enum import EnumDefinition
+from strawberry.types.enum import StrawberryEnum
 from strawberry.types.enum import enum as strawberry_enum
 from strawberry.types.lazy_type import LazyType
 from strawberry.types.maybe import _annotation_is_maybe
@@ -213,11 +213,11 @@ def create_concrete_type(self, evaled_type: type) -> type:
             return evaled_type.__strawberry_definition__.resolve_generic(evaled_type)
         raise ValueError(f"Not supported {evaled_type}")
 
-    def create_enum(self, evaled_type: Any) -> EnumDefinition:
+    def create_enum(self, evaled_type: Any) -> StrawberryEnum:
         try:
-            return evaled_type._enum_definition
+            return evaled_type.__strawberry_definition__
         except AttributeError:
-            return strawberry_enum(evaled_type)._enum_definition
+            return strawberry_enum(evaled_type).__strawberry_definition__
 
     def create_list(self, evaled_type: Any) -> StrawberryList:
         item_type, *_ = get_args(evaled_type)
@@ -386,7 +386,7 @@ def _is_strawberry_type(cls, evaled_type: Any) -> bool:
         # Prevent import cycles
         from strawberry.types.union import StrawberryUnion
 
-        if isinstance(evaled_type, EnumDefinition):
+        if isinstance(evaled_type, StrawberryEnum):
             return True
         elif _is_input_type(evaled_type):  # TODO: Replace with StrawberryInputObject
             return True

strawberry/codegen/query_codegen.py

@@ -45,7 +45,7 @@
     get_object_definition,
     has_object_definition,
 )
-from strawberry.types.enum import EnumDefinition
+from strawberry.types.enum import StrawberryEnum
 from strawberry.types.lazy_type import LazyType
 from strawberry.types.scalar import ScalarDefinition, ScalarWrapper
 from strawberry.types.union import StrawberryUnion
@@ -551,7 +551,7 @@ def _get_field_type(
         if isinstance(field_type, ScalarDefinition):
             return self._collect_scalar(field_type, None)
 
-        if isinstance(field_type, EnumDefinition):
+        if isinstance(field_type, StrawberryEnum):
             return self._collect_enum(field_type)
 
         raise ValueError(f"Unsupported type: {field_type}")  # pragma: no cover
@@ -907,7 +907,7 @@ def _collect_scalar(
 
         return graphql_scalar
 
-    def _collect_enum(self, enum: EnumDefinition) -> GraphQLEnum:
+    def _collect_enum(self, enum: StrawberryEnum) -> GraphQLEnum:
         graphql_enum = GraphQLEnum(
             enum.name,
             [value.name for value in enum.values],

strawberry/experimental/pydantic/conversion.py

@@ -9,7 +9,7 @@
     StrawberryOptional,
     has_object_definition,
 )
-from strawberry.types.enum import EnumDefinition
+from strawberry.types.enum import StrawberryEnum
 from strawberry.types.union import StrawberryUnion
 
 if TYPE_CHECKING:
@@ -40,7 +40,7 @@ def _convert_from_pydantic_to_strawberry_type(
                 return _convert_from_pydantic_to_strawberry_type(
                     option_type, data_from_model=data, extra=extra
                 )
-    if isinstance(type_, EnumDefinition):
+    if isinstance(type_, StrawberryEnum):
         return data
     if isinstance(type_, StrawberryList):
         items = []

strawberry/federation/schema.py

@@ -34,7 +34,7 @@
     from strawberry.federation.schema_directives import ComposeDirective
     from strawberry.schema.config import StrawberryConfig
     from strawberry.schema_directive import StrawberrySchemaDirective
-    from strawberry.types.enum import EnumDefinition
+    from strawberry.types.enum import StrawberryEnum
     from strawberry.types.scalar import ScalarDefinition, ScalarWrapper
 
 
@@ -373,7 +373,7 @@ def _has_federation_keys(
     definition: Union[
         StrawberryObjectDefinition,
         "ScalarDefinition",
-        "EnumDefinition",
+        "StrawberryEnum",
         "StrawberryUnion",
     ],
 ) -> bool:

strawberry/printer/printer.py

@@ -39,7 +39,7 @@
     StrawberryObjectDefinition,
     has_object_definition,
 )
-from strawberry.types.enum import EnumDefinition
+from strawberry.types.enum import StrawberryEnum
 from strawberry.types.scalar import ScalarWrapper
 from strawberry.types.unset import UNSET
 
@@ -183,7 +183,7 @@ def print_schema_directive(
             if hasattr(f_type, "_scalar_definition"):
                 extras.types.add(cast("type", f_type))
 
-            if isinstance(f_type, EnumDefinition):
+            if isinstance(f_type, StrawberryEnum):
                 extras.types.add(cast("type", f_type))
 
     return f" @{gql_directive.name}{params}"

strawberry/schema/base.py

@@ -22,7 +22,7 @@
         StrawberryObjectDefinition,
         WithStrawberryObjectDefinition,
     )
-    from strawberry.types.enum import EnumDefinition
+    from strawberry.types.enum import StrawberryEnum
     from strawberry.types.graphql import OperationType
     from strawberry.types.scalar import ScalarDefinition
     from strawberry.types.union import StrawberryUnion
@@ -83,7 +83,7 @@ def get_type_by_name(
         Union[
             StrawberryObjectDefinition,
             ScalarDefinition,
-            EnumDefinition,
+            StrawberryEnum,
             StrawberryUnion,
         ]
     ]:

strawberry/schema/compat.py

@@ -4,6 +4,7 @@
 
 from strawberry.scalars import is_scalar as is_strawberry_scalar
 from strawberry.types.base import StrawberryType, has_object_definition
+from strawberry.types.enum import StrawberryEnum
 
 # TypeGuard is only available in typing_extensions => 3.10, we don't want
 # to force updates to the typing_extensions package so we only use it when
@@ -36,7 +37,10 @@ def is_scalar(
 
 
 def is_enum(type_: Union[StrawberryType, type]) -> TypeGuard[type]:
-    return hasattr(type_, "_enum_definition")
+    if hasattr(type_, "__strawberry_definition__"):
+        return isinstance(type_.__strawberry_definition__, StrawberryEnum)
+
+    return False
 
 
 def is_schema_directive(type_: Union[StrawberryType, type]) -> TypeGuard[type]:

strawberry/schema/name_converter.py

@@ -11,7 +11,7 @@
     StrawberryOptional,
     has_object_definition,
 )
-from strawberry.types.enum import EnumDefinition, EnumValue
+from strawberry.types.enum import EnumValue, StrawberryEnum
 from strawberry.types.lazy_type import LazyType
 from strawberry.types.scalar import ScalarDefinition
 from strawberry.types.union import StrawberryUnion
@@ -45,7 +45,7 @@ def from_type(
     ) -> str:
         if isinstance(type_, (StrawberryDirective, StrawberrySchemaDirective)):
             return self.from_directive(type_)
-        if isinstance(type_, EnumDefinition):  # TODO: Replace with StrawberryEnum
+        if isinstance(type_, StrawberryEnum):
             return self.from_enum(type_)
         if isinstance(type_, StrawberryObjectDefinition):
             if type_.is_input:
@@ -78,10 +78,10 @@ def from_input_object(self, input_type: StrawberryObjectDefinition) -> str:
     def from_interface(self, interface: StrawberryObjectDefinition) -> str:
         return self.from_object(interface)
 
-    def from_enum(self, enum: EnumDefinition) -> str:
+    def from_enum(self, enum: StrawberryEnum) -> str:
         return enum.name
 
-    def from_enum_value(self, enum: EnumDefinition, enum_value: EnumValue) -> str:
+    def from_enum_value(self, enum: StrawberryEnum, enum_value: EnumValue) -> str:
         return enum_value.name
 
     def from_directive(
@@ -152,7 +152,7 @@ def get_name_from_type(self, type_: Union[StrawberryType, type]) -> str:
         if isinstance(type_, LazyType):
             type_ = type_.resolve_type()
 
-        if isinstance(type_, EnumDefinition):
+        if isinstance(type_, StrawberryEnum):
             name = type_.name
         elif isinstance(type_, StrawberryUnion):
             name = type_.graphql_name if type_.graphql_name else self.from_union(type_)

strawberry/schema/schema.py

@@ -92,7 +92,7 @@
 
     from strawberry.directive import StrawberryDirective
     from strawberry.types.base import StrawberryType
-    from strawberry.types.enum import EnumDefinition
+    from strawberry.types.enum import StrawberryEnum
     from strawberry.types.field import StrawberryField
     from strawberry.types.scalar import ScalarDefinition, ScalarWrapper
     from strawberry.types.union import StrawberryUnion
@@ -443,7 +443,7 @@ def get_type_by_name(
         Union[
             StrawberryObjectDefinition,
             ScalarDefinition,
-            EnumDefinition,
+            StrawberryEnum,
             StrawberryUnion,
         ]
     ]:
@@ -918,7 +918,7 @@ def _resolve_node_ids(self) -> None:
         for concrete_type in self.schema_converter.type_map.values():
             type_def = concrete_type.definition
 
-            # This can be a TypeDefinition, EnumDefinition, ScalarDefinition
+            # This can be a TypeDefinition, StrawberryEnum, ScalarDefinition
             # or UnionDefinition
             if not isinstance(type_def, StrawberryObjectDefinition):
                 continue

strawberry/schema/schema_converter.py

@@ -67,7 +67,7 @@
     has_object_definition,
 )
 from strawberry.types.cast import get_strawberry_type_cast
-from strawberry.types.enum import EnumDefinition
+from strawberry.types.enum import StrawberryEnum
 from strawberry.types.field import UNRESOLVED
 from strawberry.types.lazy_type import LazyType
 from strawberry.types.private import is_private
@@ -157,7 +157,7 @@ def _get_thunk_mapping(
 class CustomGraphQLEnumType(GraphQLEnumType):
     def __init__(
         self,
-        enum: EnumDefinition,
+        enum: StrawberryEnum,
         *args: Any,
         **kwargs: Any,
     ) -> None:
@@ -301,7 +301,7 @@ def from_argument(self, argument: StrawberryArgument) -> GraphQLArgument:
             },
         )
 
-    def from_enum(self, enum: EnumDefinition) -> CustomGraphQLEnumType:
+    def from_enum(self, enum: StrawberryEnum) -> CustomGraphQLEnumType:
         enum_name = self.config.name_converter.from_type(enum)
 
         assert enum_name is not None
@@ -869,7 +869,7 @@ def from_type(self, type_: Union[StrawberryType, type]) -> GraphQLNullableType:
         if compat.is_graphql_generic(type_):
             raise MissingTypesForGenericError(type_)
 
-        if isinstance(type_, EnumDefinition):  # TODO: Replace with StrawberryEnum
+        if isinstance(type_, StrawberryEnum):
             return self.from_enum(type_)
         if compat.is_input_type(type_):  # TODO: Replace with StrawberryInputObject
             return self.from_input_object(type_)
@@ -882,8 +882,8 @@ def from_type(self, type_: Union[StrawberryType, type]) -> GraphQLNullableType:
             return self.from_interface(type_definition)
         if has_object_definition(type_):
             return self.from_object(type_.__strawberry_definition__)
-        if compat.is_enum(type_):  # TODO: Replace with StrawberryEnum
-            enum_definition: EnumDefinition = type_._enum_definition  # type: ignore
+        if compat.is_enum(type_):
+            enum_definition: StrawberryEnum = type_.__strawberry_definition__  # type: ignore
             return self.from_enum(enum_definition)
         if isinstance(type_, StrawberryObjectDefinition):
             return self.from_object(type_)
@@ -1018,14 +1018,14 @@ def validate_same_type_definition(
 
         if isinstance(second_type_definition, StrawberryObjectDefinition):
             first_origin = second_type_definition.origin
-        elif isinstance(second_type_definition, EnumDefinition):
+        elif isinstance(second_type_definition, StrawberryEnum):
             first_origin = second_type_definition.wrapped_cls
         else:
             first_origin = None
 
         if isinstance(first_type_definition, StrawberryObjectDefinition):
             second_origin = first_type_definition.origin
-        elif isinstance(first_type_definition, EnumDefinition):
+        elif isinstance(first_type_definition, StrawberryEnum):
             second_origin = first_type_definition.wrapped_cls
         else:
             second_origin = None

strawberry/schema/types/concrete_type.py

@@ -7,7 +7,7 @@
 
 if TYPE_CHECKING:
     from strawberry.types.base import StrawberryObjectDefinition
-    from strawberry.types.enum import EnumDefinition
+    from strawberry.types.enum import StrawberryEnum
     from strawberry.types.scalar import ScalarDefinition
     from strawberry.types.union import StrawberryUnion
 
@@ -17,7 +17,7 @@
 @dataclasses.dataclass
 class ConcreteType:
     definition: Union[
-        StrawberryObjectDefinition, EnumDefinition, ScalarDefinition, StrawberryUnion
+        StrawberryObjectDefinition, StrawberryEnum, ScalarDefinition, StrawberryUnion
     ]
     implementation: GraphQLType
 

strawberry/types/arguments.py

@@ -21,7 +21,7 @@
     StrawberryOptional,
     has_object_definition,
 )
-from strawberry.types.enum import EnumDefinition
+from strawberry.types.enum import StrawberryEnum
 from strawberry.types.lazy_type import LazyType, StrawberryLazyReference
 from strawberry.types.maybe import Some
 from strawberry.types.unset import UNSET as _deprecated_UNSET  # noqa: N811
@@ -155,16 +155,12 @@ def _is_leaf_type(
     if is_scalar(type_, scalar_registry):
         return True
 
-    if isinstance(type_, EnumDefinition):
+    if isinstance(type_, StrawberryEnum):
         return True
 
     if isinstance(type_, LazyType):
         return _is_leaf_type(type_.resolve_type(), scalar_registry)
 
-    if hasattr(type_, "_enum_definition"):
-        enum_definition: EnumDefinition = type_._enum_definition
-        return _is_leaf_type(enum_definition, scalar_registry)
-
     return False
 
 
@@ -242,8 +238,10 @@ def convert_argument(
     if isinstance(type_, LazyType):
         return convert_argument(value, type_.resolve_type(), scalar_registry, config)
 
-    if hasattr(type_, "_enum_definition"):
-        enum_definition: EnumDefinition = type_._enum_definition
+    if hasattr(type_, "__strawberry_definition__") and isinstance(
+        type_.__strawberry_definition__, StrawberryEnum
+    ):
+        enum_definition: StrawberryEnum = type_.__strawberry_definition__
         return convert_argument(value, enum_definition, scalar_registry, config)
 
     if has_object_definition(type_):

strawberry/types/base.py

@@ -10,6 +10,7 @@
     Optional,
     TypeVar,
     Union,
+    cast,
     overload,
 )
 from typing_extensions import Literal, Protocol, Self, deprecated
@@ -204,15 +205,15 @@ def has_object_definition(
     obj: Any,
 ) -> TypeGuard[type[WithStrawberryObjectDefinition]]:
     if hasattr(obj, "__strawberry_definition__"):
-        return True
+        return isinstance(obj.__strawberry_definition__, StrawberryObjectDefinition)
     # TODO: Generics remove dunder members here, so we inject it here.
     #  Would be better to avoid it somehow.
     # https://github.com/python/cpython/blob/3a314f7c3df0dd7c37da7d12b827f169ee60e1ea/Lib/typing.py#L1152
     if is_concrete_generic(obj):
         concrete = obj.__origin__
         if hasattr(concrete, "__strawberry_definition__"):
             obj.__strawberry_definition__ = concrete.__strawberry_definition__
-            return True
+            return isinstance(obj.__strawberry_definition__, StrawberryObjectDefinition)
     return False
 
 
@@ -425,9 +426,17 @@ def is_implemented_by(self, root: type[WithStrawberryObjectDefinition]) -> bool:
 
             # TODO: uniform type var map, at the moment we map object types
             # to their class (not to TypeDefinition) while we map enum to
-            # the EnumDefinition class. This is why we do this check here:
-            if hasattr(real_concrete_type, "_enum_definition"):
-                real_concrete_type = real_concrete_type._enum_definition
+            # the StrawberryEnum class. This is why we do this check here:
+
+            if hasattr(real_concrete_type, "__strawberry_definition__"):
+                from strawberry.types.enum import StrawberryEnum
+
+                if isinstance(
+                    real_concrete_type.__strawberry_definition__, StrawberryEnum
+                ):
+                    real_concrete_type = cast(
+                        "Any", real_concrete_type.__strawberry_definition__
+                    )
 
             if (
                 isinstance(expected_concrete_type, type)