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