init

util/json_type.py

@@ -0,0 +1,24 @@
+from typing import Any
+import json
+from sqlalchemy import types
+from sqlalchemy.dialects import postgresql
+
+class JSONType(types.TypeDecorator): # type: ignore
+	impl = types.TEXT
+	
+	def load_dialect_impl(self, dialect: Any) -> Any:
+		if dialect.name == 'postgresql':
+			t = postgresql.JSON()
+		else:
+			t = types.TEXT()
+		return dialect.type_descriptor(t)
+	
+	def process_bind_param(self, value: Any, dialect: Any) -> Any:
+		if value is None or dialect.name == 'postgresql':
+			return value
+		return json.dumps(value)
+	
+	def process_result_value(self, value: Any, dialect: Any) -> Any:
+		if value is None or dialect.name == 'postgresql':
+			return value
+		return json.loads(value)