Add support for document retention policies. Closes gh-issue #189.

mayan/apps/documents/tasks.py

@@ -1,15 +1,19 @@
 from __future__ import unicode_literals
 
+from datetime import timedelta
 import logging
 
 from django.contrib.auth.models import User
 from django.core.files import File
+from django.utils.timezone import now
 
 from mayan.celery import app
 
 from common.models import SharedUploadedFile
 
-from .models import Document, DocumentPage, DocumentType, DocumentVersion
+from .models import (
+    DeletedDocument, Document, DocumentPage, DocumentType, DocumentVersion
+)
 
 logger = logging.getLogger(__name__)
 
@@ -77,3 +81,41 @@ def task_upload_new_version(document_id, shared_uploaded_file_id, user_id, comme
             logger.info('Warning during attempt to create new document version for document:%s ; %s', document, warning)
         finally:
             shared_file.delete()
+
+
+@app.task(ignore_result=True)
+def task_check_trash_periods():
+    logger.info('Executing')
+
+    for document_type in DocumentType.objects.all():
+        logger.info('Checking trash period of document type: %s', document_type)
+        if document_type.trash_time_period and document_type.trash_time_unit:
+            delta = timedelta(**{document_type.trash_time_unit: document_type.trash_time_period})
+            logger.info('Document type: %s, has a trash period delta of: %s', document_type, delta)
+            for document in Document.objects.filter(document_type=document_type):
+                if now() > document.date_added + delta:
+                    logger.info('Document "%s" with id: %d, added on: %s, exceded trash period', document, document.pk, document.date_added)
+                    document.delete()
+        else:
+            logger.info('Document type: %s, has a no retention delta', document_type)
+
+    logger.info('Finshed')
+
+
+@app.task(ignore_result=True)
+def task_check_delete_periods():
+    logger.info('Executing')
+
+    for document_type in DocumentType.objects.all():
+        logger.info('Checking deletion period of document type: %s', document_type)
+        if document_type.delete_time_period and document_type.delete_time_unit:
+            delta = timedelta(**{document_type.delete_time_unit: document_type.delete_time_period})
+            logger.info('Document type: %s, has a deletion period delta of: %s', document_type, delta)
+            for document in DeletedDocument.objects.filter(document_type=document_type):
+                if now() > document.deleted_date_time + delta:
+                    logger.info('Document "%s" with id: %d, trashed on: %s, exceded delete period', document, document.pk, document.deleted_date_time)
+                    document.delete()
+        else:
+            logger.info('Document type: %s, has a no retention delta', document_type)
+
+    logger.info('Finshed')