mirror of
https://git.ugnet.gay/CrossTalk/azul.git
synced 2026-05-27 22:59:49 +00:00
Athena Funderburg
25 lines
680 B
Python
25 lines
680 B
Python
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)
|