Add support for using raw Python enum types in schema

RELEASE.md

@@ -0,0 +1,25 @@
+Release type: patch
+
+This release adds support for using rwa Python enum types in your schema
+(enums that are not decorated with `@strawberry.enum`)
+
+This is useful if you have enum types from other places in your code
+that you want to use in strawberry.
+i.e
+```py
+# somewhere.py
+from enum import Enum
+
+
+class AnimalKind(Enum):
+    AXOLOTL, CAPYBARA = range(2)
+
+
+# gql/animals
+from somewhere import AnimalKind
+
+
+@strawberry.type
+class AnimalType:
+    kind: AnimalKind
+```

strawberry/annotation.py

@@ -20,7 +20,6 @@
 )
 from typing_extensions import Annotated, Self, get_args, get_origin
 
-from strawberry.exceptions.not_a_strawberry_enum import NotAStrawberryEnumError
 from strawberry.types.base import (
     StrawberryList,
     StrawberryObjectDefinition,
@@ -30,6 +29,7 @@
     has_object_definition,
 )
 from strawberry.types.enum import EnumDefinition
+from strawberry.types.enum import enum as strawberry_enum
 from strawberry.types.lazy_type import LazyType
 from strawberry.types.private import is_private
 from strawberry.types.scalar import ScalarDefinition
@@ -187,7 +187,7 @@ def create_enum(self, evaled_type: Any) -> EnumDefinition:
         try:
             return evaled_type._enum_definition
         except AttributeError:
-            raise NotAStrawberryEnumError(evaled_type)
+            return strawberry_enum(evaled_type)._enum_definition
 
     def create_list(self, evaled_type: Any) -> StrawberryList:
         item_type, *_ = get_args(evaled_type)

tests/enums/test_enum.py

@@ -4,7 +4,7 @@
 
 import strawberry
 from strawberry.exceptions import ObjectIsNotAnEnumError
-from strawberry.exceptions.not_a_strawberry_enum import NotAStrawberryEnumError
+from strawberry.types.base import get_object_definition
 from strawberry.types.enum import EnumDefinition
 
 
@@ -120,25 +120,6 @@ class IceCreamFlavour(Enum):
     assert definition.values[2].description is None
 
 
-@pytest.mark.raises_strawberry_exception(
-    NotAStrawberryEnumError, match='Enum "IceCreamFlavour" is not a Strawberry enum'
-)
-def test_raises_error_when_using_enum_not_decorated():
-    class IceCreamFlavour(Enum):
-        VANILLA = strawberry.enum_value("vanilla")
-        STRAWBERRY = strawberry.enum_value(
-            "strawberry",
-            description="Our favourite",
-        )
-        CHOCOLATE = "chocolate"
-
-    @strawberry.type
-    class Query:
-        flavour: IceCreamFlavour
-
-    strawberry.Schema(query=Query)
-
-
 def test_can_use_enum_values():
     @strawberry.enum
     class TestEnum(Enum):
@@ -169,3 +150,35 @@ class TestEnum(IntEnum):
     assert TestEnum.D.value == 4
 
     assert [x.value for x in TestEnum.__members__.values()] == [1, 2, 3, 4]
+
+
+def test_default_enum_implementation() -> None:
+    class Foo(Enum):
+        BAR = "bar"
+        BAZ = "baz"
+
+    @strawberry.type
+    class Query:
+        @strawberry.field
+        def foo(self, foo: Foo) -> Foo:
+            return foo
+
+    schema = strawberry.Schema(Query)
+    res = schema.execute_sync("{ foo(foo: BAR) }")
+    assert not res.errors
+    assert res.data
+    assert res.data["foo"] == "BAR"
+
+
+def test_default_enum_reuse() -> None:
+    class Foo(Enum):
+        BAR = "bar"
+        BAZ = "baz"
+
+    @strawberry.type
+    class SomeType:
+        foo: Foo
+        bar: Foo
+
+    definition = get_object_definition(SomeType, strict=True)
+    assert definition.fields[1].type is definition.fields[1].type