Add the remove_document method to the Folder class

This commit is contained in:
Roberto Rosario
2012-01-02 05:41:30 -04:00
parent 14ac963fca
commit 7a14cd1854

View File

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