Return a QuerySet of documents instead

2012-02-02 00:38:58 -04:00
parent 556c0824fb
commit 4e3a16da82
1 changed files with 1 additions and 1 deletions
--- a/apps/folders/models.py
+++ b/apps/folders/models.py
@@ -26,7 +26,7 @@ class Folder(models.Model):

    @property
    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):
        folder_document = self.folderdocument_set.get(document=document)