From d50f0f9910a71428205f7c8dbf5ab958cdf3dc9c Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Mon, 2 Jan 2012 06:44:12 -0400 Subject: [PATCH] Add an 'add_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 0f48eead9a..20c2eda7a7 100644 --- a/apps/folders/models.py +++ b/apps/folders/models.py @@ -31,6 +31,10 @@ class Folder(models.Model): def remove_document(self, document): folder_document = self.folderdocument_set.get(document=document) folder_document.delete() + + def add_document(self, document): + folder_document, created = FolderDocument.objects.get_or_create(folder=self, document=document) + return created class Meta: unique_together = ('title', 'user')