Add Enum and JSONString type

nebula-contrib · Oct 18, 2024 · 183e2d4 · 183e2d4
1 parent b010157
commit 183e2d4
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/nebula_carina/ngql/schema/data_types.py b/nebula_carina/ngql/schema/data_types.py
@@ -115,6 +115,20 @@ def __eq__(self, other):
     def value2db_str(cls, value):
         return 'NULL' if value is None else f'"{value}"'
 
+class JSONString(DataType):
+    python_data_type = str
+
+    @classmethod
+    def value2db_str(cls, value):
+        return "NULL" if value is None else f"'{value}'"
+
+
+class EnumType(DataType):
+    python_data_type = str
+
+    @classmethod
+    def value2db_str(cls, enum_value):
+        return "NULL" if enum_value is None else f'"{enum_value.value}"'
 
 class Bool(DataType):
     python_data_type = bool