Fallback to String / Implemented Enum support.

Author: flipbit03

graphene_sqlalchemy/converter.py

@@ -1,5 +1,7 @@
 import datetime
+import enum
 import typing
+import warnings
 from decimal import Decimal
 from functools import singledispatch
 from typing import Any
@@ -18,7 +20,8 @@
                      default_connection_field_factory)
 from .registry import get_global_registry
 from .resolvers import get_attr_resolver, get_custom_resolver
-from .utils import singledispatchbymatchfunction, value_equals
+from .utils import (singledispatchbymatchfunction, value_equals,
+                    value_is_subclass)
 
 try:
     from sqlalchemy_utils import ChoiceType, JSONType, ScalarListType, TSVectorType
@@ -271,7 +274,13 @@ def convert_hybrid_property_return_type_inner(arg: Any):
     existing_graphql_type = get_global_registry().get_type_for_model(arg)
     if existing_graphql_type:
         return existing_graphql_type
-    raise Exception(f"I don't know how to generate a GraphQL type out of a \"{arg}\" type")
+
+    # No valid type found, warn and fall back to graphene.String
+    warnings.warn(
+        (f"I don't know how to generate a GraphQL type out of a \"{arg}\" type."
+         "Falling back to \"graphene.String\"")
+    )
+    return String
 
 
 @convert_hybrid_property_return_type_inner.register(value_equals(str))
@@ -317,6 +326,11 @@ def convert_hybrid_property_return_type_inner_time(arg):
     return Time
 
 
+@convert_hybrid_property_return_type_inner.register(value_is_subclass(enum.Enum))
+def convert_hybrid_property_return_type_inner_enum(arg):
+    return Enum.from_enum(arg)
+
+
 @convert_hybrid_property_return_type_inner.register(lambda x: getattr(x, '__origin__', None) in [list, typing.List])
 def convert_hybrid_property_return_type_inner_list(arg):
     # type is either list[T] or List[T], generic argument at __args__[0]

graphene_sqlalchemy/utils.py

@@ -188,3 +188,13 @@ def value_equals(value):
     """A simple function that makes the equality based matcher functions for
      SingleDispatchByMatchFunction prettier"""
     return lambda x: x == value
+
+
+def value_is_subclass(cls):
+    def safe_subclass_checker(o):
+        try:
+            return issubclass(o, cls)
+        except TypeError:
+            return False
+
+    return safe_subclass_checker