SQLAlchemy映射将表的列连接到一个对象

时间:2014-03-01 21:28:41

标签: python orm sqlalchemy pyramid

我有三个表:UserTypeMapper,User和SystemAdmin。在我的get_user方法中,根据UserTypeMapper.is_admin行,我然后查询User或SystemAdmin表。 user_id行与User和SystemAdmin表中的主键id相关联。

class UserTypeMapper(Base):
    __tablename__ = 'user_type_mapper'

    id = Column(BigInteger, primary_key=True)
    is_admin = Column(Boolean, default=False)
    user_id = Column(BigInteger, nullable=False)

class SystemAdmin(Base):
    __tablename__ = 'system_admin'

    id = Column(BigInteger, primary_key=True)
    name = Column(Unicode)
    email = Column(Unicode)

class User(Base):
    __tablename__ = 'user'

    id = Column(BigInteger, primary_key=True)
    name = Column(Unicode)
    email = Column(Unicode)

我希望能够从一个查询中获得任何用户 - 系统管理员或普通用户 - 因此我在UserSystemAdmin上进行加入,具体取决于is_admin行。例如:

DBSession.query(UserTypeMapper, SystemAdmin).join(SystemAdmin, UserTypeMapper.user_id==SystemAdmin.id).first()

DBSession.query(UserTypeMapper, User).join(User, UserTypeMapper.user_id==User.id).first()

这很好用;但是,我想访问这些,如下:

>>> my_admin_obj.is_admin
True
>>> my_admin_obj.name
Bob Smith

>>> my_user_obj.is_admin
False
>>> my_user_obj.name
Bob Stevens

目前,我必须指定:my_user_obj.UserTypeMapper.is_adminmy_user_obj.User.name。从我读过的内容来看,我需要映射表,这样我就不需要指定属性所属的表。我的问题是,我不知道如何指定这个,因为我有两个可能的表,name属性,例如,可能来自。

这是我所指的例子:Mapping a Class against Multiple Tables

我怎样才能做到这一点?谢谢。

1 个答案:

答案 0 :(得分:2)

您已经发现了为什么“双重目的外键”是antipattern

你有一个相关的问题,你没有完全指出;没有办法使用外键约束强制数据处于有效状态。你想确保UserTypeMapper中每一行都有一个东西,但是'某些'不是任何一个表。正式地,你想要一个功能依赖

  

user_type_mapper→(system_admin×1)∪(user×0)

但是大多数sql数据库都不允许你编写表达它的外键约束。

它看起来很复杂,因为很复杂

相反,让我们考虑一下 想要说的话; “每个system_admin应为user;或

  

system_adminuser

在sql中,将写入:

CREATE TABLE user (
    id INTEGER PRIMARY KEY,
    name VARCHAR,
    email VARCHAR
);

CREATE TABLE system_admin (
    user_id INTEGER PRIMARY KEY REFERENCES user(id)
);

或者,在sqlalchemy声明式样式

class User(Base):
    __tablename__ = 'user'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    email = Column(String)

class SystemAdmin(Base):
    __tablename__ = 'system_admin'
    user_id = Column(ForeignKey(User.id), primary_key=True)

这个架构允许我们提出什么样的问题?

  • “是否有一个名为'john doe'的SystemAdmin?”
>>> print session.query(User).join(SystemAdmin).filter(User.name == 'john doe').exists()

EXISTS (SELECT 1 
FROM "user" JOIN system_admin ON "user".id = system_admin.user_id 
WHERE "user".name = :name_1)
  • “有多少用户?有多少个系统管理员?”
>>> print session.query(func.count(User.id), func.count(SystemAdmin.user_id)).outerjoin(SystemAdmin)

SELECT count("user".id) AS count_1, count(system_admin.user_id) AS count_2 
FROM "user" LEFT OUTER JOIN system_admin ON "user".id = system_admin.user_id

我希望您能明白为什么上述内容可以优先考虑您在问题中描述的设计;但是在你不能做出选择的情况下(只有在那种情况下,如果你仍然觉得你有更好的东西,请改进你的问题),你仍然可以填写那些数据通过提供表的备用映射,将一个python对象放入一个非常难以使用的对象中;特别是在第一个等式中遵循粗略结构的那个。

我们需要提两次UserTypeMapper,一次为union的每一边,为此,我们需要给别名。

>>> from sqlalchemy.orm import aliased
>>> utm1 = aliased(UserTypeMapper)
>>> utm2 = aliased(UserTypeMapper)

对于联合体,将每个别名连接到相应的表:由于SystemAdminUser具有相同顺序的相同列,因此我们不需要详细描述它们,但是如果它们完全不同,我们需要通过明确提及每一列来使它们“联合兼容”;这是一个练习。

>>> utm_sa = Query([utm1, SystemAdmin]).join(SystemAdmin, (utm1.user_id == SystemAdmin.id) & (utm1.is_admin == True))
>>> utm_u = Query([utm2, User]).join(User, (utm2.user_id == User.id) & (utm2.is_admin == False))

然后我们一起加入他们......

>>> print utm_sa.union(utm_u)
SELECT anon_1.user_type_mapper_1_id AS anon_1_user_type_mapper_1_id, anon_1.user_type_mapper_1_is_admin AS anon_1_user_type_mapper_1_is_admin, anon_1.user_type_mapper_1_user_id AS anon_1_user_type_mapper_1_user_id, anon_1.system_admin_id AS anon_1_system_admin_id, anon_1.system_admin_name AS anon_1_system_admin_name, anon_1.system_admin_email AS anon_1_system_admin_email 
FROM (SELECT user_type_mapper_1.id AS user_type_mapper_1_id, user_type_mapper_1.is_admin AS user_type_mapper_1_is_admin, user_type_mapper_1.user_id AS user_type_mapper_1_user_id, system_admin.id AS system_admin_id, system_admin.name AS system_admin_name, system_admin.email AS system_admin_email 
FROM user_type_mapper AS user_type_mapper_1 JOIN system_admin ON user_type_mapper_1.user_id = system_admin.id AND user_type_mapper_1.is_admin = 1 UNION SELECT user_type_mapper_2.id AS user_type_mapper_2_id, user_type_mapper_2.is_admin AS user_type_mapper_2_is_admin, user_type_mapper_2.user_id AS user_type_mapper_2_user_id, "user".id AS user_id, "user".name AS user_name, "user".email AS user_email 
FROM user_type_mapper AS user_type_mapper_2 JOIN "user" ON user_type_mapper_2.user_id = "user".id AND user_type_mapper_2.is_admin = 0) AS anon_1

虽然理论上可以把这一切都包装成一个看起来有点像标准sqlalchemy orm东西的python类,但我肯定那样做。使用非表映射,特别是当它们不仅仅是简单的连接(这是一个联合)时,需要付出大量的工作才能获得零回报。