check to make sure there's a $ref

CanDIG · Nov 29, 2023 · 2a2721c · 2a2721c
1 parent 927c49b
commit 2a2721c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/schema.py b/schema.py
@@ -136,8 +136,10 @@ def fail(self, message):
 
 
     def expand_ref(self, ref):
-        refName = ref["$ref"].replace("#/components/schemas/", "")
-        return self.generate_schema_scaffold(json.loads(json.dumps(self.schema[refName])))
+        if "$ref" in ref:
+            refName = ref["$ref"].replace("#/components/schemas/", "")
+            return self.generate_schema_scaffold(json.loads(json.dumps(self.schema[refName])))
+        return ref["type"]
 
 
     def generate_schema_scaffold(self, schema_obj):