Return a QuerySet of documents instead
This commit is contained in:
@@ -26,7 +26,7 @@ class Folder(models.Model):
|
|||||||
|
|
||||||
@property
|
@property
|
||||||
def documents(self):
|
def documents(self):
|
||||||
return [folder_document.document for folder_document in self.folderdocument_set.all()]
|
return Document.objects.filter(folderdocument__folder=self)
|
||||||
|
|
||||||
def remove_document(self, document):
|
def remove_document(self, document):
|
||||||
folder_document = self.folderdocument_set.get(document=document)
|
folder_document = self.folderdocument_set.get(document=document)
|
||||||
|
|||||||
Reference in New Issue
Block a user