Make it easy to customise expiration for cloudfront

jschneier · Mar 28, 2023 · ec64e58 · ec64e58
1 parent 90e2e78
commit ec64e58
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/storages/backends/s3boto3.py b/storages/backends/s3boto3.py
@@ -561,6 +561,9 @@ def _strip_signing_parameters(self, url):
         joined_qs = ('='.join(keyval) for keyval in filtered_qs)
         split_url = split_url._replace(query='&'.join(joined_qs))
         return split_url.geturl()
+
+    def _get_expiration_for_cloudfront(self, expire):
+        return datetime.utcnow() + timedelta(seconds=expire)
 
     def url(self, name, parameters=None, expire=None, http_method=None):
         # Preserve the trailing slash after normalizing the path.
@@ -578,7 +581,7 @@ def url(self, name, parameters=None, expire=None, http_method=None):
             )
 
             if self.querystring_auth and self.cloudfront_signer:
-                expiration = datetime.utcnow() + timedelta(seconds=expire)
+                expiration = self._get_expiration_for_cloudfront(expire)
                 return self.cloudfront_signer.generate_presigned_url(url, date_less_than=expiration)
 
             return url