fix for dlname not getting set on dmcbucket files

broadinstitute · Dec 7, 2024 · 279959a · 279959a
1 parent f378088
commit 279959a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/portal-backend/depmap/download/models.py b/portal-backend/depmap/download/models.py
@@ -180,8 +180,8 @@ def __init__(self, file_name):
 class DmcBucketUrl(BucketUrl):
     BUCKET = "depmap-dmc-downloads"
 
-    def __init__(self, file_name):
-        super().__init__(DmcBucketUrl.BUCKET, file_name)
+    def __init__(self, file_name, dl_name=None):
+        super().__init__(DmcBucketUrl.BUCKET, file_name=file_name, dl_name=dl_name)
 
     def __repr__(self):
         return "DmcBucketUrl({})".format(repr(self.file_name))

diff --git a/portal-backend/depmap/settings/parse_downloads.py b/portal-backend/depmap/settings/parse_downloads.py
@@ -71,10 +71,12 @@ def get_summary_stats(stats: List[Dict[str, Any]]) -> SummaryStats:
 
 def get_bucket(url: dict):
     if url.get("bucket", "") == DmcBucketUrl.BUCKET:
-        return DmcBucketUrl(url.get("file_name", ""))
+        return DmcBucketUrl(url.get("file_name", ""), dl_name=url.get("dl_name", ""))
     else:
         return BucketUrl(
-            url.get("bucket", ""), url.get("file_name", ""), url.get("dl_name", "")
+            url.get("bucket", ""),
+            url.get("file_name", ""),
+            dl_name=url.get("dl_name", ""),
         )