strawberry-graphql 0.288.4__py3-none-any.whl → 0.289.0__py3-none-any.whl

strawberry/types/enum.py

@@ -1,7 +1,7 @@
 import dataclasses
 from collections.abc import Callable, Iterable, Mapping
 from enum import EnumMeta
-from typing import TYPE_CHECKING, Any, TypeGuard, TypeVar, overload
+from typing import TYPE_CHECKING, Any, Literal, TypeGuard, TypeVar, overload
 
 from strawberry.exceptions import ObjectIsNotAnEnumError
 from strawberry.types.base import (
@@ -103,6 +103,7 @@ def enum_value(
 
 
 EnumType = TypeVar("EnumType", bound=EnumMeta)
+GraphqlEnumNameFrom = Literal["key", "value"]
 
 
 def _process_enum(
@@ -110,6 +111,7 @@ def _process_enum(
     name: str | None = None,
     description: str | None = None,
     directives: Iterable[object] = (),
+    graphql_name_from: GraphqlEnumNameFrom = "key",
 ) -> EnumType:
     if not isinstance(cls, EnumMeta):
         raise ObjectIsNotAnEnumError(cls)
@@ -119,6 +121,7 @@ def _process_enum(
 
     values = []
     for item in cls:  # type: ignore
+        graphql_name = None
         item_value = item.value
         item_name = item.name
         deprecation_reason = None
@@ -135,13 +138,19 @@ def _process_enum(
             cls._value2member_map_[item_value.value] = item
             cls._member_map_[item_name]._value_ = item_value.value
 
-            if item_value.graphql_name:
-                item_name = item_value.graphql_name
-
+            graphql_name = item_value.graphql_name
             item_value = item_value.value
 
+        if not graphql_name:
+            if graphql_name_from == "key":
+                graphql_name = item_name
+            elif graphql_name_from == "value":
+                graphql_name = item_value
+            else:
+                raise ValueError(f"Invalid mode: {graphql_name_from}")
+
         value = EnumValue(
-            item_name,
+            graphql_name,
             item_value,
             deprecation_reason=deprecation_reason,
             directives=item_directives,
@@ -174,6 +183,7 @@ def enum(
     name: str | None = None,
     description: str | None = None,
     directives: Iterable[object] = (),
+    graphql_name_from: GraphqlEnumNameFrom = "key",
 ) -> EnumType: ...
 
 
@@ -184,6 +194,7 @@ def enum(
     name: str | None = None,
     description: str | None = None,
     directives: Iterable[object] = (),
+    graphql_name_from: GraphqlEnumNameFrom = "key",
 ) -> Callable[[EnumType], EnumType]: ...
 
 
@@ -193,18 +204,20 @@ def enum(
     name: str | None = None,
     description: str | None = None,
     directives: Iterable[object] = (),
+    graphql_name_from: GraphqlEnumNameFrom = "key",
 ) -> EnumType | Callable[[EnumType], EnumType]:
     """Annotates an Enum class a GraphQL enum.
 
     GraphQL enums only have names, while Python enums have names and values,
-    Strawberry will use the names of the Python enum as the names of the
-    GraphQL enum values.
+    Strawberry will by default use the names of the Python enum as the names of the
+    GraphQL enum values. You can use the values instead by using mode="value".
 
     Args:
         cls: The Enum class to be annotated.
         name: The name of the GraphQL enum.
         description: The description of the GraphQL enum.
         directives: The directives to attach to the GraphQL enum.
+        graphql_name_from: Whether to use the names (key) or values of the Python enums in GraphQL.
 
     Returns:
         The decorated Enum class.
@@ -235,7 +248,13 @@ def enum(
     """
 
     def wrap(cls: EnumType) -> EnumType:
-        return _process_enum(cls, name, description, directives=directives)
+        return _process_enum(
+            cls,
+            name,
+            description,
+            directives=directives,
+            graphql_name_from=graphql_name_from,
+        )
 
     if not cls:
         return wrap

{strawberry_graphql-0.288.4.dist-info → strawberry_graphql-0.289.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: strawberry-graphql
-Version: 0.288.4
+Version: 0.289.0
 Summary: A library for creating GraphQL APIs
 License: MIT
 License-File: LICENSE

{strawberry_graphql-0.288.4.dist-info → strawberry_graphql-0.289.0.dist-info}/RECORD

@@ -210,7 +210,7 @@ strawberry/types/arguments.py,sha256=3Rt1WlVDMc0KKYmJOnMUckeTB0egJv96Uw7RMopM4DM
 strawberry/types/auto.py,sha256=4sDflIg-Kz-QpK7Qo7bCSFblw7VE-0752rgayukFznU,2954
 strawberry/types/base.py,sha256=Hubo7u6OsU1v1v-w-vBCwlNVTqHaNn4zPOJuX93WR5U,15699
 strawberry/types/cast.py,sha256=fx86MkLW77GIximBAwUk5vZxSGwDqUA6XicXvz8EXwQ,916
-strawberry/types/enum.py,sha256=Dc1emdlBi1m9pK9h6rdyq8VRaBLafQO7MQdKhVgYXvs,7286
+strawberry/types/enum.py,sha256=sL7SMLx7W9h0-zxtKews6drfYtLUVZ1VorXuXp83mj4,8103
 strawberry/types/execution.py,sha256=z84AekAD_V_F4vGGHMPMGlAT0Htiw5JhEVvw1F9sGYQ,4019
 strawberry/types/field.py,sha256=-E5jfq7YXzI2CUepttfGzgmXFA1BTR2hmts44P-JpEk,20422
 strawberry/types/fields/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -238,8 +238,8 @@ strawberry/utils/logging.py,sha256=Dnivjd0ZhK_lAvjvuyCDkEWDhuURBoK9d3Kt_mIqbRg,7
 strawberry/utils/operation.py,sha256=Qs3ttbuC415xEVqmJ6YsWQpJNUo8CZJq9AoMB-yV65w,1215
 strawberry/utils/str_converters.py,sha256=-eH1Cl16IO_wrBlsGM-km4IY0IKsjhjnSNGRGOwQjVM,897
 strawberry/utils/typing.py,sha256=eE9NeMfASeXRstbjLnQFfOPymcSX8xwg3FGw_HCp95E,11828
-strawberry_graphql-0.288.4.dist-info/METADATA,sha256=LiS09hr9ulZqpMfHbqvHv4rVtV7Ej_WIgApx4CQU-C0,7627
-strawberry_graphql-0.288.4.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-strawberry_graphql-0.288.4.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
-strawberry_graphql-0.288.4.dist-info/licenses/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
-strawberry_graphql-0.288.4.dist-info/RECORD,,
+strawberry_graphql-0.289.0.dist-info/METADATA,sha256=kS72_h2N1ZRutWs8b46_u8-eKJHscc495FQ56I_qNSE,7627
+strawberry_graphql-0.289.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+strawberry_graphql-0.289.0.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
+strawberry_graphql-0.289.0.dist-info/licenses/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
+strawberry_graphql-0.289.0.dist-info/RECORD,,