Added short_uid param in utils.get_file_download_url

Author: sgeulette

src/imio/esign/tests/test_utils.py

@@ -325,6 +325,10 @@ def test_get_file_download_url(self):
         self.assertTrue(result2[0].startswith("https://downloads.files.com/"))
         self.assertEqual(shortuid_decode_id(result2[1], separator="-"), uid2)  # correctly decoded
 
+        # Test with pre-computed short_uid parameter
+        result3 = get_file_download_url(None, short_uid="MyCustom-Short-UID")
+        self.assertEqual(result3, ("https://downloads.files.com/MyCustom-Short-UID", "MyCustom-Short-UID"))
+
     def test_get_max_download_date(self):
         annex = self.folders[0]
         mod_date = annex.modified().asdatetime().date()

src/imio/esign/utils.py

@@ -384,21 +384,23 @@ def remove_session(session_id):
     # logger.info("Session %s removed", session_id)
 
 
-def get_file_download_url(uid, separator="-", block_size=5, root_url=None):
+def get_file_download_url(uid, separator="-", block_size=5, root_url=None, short_uid=None):
     """Get the file download URL for a given file UID.
 
     :param uid: file UID
     :param separator: separator used in short UID encoding
     :param block_size: block size used in short UID encoding
     :param root_url: root URL. If not provided, the settings value is used
+    :param short_uid: take this short UID instead of computing it
     :return: file download URL, short_uid
     """
     if not root_url:
         root_url = get_registry_file_url()
 
     if not root_url:
         raise Exception("No root URL provided for file download url.")
-    short_uid = shortuid_encode_id(uid, separator=separator, block_size=block_size)
+    if not short_uid:
+        short_uid = shortuid_encode_id(uid, separator=separator, block_size=block_size)
     return "{}/{}".format(root_url.strip('/'), short_uid), short_uid