X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/02f8116c468682e76e7d7d81ab345c92d8188152..049c5959e33d29e4c0044c235d291144d49d3c45:/src/sources/models.py

diff --git a/src/sources/models.py b/src/sources/models.py
index 0b55c890..165cb67b 100644
--- a/src/sources/models.py
+++ b/src/sources/models.py
@@ -38,6 +38,10 @@ class Source(models.Model):
     def get_ocr_directory(self):
         return f'sources/ocr/{self.pk}/'
 
+    def has_upload_files(self):
+        d = os.path.join(settings.MEDIA_ROOT, self.get_upload_directory())
+        return os.path.isdir(d) and os.listdir(d)
+    
     def get_view_files(self):
         d = self.get_view_directory()
         return [
@@ -47,6 +51,10 @@ class Source(models.Model):
             ))
         ]
 
+    def has_view_files(self):
+        d = os.path.join(settings.MEDIA_ROOT, self.get_view_directory())
+        return os.path.isdir(d) and os.listdir(d)
+    
     def get_ocr_files(self):
         d = os.path.join(settings.MEDIA_ROOT, self.get_ocr_directory())
         return [
@@ -54,6 +62,10 @@ class Source(models.Model):
             for name in sorted(os.listdir(d))
         ]
 
+    def has_ocr_files(self):
+        d = os.path.join(settings.MEDIA_ROOT, self.get_ocr_directory())
+        return os.path.isdir(d) and os.listdir(d)
+
     def process(self):
         updir = os.path.join(
             settings.MEDIA_ROOT,