heckyel/mediagoblin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implement Collection.delete()

Browse Source 
Deleting a Collection should automatically delete all containing items.

Signed-off-by: Sebastian Spaeth <Sebastian@SSpaeth.de>
This commit is contained in:
Sebastian Spaeth 2012-11-28 16:15:46 +01:00
parent fdc34b8ba7
commit 242776e363
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mediagoblin/db/models.py
View File

@ -377,6 +377,10 @@ class MediaComment(Base, MediaCommentMixin):
class Collection(Base, CollectionMixin): class Collection(Base, CollectionMixin):
"""An 'album' or 'set' of media by a user.
On deletion, contained CollectionItems get automatically reaped via
SQL cascade"""
__tablename__ = "core__collections" __tablename__ = "core__collections"
id = Column(Integer, primary_key=True) id = Column(Integer, primary_key=True)
@ -386,11 +390,13 @@ class Collection(Base, CollectionMixin):
index=True) index=True)
description = Column(UnicodeText) description = Column(UnicodeText)
creator = Column(Integer, ForeignKey(User.id), nullable=False) creator = Column(Integer, ForeignKey(User.id), nullable=False)
# TODO: No of items in Collection. Badly named, can we migrate to num_items?
items = Column(Integer, default=0) items = Column(Integer, default=0)
get_creator = relationship(User) get_creator = relationship(User)
def get_collection_items(self, ascending=False): def get_collection_items(self, ascending=False):
#TODO, is this still needed with self.collection_items being available?
order_col = CollectionItem.position order_col = CollectionItem.position
if not ascending: if not ascending:
order_col = desc(order_col) order_col = desc(order_col)
@ -408,7 +414,10 @@ class CollectionItem(Base, CollectionItemMixin):
note = Column(UnicodeText, nullable=True) note = Column(UnicodeText, nullable=True)
added = Column(DateTime, nullable=False, default=datetime.datetime.now) added = Column(DateTime, nullable=False, default=datetime.datetime.now)
position = Column(Integer) position = Column(Integer)
in_collection = relationship("Collection") in_collection = relationship("Collection",
backref=backref(
"collection_items",
cascade="all, delete-orphan"))
get_media_entry = relationship(MediaEntry) get_media_entry = relationship(MediaEntry)