a stab at supporting a value getter to enable dataclass support

Author: bckohan

django_enum/converters.py

@@ -4,11 +4,11 @@
 """
 
 from enum import Enum
-from typing import Dict, Type
+from typing import Dict, Optional, Type, Union, cast
 
 from django.urls.converters import register_converter
 
-from django_enum.utils import determine_primitive
+from django_enum.utils import ValueGetter, determine_primitive
 
 __all__ = ["register_enum_converter"]
 
@@ -36,14 +36,21 @@ def to_url(self, value):
         return str(getattr(value, self.prop))
 
 
-def register_enum_converter(enum: Type[Enum], type_name="", prop="value"):
+def register_enum_converter(
+    enum: Type[Enum],
+    type_name="",
+    prop="value",
+    value: Optional[Union[str, ValueGetter]] = None,
+):
     """
     Register an enum converter for Django's URL dispatcher.
 
     :param enum: The enumeration type to register.
     :param type_name: the name to use for the converter
     :param prop: The property name to use to generate urls - by default the
         value is used.
+    :param value: The property or method to use to get the primitive value
+        from the enumeration instance (if different than "value").
     """
     register_converter(
         type(
@@ -52,7 +59,12 @@ def register_enum_converter(enum: Type[Enum], type_name="", prop="value"):
             {
                 "enum": enum,
                 "prop": prop,
-                "primitive": determine_primitive(enum),
+                "primitive": determine_primitive(
+                    enum,
+                    value=cast(ValueGetter, lambda e: getattr(e, value))
+                    if isinstance(value, str)
+                    else value,
+                ),
                 "regex": "|".join([str(getattr(env, prop)) for env in enum]),
                 "_lookup_": {str(getattr(env, prop)): env for env in enum},
             },