Question

我有一个用例，我希望能够使用声明性API部分定义一个表，然后再添加一些与我在定义表类后自动生成的主键相对应的列属性。 / p>

例如，假设我以这张桌子为起点：

Base = declarative_base(cls=DeferredReflection)

class my_table(Base):
    __tablename__ = "my_table"
    __table_args__ = {'schema': 'my_schema'}

    @hybrid_property
    def my_attr(self):
        return func.coalesce(self.attr_1, self.attr_2, self.attr_3)

该表没有主键，但是它具有一个有效地为主键的主索引。默认情况下，SQLAlchemy不使用主索引作为主键。因此，我想做的就是尝试从表上已经可用的主索引中构建主键。

这是我为此编写的课程：

class AddPrimaryKeys():
    '''
    This class provides a function that will automatically add primary key
    definitions using the established primary index.
    '''
    def __init__(self, inspector):
        self.inspector = inspector

    def __call__(self, table, metadata):
        pkey_cols = self.__build_pkey_from_index(table)
        return self.__add_primary_keys(table, pkey_cols, metadata)

    def __build_pkey_from_index(self, table):
        tablename = table.__tablename__
        schema = table.__table_args__['schema']
        indices = self.inspector.get_indexes(tablename, schema=schema)
        pkey_info = self.__get_column_info(tablename, schema, indices)
        pkey_cols = self.__build_columns(pkey_info, table)
        return pkey_cols

    def __get_column_info(self, tablename, schema, indices):
        columns = self.inspector.get_columns(tablename, schema=schema)
        index_names = indices[0]['column_names']
        pkey_info = []
        for c in columns:
            if c['name'] in index_names:
                pkey_info.append((c['name'], c['type']))
        return pkey_info

    def __build_columns(self, pkey_info, table):
        pkey_cols = []
        for key_name, key_type in pkey_info:
            column = Column(key_name, key_type, primary_key=True)
            pkey_cols.append(column)
        print(pkey_cols)
        return pkey_cols

    def __add_primary_keys(self, table, pkey_cols, metadata):
        tablename = table.__tablename__
        schema = table.__table_args__['schema']
        table_map = Table(name=tablename, metadata=metadata, schema=schema, *pkey_cols)
        # properties={'primary_key':pkey_cols}
        # mapper(table, table_map, properties=properties)
        mapper(table, table_map)
        return metadata

我这样叫课：

insp = inspect(td_engine)
from lib.pkey_lookup import AddPrimaryKeys
add_pkeys = AddPrimaryKeys(insp)
add_pkeys(my_table, Base.metadata)

这是列列表的外观：

[Column('some_attr', DECIMAL(precision=18), table=None, primary_key=True, nullable=False), Column('another_attr', DECIMAL(precision=18), table=None, primary_key=True, nullable=False)]

这将引发以下错误：

AttributeError: Neither 'Column' object nor 'Comparator' object has an attribute 'tables'

您还可以看到，我只是在尝试声明primary_key列作为属性dict的一部分，在这种情况下，我将丢弃整个__build_columns方法。

是否有人对如何解决该错误有任何建议，也许是一种将insp.get_indexes()所标识的列分配为表上的主键的更好方法？

编辑：最后还应该添加一点，我想调用Base.prepare(engine)来填充数据库中的其余属性或表属性。

SQLAlchemy：将表映射到半配置的类

0 个答案: