Version 2.3.2 (Compat with API Star schemas)

lovelydinosaur · lovelydinosaur · commit 27f4f6ed7170 · 2017-10-05T15:02:10.000+01:00
diff --git a/coreapi/__init__.py b/coreapi/__init__.py
@@ -4,7 +4,7 @@
 from coreapi.document import Array, Document, Link, Object, Error, Field
 
 
-__version__ = '2.3.1'
+__version__ = '2.3.2'
 __all__ = [
     'Array', 'Document', 'Link', 'Object', 'Error', 'Field',
     'Client',
diff --git a/coreapi/codecs/corejson.py b/coreapi/codecs/corejson.py
@@ -32,13 +32,16 @@
 
 
 def encode_schema_to_corejson(schema):
-    type_id = SCHEMA_CLASS_TO_TYPE_ID.get(schema.__class__, 'anything')
+    if hasattr(schema, 'typename'):
+        type_id = schema.typename
+    else:
+        type_id = SCHEMA_CLASS_TO_TYPE_ID.get(schema.__class__, 'anything')
     retval = {
         '_type': type_id,
         'title': schema.title,
         'description': schema.description
     }
-    if isinstance(schema, coreschema.Enum):
+    if hasattr(schema, 'enum'):
         retval['enum'] = schema.enum
     return retval
 
