tobias/radicale
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Store collection's parent in database (bug #77)

Browse Source
This commit is contained in:
Guillaume Ayoub 2013-10-31 16:07:37 +01:00
parent a920518a26
commit 0c2c9c3a7e
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
radicale/storage/database.py
View File

@ -45,10 +45,7 @@ class DBCollection(Base):
__tablename__ = "collection"
path = Column(String, primary_key=True)
parent_path = Column(String, ForeignKey("collection.path"))
parent = relationship(
"DBCollection", backref="children", remote_side=[path])
parent_path = Column(String)
class DBItem(Base):
@ -152,6 +149,7 @@ class Collection(ical.Collection):
else:
db_collection = DBCollection()
db_collection.path = self.path
db_collection.parent_path = "/".join(self.path.split("/")[:-1])
self.session.add(db_collection)
for header in headers:
@ -197,10 +195,9 @@ class Collection(ical.Collection):
@classmethod
def children(cls, path):
session = Session()
if path:
children = session.query(DBCollection).get(path).children
else:
children = session.query(DBCollection).filter_by(parent=None).all()
children = (
session.query(DBCollection)
.filter_by(parent_path=path or "").all())
collections = [cls(child.path) for child in children]
session.close()
return collections
@ -212,7 +209,7 @@ class Collection(ical.Collection):
session = Session()
result = (
session.query(DBCollection)
.filter_by(parent_path=path).count() > 0)
.filter_by(parent_path=path or "").count() > 0)
session.close()
return result
@ -223,7 +220,7 @@ class Collection(ical.Collection):
session = Session()
result = (
session.query(DBItem)
.filter_by(collection_path=path).count() > 0)
.filter_by(collection_path=path or "").count() > 0)
session.close()
return result