修改SqlAlchemy模型时出现Unhashable Type错误

时间:2014-01-23 17:44:20

标签: python flask flask-sqlalchemy flask-security

我正在使用几乎从文档中复制的最基本的烧瓶应用程序,并且收到了非常烦人的错误。我已经能够追踪它,但不知道如何解决它。我知道在实现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上,所有的软件包都是最新的。

1 个答案:

答案 0 :(得分:5)

我有一个“解决方案”,但我想确保它是一个可行的解决方案,而不是一个不稳定的黑客。

我意识到将 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)

如果我做得好,请告诉我,谢谢!