Skip to content

Commit 1a5e2b7

Browse files
committed
new method should work for uploading files to either a dataset or a folder, most of these will be uploaded to folders following the 'collection becomes dataset' model
1 parent ff8c62c commit 1a5e2b7

File tree

1 file changed

+11
-14
lines changed

1 file changed

+11
-14
lines changed

scripts/migration/migrate.py

Lines changed: 11 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -640,29 +640,26 @@ def download_and_upload_file_to_folder(file, folder, dataset_v2_id, headers_v2):
640640

641641
# Upload the file to Clowder v2
642642
dataset_file_upload_endpoint = f"{CLOWDER_V2}/api/v2/datasets/{dataset_v2_id}/files"
643-
if folder:
643+
if folder is not None:
644+
# add folder if it is not None
644645
dataset_file_upload_endpoint += f"Multiple?folder_id={folder['id']}"
645-
response = requests.post(
646-
dataset_file_upload_endpoint,
647-
headers=headers_v2,
648-
files={"file": open(filename, "rb")},
649-
)
646+
response = requests.post(
647+
dataset_file_upload_endpoint,
648+
headers=headers_v2,
649+
files={"file": open(filename, "rb")},
650+
)
651+
if response.status_code == 200:
652+
print(f"Uploaded file: {filename} to dataset {dataset_v2_id}")
653+
return response.json().get("id")
650654
else:
651-
print(f"This file is not in a folder")
655+
print(f"Failed to upload file: {filename} to dataset {dataset_v2_id}")
652656

653657
# Clean up the local file after upload
654658
try:
655659
os.remove(filename)
656660
except Exception as e:
657661
print(f"Could not delete locally downloaded file: {filename}")
658662
print(e)
659-
660-
if response.status_code == 200:
661-
print(f"Uploaded file: {filename} to dataset {dataset_v2_id}")
662-
return response.json().get("id")
663-
else:
664-
print(f"Failed to upload file: {filename} to dataset {dataset_v2_id}")
665-
666663
return None
667664

668665

0 commit comments

Comments
 (0)