Change retrieve study output file to use .multipart extension (GoogleCloudPlatform#2345)

noerog · gguuss · commit 41161fbdad8c · 2019-08-22T12:30:47.000-07:00
diff --git a/healthcare/api-client/dicom/dicomweb.py b/healthcare/api-client/dicom/dicomweb.py
@@ -131,7 +131,10 @@ def dicomweb_retrieve_study(
     dicomweb_path = '{}/datasets/{}/dicomStores/{}/dicomWeb/studies/{}'.format(
         url, dataset_id, dicom_store_id, study_uid)
 
-    file_name = 'study.dcm'
+    # When specifying the output file, use an extension like ".multipart."
+    # Then, parse the downloaded multipart file to get each individual
+    # DICOM file.
+    file_name = 'study.multipart'
 
     # Make an authenticated API request
     session = get_session(service_account_json)
diff --git a/healthcare/api-client/dicom/dicomweb_test.py b/healthcare/api-client/dicom/dicomweb_test.py
@@ -159,15 +159,15 @@ def test_dicomweb_retrieve_study(test_dataset, test_dicom_store, capsys):
         study_uid)
 
     # Assert study was downloaded
-    assert os.path.isfile('study.dcm')
+    assert os.path.isfile('study.multipart')
 
     out, _ = capsys.readouterr()
 
     # Check that retrieve study worked
     assert 'Retrieved study' in out
 
     # Delete downloaded study
-    os.remove('study.dcm')
+    os.remove('study.multipart')
 
     dicomweb.dicomweb_delete_study(
         service_account_json,
