the 'roles' attribute on the "User" and "Admin" mappers (inheriting properties copy across a hierarchy) were both trying to insert a row...added some more uniqueness + unittest to fix, revision 1653.On Jun 22, 2006, at 5:04 PM, Tim Van Steenburgh wrote: import sqlalchemy.mods.threadlocal
from sqlalchemy import *
sqlite = 'sqlite:///:memory:'
db = create_engine(sqlite,
strategy='plain',
echo=True
)
metadata = BoundMetaData(db)
################################################################################
# TABLE DEFS
################################################################################
users = Table('user', metadata,
Column('id', Integer, primary_key=True),
Column('first_name', String(64)),
Column('last_name', String(64)),
Column('email', String(128)),
Column('password', String(16)),
Column('is_active', Boolean)
)
roles = Table('role', metadata,
Column('id', Integer, primary_key=True),
Column('description', String(32))
)
user_roles = Table('user_role', metadata,
Column('user_id', Integer, ForeignKey('user.id'), primary_key=True),
Column('role_id', Integer, ForeignKey(' role.id'), primary_key=True)
)
admins = Table('admin', metadata,
Column('id', Integer, primary_key=True),
Column('user_id', Integer, ForeignKey('user.id'))
)
################################################################################
# CLASS DEFS
################################################################################
class User(object):
def __init__(self, fname=None, lname=None, email=None, is_active=1):
self.first_name = fname
self.last_name = lname
self.email = email
self.is_active = is_active
def has_role(self, role_name):
if role_name in [role.description for role in self.roles]:
return True
else:
return False
class Role(object):
def __init__(self, description=None):
self.description = description
class UserRole(object): pass
class Admin(User):
def __init__(self, fname=None, lname=None, email=None, is_active=1):
User.__init__(self, fname, lname, email, is_active)
# give user 'admin' role
self.roles.append(Query(Role).get_by(description='admin'))
################################################################################
# MAPPER DEFS
################################################################################
role_mapper = mapper(Role, roles)
user_mapper = mapper(User, users, properties = {
'roles' : relation(Role, secondary=user_roles, lazy=False, private=False)
}
)
admin_mapper = mapper(Admin, admins, inherits=user_mapper)
################################################################################
# MAIN
################################################################################
if __name__ == "__main__":
# create tables
users.create()
roles.create()
user_roles.create()
admins.create()
# create roles
adminrole = Role('admin')
clientrole = Role('client')
adjusterrole = Role('adjuster')
objectstore.flush()
# create admin user
a = Admin('Tim',
'Van Steenburgh',
' [EMAIL PROTECTED]'
)
a.password = 'admin'
objectstore.flush()
|
