Added new param to utils.get_max_download_date

Author: sgeulette

src/imio/esign/tests/test_utils.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 """utils tests for this package."""
+from datetime import date
 from datetime import timedelta
 from imio.esign.testing import IMIO_ESIGN_INTEGRATION_TESTING  # noqa: E501
 from imio.esign.utils import add_files_to_session
@@ -327,8 +328,10 @@ def test_get_file_download_url(self):
     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), mod_date + timedelta(days=90))
         self.assertEqual(get_max_download_date(annex, timedelta(days=0)), mod_date)
+        today = date.today()
+        self.assertEqual(get_max_download_date(None, timedelta(days=0), today), today)
 
 
 # example of annotation content

src/imio/esign/utils.py

@@ -402,12 +402,14 @@ def get_file_download_url(uid, separator="-", block_size=5, root_url=None):
     return "{}/{}".format(root_url.strip('/'), short_uid), short_uid
 
 
-def get_max_download_date(obj, delta=timedelta(days=120)):
+def get_max_download_date(obj, delta=timedelta(days=90), adate=None):
     """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
+    :param adate: use date instead of modification date
     :return: maximum download date
     """
-    mod_date = obj.modified().asdatetime().date()
-    return mod_date + delta
+    if adate is None:
+        adate = obj.modified().asdatetime().date()
+    return adate + delta