使用带有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进行操作,如果不行,我很想听听关于其他方法的建议。
答案 0 :(得分:1)
在类Db
中,我认为唯一索引就足够了
__table_args__ = (UniqueConstraint('parent_id', 'name'), )