abolish jsonschema type

Author: Lingling Peng

synapseclient/extensions/curator/schema_generation.py

@@ -98,15 +98,6 @@ class ValidationRuleName(Enum):
     IN_RANGE = "inRange"
 
 
-class JSONSchemaType(Enum):
-    """This enum is the currently supported JSON Schema types"""
-
-    STRING = "string"
-    NUMBER = "number"
-    INTEGER = "integer"
-    BOOLEAN = "boolean"
-
-
 class RegexModule(Enum):
     """This enum are allowed modules for the regex validation rule"""
 
@@ -1645,7 +1636,7 @@ def is_class_in_schema(self, node_label: str) -> bool:
 
     def get_node_column_type(
         self, node_label: Optional[str] = None, node_display_name: Optional[str] = None
-    ) -> Optional[JSONSchemaType]:
+    ) -> str:
         """Gets the column type of the node
 
         Args:
@@ -1658,9 +1649,8 @@ def get_node_column_type(
         node_label = self._get_node_label(node_label, node_display_name)
         rel_node_label = self.dmr.get_relationship_value("columnType", "node_label")
         type_string = self.graph.nodes[node_label][rel_node_label]
-        if type_string is None:
-            return type_string
-        return JSONSchemaType(type_string)
+        return type_string
+        # return JSONSchemaType(type_string)
 
     def _get_node_label(
         self, node_label: Optional[str] = None, node_display_name: Optional[str] = None
@@ -2778,7 +2768,7 @@ def define_data_model_relationships(self) -> dict:
                 "required_header": False,
                 "edge_rel": False,
                 "node_attr_dict": {"default": None},
-                "allowed_values": [enum.value for enum in JSONSchemaType],
+                "allowed_values": ["string", "number", "integer", "boolean"],
             },
         }
 
@@ -4360,7 +4350,7 @@ class TraversalNode:  # pylint: disable=too-many-instance-attributes
     dependencies: list[str] = field(init=False)
     description: str = field(init=False)
     is_array: bool = field(init=False)
-    type: Optional[JSONSchemaType] = field(init=False)
+    type: str = field(init=False)
     format: Optional[JSONSchemaFormat] = field(init=False)
     minimum: Optional[float] = field(init=False)
     maximum: Optional[float] = field(init=False)
@@ -4846,7 +4836,7 @@ def _create_array_property(node: Node) -> Property:
 
     items: Items = {}
     if node.type:
-        items["type"] = node.type.value
+        items["type"] = node.type
         _set_type_specific_keywords(items, node)
 
     array_type_dict: TypeDict = {"type": "array", "title": "array"}
@@ -4919,10 +4909,10 @@ def _create_simple_property(node: Node) -> Property:
 
     if node.type:
         if node.is_required:
-            prop["type"] = node.type.value
+            prop["type"] = node.type
         else:
             prop["oneOf"] = [
-                {"type": node.type.value, "title": node.type.value},
+                {"type": node.type, "title": node.type},
                 {"type": "null", "title": "null"},
             ]
     elif node.is_required:

tests/unit/synapseclient/extensions/unit_test_create_json_schema.py

@@ -20,7 +20,6 @@
     GraphTraversalState,
     JSONSchema,
     JSONSchemaFormat,
-    JSONSchemaType,
     TraversalNode,
     _create_array_property,
     _create_enum_array_property,
@@ -247,7 +246,7 @@ def test_update_property(self) -> None:
 )
 def test_node_init(
     node_name: str,
-    expected_type: Optional[JSONSchemaType],
+    expected_type: str,
     expected_is_array: bool,
     expected_min: Optional[float],
     expected_max: Optional[float],