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')