DCGs are based on atlas jpg name

Author: stephen-riggs

src/murfey/client/contexts/atlas.py

@@ -59,9 +59,9 @@ def post_transfer(
             source = _get_source(transferred_file, environment)
             if source:
                 atlas_mrc = transferred_file.with_suffix(".mrc")
-                transferred_atlas_name = _atlas_destination(
+                transferred_atlas_jpg = _atlas_destination(
                     environment, source, atlas_mrc, self._token
-                ) / atlas_mrc.relative_to(source.parent)
+                ) / atlas_mrc.relative_to(source.parent).with_suffix(".jpg")
 
                 with open(transferred_file, "rb") as atlas_xml:
                     atlas_xml_data = xmltodict.parse(atlas_xml)
@@ -87,7 +87,7 @@ def post_transfer(
                 dcg_data = {
                     "experiment_type_id": 44,  # Atlas
                     "tag": str(transferred_file.parent),
-                    "atlas": str(transferred_atlas_name).replace("//", "/"),
+                    "atlas": str(transferred_atlas_jpg).replace("//", "/"),
                     "sample": sample,
                     "atlas_pixel_size": atlas_pixel_size,
                 }
@@ -101,5 +101,5 @@ def post_transfer(
                     data=dcg_data,
                 )
                 logger.info(
-                    f"Registered data collection group for atlas {str(transferred_atlas_name)!r}"
+                    f"Registered data collection group for atlas {str(transferred_atlas_jpg)!r}"
                 )

tests/client/contexts/test_atlas.py

@@ -77,7 +77,7 @@ def test_atlas_context_xml(mock_capture_post, tmp_path):
     dcg_data = {
         "experiment_type_id": 44,  # Atlas
         "tag": str(atlas_xml.parent),
-        "atlas": f"{tmp_path}/destination/{atlas_xml.relative_to(tmp_path).with_suffix('.mrc')}",
+        "atlas": f"{tmp_path}/destination/{atlas_xml.relative_to(tmp_path).with_suffix('.jpg')}",
         "sample": 2,
         "atlas_pixel_size": atlas_pixel_size * 7.8,
     }