Ошибка Unhashable Type при изменении моделей SqlAlchemy

Я использую самое простое приложение для фляг, практически скопированное из документации, и получаю крайне раздражающую ошибку. Я смог отследить его, но не знаю, как его решить. Я знаю, что проблема возникает при реализации flask-security, но ошибка возникает внутри sqlalchemy. Любой совет?

from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy
import logging
app = Flask(__name__)

app.config['DEBUG'] = True
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:' #

db = SQLAlchemy(app)

roles_users = db.Table('roles_users',
        db.Column('user_id', db.Integer(), db.ForeignKey('user.id')),
        db.Column('role_id', db.Integer(), db.ForeignKey('role.id')))

class Role(db.Model):
    id = db.Column(db.Integer(), primary_key=True)
    name = db.Column(db.String(80), unique=True)
    description = db.Column(db.String(255))

    def __init__(self, name, desc):
        self.name = name 
        self.description = desc

    def __repr__(self):
        return '<Role: {}>'.format(str(self.name))

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(255), unique=True)
    password = db.Column(db.String(255))
    roles = db.relationship('Role', secondary=roles_users,
                            backref=db.backref('users', lazy='dynamic'))

    def __init__(self, username, password):
        self.username = username
        self.password = password

    def __repr__(self):
        return '<User: {}>'.format(str(self.username))

def main():
    db.create_all()
    adminRole = Role('Admin', 'Unrestricted')
    adminUser = User('admin', 'adminpassword')
    adminUser.roles = [adminRole]
    db.session.add(adminUser)
    db.session.commit()
    app.run(debug=True)

if __name__ == '__main__':
    LOG_FILENAME = 'test.log'
    logging.basicConfig(filename=LOG_FILENAME,level=logging.DEBUG,)
    logging.debug('This message should go to the log file')
    try:
       main()
    except:
       logging.exception('Got exception on main handler')
       raise

Приведенный выше код работает совершенно нормально. Проблема возникает при использовании Flask-Security и создании подкласса RoleMixin (который добавляет в модель функции eq и ne. Как только класс становится следующим:

class Role(db.Model):
    id = db.Column(db.Integer(), primary_key=True)
    name = db.Column(db.String(80), unique=True)
    description = db.Column(db.String(255))

    def __init__(self, name, desc):
        self.name = name 
        self.description = desc

    def __repr__(self):
        return '<Role: {}>'.format(str(self.name))


    def __eq__(self, other):
        return (self.name == other or
                self.name == getattr(other, 'name', None))

    def __ne__(self, other):
        return not self.__eq__(other)

Я получаю следующую ошибку:

ERROR:root:Got exception on main handler
Traceback (most recent call last):
  File "test.py", line 66, in <module>
    main()
  File "test.py", line 53, in main
    adminUser.roles = [adminRole]
  File "C:\Users\ramabodhi\Envs\test\lib\site-packages\sqlalchemy\orm\attributes.py", line 220, in __set__
    instance_dict(instance), value, None)
  File "C:\Users\ramabodhi\Envs\test\lib\site-packages\sqlalchemy\orm\attributes.py", line 975, in set
    lambda adapter, i: adapter.adapt_like_to_iterable(i))
  File "C:\Users\ramabodhi\Envs\test\lib\site-packages\sqlalchemy\orm\attributes.py", line 1010, in _set_iterable
    collections.bulk_replace(new_values, old_collection, new_collection)
  File "C:\Users\ramabodhi\Envs\test\lib\site-packages\sqlalchemy\orm\collections.py", line 782, in bulk_replace
    constants = existing_idset.intersection(values or ())
  File "C:\Users\ramabodhi\Envs\test\lib\site-packages\sqlalchemy\util\_collections.py", line 592, in intersection
    result._members.update(self._working_set(members).intersection(other))
TypeError: unhashable type: 'Role'
DEBUG:root:This message should go to the log file
ERROR:root:Got exception on main handler
Traceback (most recent call last):
  File "test.py", line 66, in <module>
    main()
  File "test.py", line 53, in main
    adminUser.roles = [adminRole]
  File "C:\Users\ramabodhi\Envs\test\lib\site-packages\sqlalchemy\orm\attributes.py", line 220, in __set__
    instance_dict(instance), value, None)
  File "C:\Users\ramabodhi\Envs\test\lib\site-packages\sqlalchemy\orm\attributes.py", line 975, in set
    lambda adapter, i: adapter.adapt_like_to_iterable(i))
  File "C:\Users\ramabodhi\Envs\test\lib\site-packages\sqlalchemy\orm\attributes.py", line 1010, in _set_iterable
    collections.bulk_replace(new_values, old_collection, new_collection)
  File "C:\Users\ramabodhi\Envs\test\lib\site-packages\sqlalchemy\orm\collections.py", line 782, in bulk_replace
    constants = existing_idset.intersection(values or ())
  File "C:\Users\ramabodhi\Envs\test\lib\site-packages\sqlalchemy\util\_collections.py", line 592, in intersection
    result._members.update(self._working_set(members).intersection(other))
TypeError: unhashable type: 'Role'

Я использую Python 3.3 Windows 7, и все мои пакеты обновлены.


person tryexceptcontinue    schedule 23.01.2014    source источник
comment
Вы говорите, что являетесь подклассом RoleMixin, но ваша роль — это класс db.Model.   -  person Paolo Casciello    schedule 23.01.2014
comment
Я был непоследователен в своем объяснении, извините. Все, что делает RoleMixin, — это добавляет в модель функции eq и ne, поэтому я просто закодировал их для простоты.   -  person tryexceptcontinue    schedule 23.01.2014


Ответы (1)


У меня есть «решение» для этого, но я хочу убедиться, что это жизнеспособное решение, а не нестабильный взлом.

Я понимаю, что добавление функций ne и eq в класс требует добавления функции hash, чтобы модель можно было хэшировать. Итак, теперь это работает:

class Role(db.Model):
    id = db.Column(db.Integer(), primary_key=True)
    name = db.Column(db.String(80), unique=True)
    description = db.Column(db.String(255))

    def __init__(self, name, desc):
        self.name = name 
        self.description = desc

    def __repr__(self):
        return '<Role: {}>'.format(str(self.name))


    def __eq__(self, other):
        return (self.name == other or
                self.name == getattr(other, 'name', None))

    def __ne__(self, other):
        return not self.__eq__(other)

    def __hash__(self):
        return hash(self.name)

Дайте мне знать, если я сделал это правильно, спасибо!

person tryexceptcontinue    schedule 23.01.2014
comment
Обычно я реализую __hash__ как return id(self). - person dirn; 23.01.2014
comment
@dim: тебе следует избавиться от этой привычки; ограничение хэша в python заключается в том, что a == b подразумевает hash(a) == hash(b), но если вы переопределили __eq__, как это делает ramabodhi; это ограничение больше не будет выполняться, и это приведет к поломке коллекций, ориентированных на хэш. - person SingleNegationElimination; 24.01.2014
comment
У меня была такая же ошибка в аналогичном классе Role, хотя мой не включал переопределения для функций eq и ne. Я тоже обнаружил, что добавление хеш-функции, как это сделал #ramabodhi, решило проблему. - person Steve Saporta; 08.04.2014