SQLAlchemy的自动映射backref错误

Question

我是新来的SQLAlchemy（奥姆斯在通常情况下），我试图在移动现有的应用程序SQLAlchemy的，这样我们就可以从现有（和繁琐的更新）查询转移一些代码的复杂性，到Python。不幸的是，我在数据库反射后立即出现错误。虽然我可以直接查询表，但我实际上没有直接访问类或类之间的关系。下面是我正在尝试做的一个大约最小的例子。

现存的Postgres数据表：

dev=> \d+ gmt_file 
             Table "public.gmt_file" 
    Column |  Type  | Modifiers | Storage | Stats target | Description 
-----------+--------------+-----------+----------+--------------+------------- 
file_id | integer  | not null | plain |    | 
     a | integer  |   | plain |    | 
     b | integer  |   | plain |    | 
Indexes: 
    "gmt_file_pk" PRIMARY KEY, btree (file_id) 
Foreign-key constraints: 
    "gmt_file_a_fk" FOREIGN KEY (a) REFERENCES cmn_user(user_id) 
    "gmt_file_b_fk" FOREIGN KEY (b) REFERENCES cmn_user(user_id) 

SQLAlchemy的应用程序（小例子）：

from sqlalchemy import create_engine 
from sqlalchemy.orm import Session,Mapper 
from sqlalchemy.ext.automap import automap_base 

engine = create_engine('postgresql://user:[email protected]:5432/dev') 
Base = automap_base() 
Base.prepare(engine, reflect=True) 
session = Session(engine,autocommit=True) 

session.query(Base.classes.gmt_file).all() 

从我目前可以告诉，这将引发因为两者a和b具有一个backref错误与不同表中的相同字段的外键关系（这通常发生在现有数据库中）。我尝试了多种处理此错误的方法，包括创建自定义命名函数（name_for_scalar_relationship()和name_for_collection_relationship()），但无济于事。有没有一种标准的方法来处理这个问题，或者在SQLAlchemy中禁用反射过程中的backref创建？

的最终目标将是反映以自动化的方式分贝，而不必编写自定义名称为数百个目前存在表的映射，但是我在适合做什么损失。任何帮助表示赞赏。

谢谢

Answer 1

看起来我们得使用自动映射，当一个以上的外键是指在同一列的属性名称冲突。

Base.prepare允许参数name_for_scalar_relationship和name_for_collection_relationship，它们采用用于生成属性名称的函数。 （请参阅AutomapBase.prepare()和name_for_collection_relationship()的文档）我能够通过定义自己的函数来解决backref错误。

修改你的小例子：

from sqlalchemy import create_engine 
from sqlalchemy.orm import Session,Mapper 
from sqlalchemy.ext.automap import automap_base, name_for_collection_relationship 

engine = create_engine('postgresql://user:[email protected]:5432/dev') 
Base = automap_base() 

def _name_for_collection_relationship(base, local_cls, referred_cls, constraint): 
    if constraint.name: 
     return constraint.name.lower() 
    # if this didn't work, revert to the default behavior 
    return name_for_collection_relationship(base, local_cls, referred_cls, constraint) 

Base.prepare(engine, reflect=True, name_for_collection_relationship=_name_for_collection_relationship) 
session = Session(engine,autocommit=True) 

session.query(Base.classes.gmt_file).all() 

这样应该可以让你与属性名称gmt_file_a_fk和gmt_file_b_fk你的类。

此方法为我工作。如果它不起作用，您也可以尝试类似地重新定义name_for_scalar_relationship()。

如果你想在所有覆盖类，你必须确保正确定义的列和关系。例如：

from sqlalchemy import Column, Integer, ForeignKey 
from sqlalchemy.orm import relationship 

class GmtFile(Base): 
    __tablename___ = 'gmt_file' 

    file_id = Column('file_id', Integer, primary_key=True) 

    a = Column('a', Integer, ForeignKey('CmnUser.user_id', name='gmt_file_a')) 
    b = Column('b', Integer, ForeignKey('CmnUser.user_id', name='gmt_file_b')) 
    # you'll need to define the class CmnUser as well 

    # these variable names need to be the same as the ForeignKey names above 
    gmt_file_a = relationship('CmnUser', foreign_keys=[a]) 
    gmt_file_b = relationship('CmnUser', foreign_keys=[b])