如何在SQLAlchemy column_property上创建索引?

时间:2019-02-19 23:25:07

标签: python sql sqlite sqlalchemy flask-sqlalchemy

使用带有SQLite引擎的SQLAlchemy,我有一个描述目录结构的自引用层次结构表。

from sqlalchemy import Column, Integer, String, ForeignKey, Index
from sqlalchemy.orm import column_property, aliased, join
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class Dr(Base):
    __tablename__ = 'directories'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    parent_id = Column(Integer, ForeignKey('directories.id'))

每个博士行仅知道其自己的“名称”及其“ parent_id”。我添加了一个名为“ path”的递归column_property,该递归column_property返回了一个字符串,其中包含来自根Dr的所有Dr的祖先。

root_anchor = (
    select([Dr.id, Dr.name, Dr.parent_id,Dr.name.label('path')])
    .where(Dr.parent_id == None).cte(recursive=True)
)

dir_alias = aliased(Dr)
cte_alias = aliased(root_anchor)

path_table = root_anchor.union_all(
    select([
        dir_alias.id, dir_alias.name,
        dir_alias.parent_id, cte_alias.c.path + "/" + dir_alias.name
    ]).select_from(join(
        dir_alias, cte_alias, onclause=cte_alias.c.id==dir_alias.parent_id)
    ))
)

Dr.path = column_property(
    select([path_table.c.path]).where(path_table.c.id==Dr.id)
)

这是输出示例:

"""
-----------------------------
| id |   name   | parent_id |
-----------------------------
|  1 | root     | NULL      |
-----------------------------
|  2 | kid      | 1         |
-----------------------------
|  3 | grandkid | 2         |
-----------------------------
"""
sqllite_engine = create_engine('sqlite:///:memory:')
Session = sessionmaker(bind=sqllite_engine)
session = Session()

instance = session.query(Dr).filter(Dr.name=='grandkid').one()
print(instance.path)
# Outputs: "root/kid/grandkid" 

我希望能够在“ path”属性上添加索引或至少一个唯一约束,以使唯一路径在表中不能存在多次。我尝试过:

Index('pathindex', Directory.path, unique=True)

...没有运气。没有引发错误,但是SQLAlchemy似乎没有注册索引,只是默默地忽略了它。它仍然允许添加重复路径,例如:

session.add(Dr(name='grandkid', parent_id=2))
session.commit()

作为进一步的证据表明Index()被忽略,检查表的“ indexes”属性将导致一个空集:

print(Dr.__table__.indexes)
#Outputs: set([])

对我来说至关重要的是,数据库中不能存在重复的路径。我不确定在SQLAlchemy中我是否可以使用column_property进行操作,如果不行,我很想听听关于其他方法的建议。

1 个答案:

答案 0 :(得分:1)

在类Db中,我认为唯一索引就足够了

__table_args__ = (UniqueConstraint('parent_id', 'name'), )