Added max download

Author: sgeulette

src/imio/esign/tests/test_utils.py

@@ -1,8 +1,10 @@
 # -*- coding: utf-8 -*-
 """utils tests for this package."""
+from datetime import timedelta
 from imio.esign.testing import IMIO_ESIGN_INTEGRATION_TESTING  # noqa: E501
 from imio.esign.utils import add_files_to_session
 from imio.esign.utils import get_file_uid_url
+from imio.esign.utils import get_max_download_date
 from imio.esign.utils import get_session_annotation
 from imio.esign.utils import remove_context_from_session
 from imio.esign.utils import remove_files_from_session
@@ -322,6 +324,12 @@ def test_get_file_uid_url(self):
         suid = result2[len("https://downloads.files.com/") :]
         self.assertEqual(shortuid_decode_id(suid, separator="-"), uid2)  # correctly decoded
 
+    def test_get_max_download_date(self):
+        annex = self.folders[0]
+        mod_date = annex.modified().asdatetime().date()
+        self.assertEqual(get_max_download_date(annex), mod_date + timedelta(days=120))
+        self.assertEqual(get_max_download_date(annex, timedelta(days=0)), mod_date)
+
 
 # example of annotation content
 """

src/imio/esign/utils.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 from datetime import datetime
+from datetime import timedelta
 from imio.esign import _tr as _
 from imio.esign import ESIGN_CREDENTIALS
 from imio.esign import ESIGN_ROOT_URL
@@ -395,3 +396,14 @@ def get_file_uid_url(uid, separator="-", block_size=5, root_url=None):
         raise Exception("No root URL provided for file download url.")
     short_uid = shortuid_encode_id(uid, separator=separator, block_size=block_size)
     return "{}/{}".format(root_url.strip('/'), short_uid)
+
+
+def get_max_download_date(obj, delta=timedelta(days=120)):
+    """Get the maximum download date for e-sign files. Is takes the modification date and adds delta.
+
+    :param obj: content object
+    :param delta: timedelta to add to modification date
+    :return: maximum download date
+    """
+    mod_date = obj.modified().asdatetime().date()
+    return mod_date + delta