Как использовать SQLAlchemy TypeDecorators для компиляции необработанного SQL - PullRequest
0 голосов
/ 21 января 2019

У меня есть объект SqlAlchemy Query, который использует uuid.UUID:

import uuid
import sqlalchemy

foreign_uuid = '822965bb-c67e-47ee-ad12-a3b060ef79ae'
qry = Query(MyModel).filter(MyOtherModel.uuid == uuid.UUID(foreign_uuid))

Теперь я хочу получить необработанный postgresql из SQLAlchemy:

qry.statement.compile(dialect=postgresql.dialect(), compile_kwargs={"literal_binds": True}))

, который выдает следующую ошибку: NotImplementedError: Don't know how to literal-quote value UUID('822965bb-c67e-47ee-ad12-a3b060ef79ae')

Кажется, это потому, что SqlAlchemy знает, как обращаться с базовыми типами. Документация предлагает использовать TypeDecorator и даже предоставляет один для GUID:

from sqlalchemy.dialects.postgresql import UUID
import uuid

class GUID(TypeDecorator):
    """Platform-independent GUID type.

    Uses PostgreSQL's UUID type, otherwise uses
    CHAR(32), storing as stringified hex values.

    """
    impl = CHAR

    def load_dialect_impl(self, dialect):
        if dialect.name == 'postgresql':
            return dialect.type_descriptor(UUID())
        else:
            return dialect.type_descriptor(CHAR(32))

    def process_bind_param(self, value, dialect):
        if value is None:
            return value
        elif dialect.name == 'postgresql':
            return str(value)
        else:
            if not isinstance(value, uuid.UUID):
                return "%.32x" % uuid.UUID(value).int
            else:
                # hexstring
                return "%.32x" % value.int

    def process_result_value(self, value, dialect):
        if value is None:
            return value
        else:
            if not isinstance(value, uuid.UUID):
                value = uuid.UUID(value)
            return value

Однако в нем не говорится, как использовать это TypeDecorator,Нужно ли ссылаться на это как-то в моей модели?Сделать ли его доступным как-нибудь для .statement.compile звонка?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...