SQLAlchemy TypeDecorator не работает - PullRequest
Новая
       11

SQLAlchemy TypeDecorator не работает

5 голосов
/ 16 декабря 2011

Я использую xml в своей базе данных postgresql, и мне нужно, чтобы пользовательский тип мог обрабатывать xml данные в SQLAlchemy.

Итак, я сделал XMLType класс, общающийся с xml.etree, но он не работает так, как я хотел.

вот код, который я написал: 

import xml.etree.ElementTree as etree

class XMLType(sqlalchemy.types.TypeDecorator):

    impl = sqlalchemy.types.UnicodeText
    type = etree.Element

    def get_col_spec(self):
        return 'xml'

    def bind_processor(self, dialect):
        def process(value):
            if value is not None:
                return etree.dump(value)
            else:
                return None
        return process

    def process_result_value(self, value, dialect):
        if value is not None:
            value = etree.fromstring(value)
        return value

Хорошо работает при извлечении значений и обработке результатов. но когда я пытался вставить строку, я получаю сообщение об ошибке (конечно, я поставил body как xml.etree.ElementTree.Element объект): 

IntegrityError: (IntegrityError) null value in column "body" violates not-null 
constraint "INSERT INTO comments (id, author_id, look_id, body, created_at) 
VALUES (nextval('object_seq'), %(author_id)s, %(look_id)s, %(body)s, now()) 
RETURNING comments.id" {'body': None, 'author_id': 1550L, 'look_id': 83293L}

Видя, что значение body равно None, очевидно, что процессор привязки работает неправильно, но я думаю, что реализовал это правильно, поэтому не знаю, что мне делать, чтобы изменить ситуацию.

process_bind_param выдает мне ту же ошибку.

Где я ошибся в своем коде?

1 Ответ

5 голосов
/ 16 декабря 2011

ElementTree.dump() функция выводит XML в поток (стандартный вывод по умолчанию) и возвращает None.Используйте ElementTree.tostring() или создайте дамп для объекта StringIO.

...