From 7a14cd1854eb2f36bb3f5142c554fd3b43ca388f Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Mon, 2 Jan 2012 05:41:30 -0400 Subject: [PATCH] Add the remove_document method to the Folder class --- apps/folders/models.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/apps/folders/models.py b/apps/folders/models.py index f2a83ff32c..0f48eead9a 100644 --- a/apps/folders/models.py +++ b/apps/folders/models.py @@ -27,6 +27,10 @@ class Folder(models.Model): @property def documents(self): return [folder_document.document for folder_document in self.folderdocument_set.all()] + + def remove_document(self, document): + folder_document = self.folderdocument_set.get(document=document) + folder_document.delete() class Meta: unique_together = ('title', 'user')