在SQLAlchemy中连接表时,是否有一种方便的方法来只对冲突的列进行别名?

StackOverflow https://stackoverflow.com/questions/1627429

  •  06-07-2019
  •  | 
  •  

使用 join 而不是单个表是有用的> SQLAlchemy 的声明性扩展。当列名冲突时,通常是一对多,因为默认情况下所有主键都命名为 id ,您可以使用 .alias()为每个列添加前缀表名。如果您已经编写了假定您的映射类具有非前缀名称的代码,则这很不方便。

例如:

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Table, Column, Integer, ForeignKeyConstraint

Base = declarative_base()

t1 = Table('t1', 
    Base.metadata, 
    Column('id', Integer, primary_key=True))

t2 = Table('t2', 
    Base.metadata, 
    Column('id', Integer, primary_key=True),
    Column('fkey', Integer),
    ForeignKeyConstraint(['fkey'], [t1.c.id]))

class ST(Base):
    __table__ = t1.join(t2)

class ST2(Base):
    __table__ = t1.join(t2).alias()

ST 具有 id fkey 属性,每个名称映射到使用重写名称的连接中的第一个表,因此映射class不公开 t2 的主键。 ST2 具有 t1_id t2_id t2_fkey 属性。

是否有一种方便的方法来只对 join 中每个表的某些列进行别名,这样映射的类为大多数映射列公开了更方便的非前缀属性名称?

有帮助吗?

解决方案

您可以使用 label()方法分别为每个列创建别名。所以它可能类似于以下内容(未经测试):

from sqlalchemy import select

def alias_dups(join):
    dups = set(col.key for col in join.left.columns) & \
                set(col.key for col in join.right.columns)
    columns = []
    for col in join.columns:
        if col.key in dups:
            col = col.label('%s_%s' % (col.table.name, col.key))
        columns.append(col)
    return select(columns, from_obj=[join]).alias()

class ST2(Base):
    __table__ = alias_dups(t1.join(t2))
许可以下: CC-BY-SA归因
不隶属于 StackOverflow
scroll top