SQLAlchemy 自动映射数据并覆盖某些列

Question

是否可以在 SQLAlchemy 中将现有的数据库表自动映射到类中但覆盖某些表的某些字段？

我正在腌制 MetaData 对象，因为它需要一些时间，它包含所有表，但是当我尝试覆盖某些对象时，它会引发 MetaData 对象未绑定到引擎或连接的异常.

from sqlalchemy.ext.automap import automap_base
from sqlalchemy import create_engine, MetaData, Column, String, Integer
import os, pickle

class MetadataCache(object):
    def __init__(self, engine, schema):
        self.engine = engine
        self.schema = schema
        self.metadata = None

    @property
    def cache_name(self):
        final_name = '{0}.{1}.cache'.format(self.engine.url.database,
                                            self.schema)
        return final_name

    def get_or_create_metadata(self):

        if os.path.exists(self.cache_name):
            with open(self.cache_name, 'r') as cachefile:
                self.metadata = pickle.load(cachefile)
        else:
            self.metadata = MetaData()
            self.metadata.reflect(bind=self.engine, schema=self.schema)
            with open(self.cache_name, 'w') as cachefile:
                pickle.dump(self.metadata, cachefile)

        return self.metadata

engine = create_engine('...')
metadata = MetadataCache(engine, 'schemaname').get_or_create_metadata()
Base = automap_base(metadata=metadata)


class User(Base):
    __tablename__ = 'user'

    id = Column('id', Integer, primary_key=True)
    name = Column('name', String)

class Profile(Base):
    __tablename__ = 'profile'
    id = Column('id', Integer, primary_key=True)
    userid = Column('userid', ForeignKey('user.id'))

Base.prepare(reflect=True)

Answer 1

引发错误是因为 Base 拥有的元数据（您传递给 automap_base 构造函数）未绑定到引擎，也不应该绑定到引擎。这只是意味着您必须将绑定传递给Base.prepare。

此外，由于您是手动反映元数据，因此不应告诉 Base.prepare 这样做。

将最后一行更改为：

Base.prepare(engine)

编辑：当然，我在考虑 Base.prepare 需要引擎进行反射，但由于您传递的是预反射 MetaData()，因此您可以使用：

Base.prepare()

编辑：

在回复下面的评论时，请注意必须将extend_existing=True 传递给Table：

class User(Base):
    __tablename__ = 'user'
    __table_args__ = {'extend_existing': True}

    id = Column('id', Integer, primary_key=True)
    name = Column('name', String)