wip:

- Passing test for raw ORM handling to find snapshot files from an
  object
- Quietens down pytest output

1 files changed, 9 insertions, 0 deletions

diff --git a/pyblackbird_cc/resources/models.py b/pyblackbird_cc/resources/models.py
index 120ecdf..2b73c89 100644
--- a/pyblackbird_cc/resources/models.py
+++ b/pyblackbird_cc/resources/models.py
@@ -79,6 +79,15 @@ class Resource(models.Model):
     def __str__(self):
         return self.name
 
+    def get_pdf_file_names(self):
+        return [p.file_name for p in self.pdf_resources.all()]
+
+    def get_pdf_snapshot_file_names(self):
+        rs = self.pdf_resources.all()
+        sh = [sh for sh in rs.values_list("pdf_page_snapshots__file_name", flat=True)]
+        out = [s for s in sh if s]
+        return out
+
 
 class ResourceType(models.Model):
     name = models.CharField(max_length=255, null=False)