Add the remove_document method to the Folder class
This commit is contained in:
@@ -28,6 +28,10 @@ class Folder(models.Model):
|
|||||||
def documents(self):
|
def documents(self):
|
||||||
return [folder_document.document for folder_document in self.folderdocument_set.all()]
|
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:
|
class Meta:
|
||||||
unique_together = ('title', 'user')
|
unique_together = ('title', 'user')
|
||||||
ordering = ('title',)
|
ordering = ('title',)
|
||||||
|
|||||||
Reference in New Issue
Block a user