use crystal plate object to get well count

Author: David Erb

src/rockingester_lib/collectors/direct_poll.py

@@ -21,6 +21,9 @@
 # Crystal well pydantic model.
 from xchembku_api.models.crystal_well_model import CrystalWellModel
 
+# Crystal plate objects factory.
+from xchembku_lib.crystal_plate_objects.crystal_plate_objects import CrystalPlateObjects
+
 # Base class for collector instances.
 from rockingester_lib.collectors.base import Base as CollectorBase
 
@@ -289,8 +292,13 @@ async def scrape_plate_directory_when_complete(
             entry.name for entry in os.scandir(plate_directory) if entry.is_file()
         ]
 
+        # Make an object corresponding to the crystal plate model's type.
+        crystal_plate_object = CrystalPlateObjects().build_object(
+            {"type": crystal_plate_model.thing_type}
+        )
+
         # Don't handle the plate directory until all images have arrived.
-        if len(well_names) < 288:
+        if len(well_names) < crystal_plate_object.get_well_count():
             return
 
         # Sort wells by name so that tests are deterministic.

tests/test_collector.py

@@ -5,6 +5,11 @@
 
 from dls_utilpack.visit import get_xchem_subdirectory
 
+# Types which the CrystalPlateObjects factory can use to build an instance.
+from xchembku_api.crystal_plate_objects.constants import (
+    ThingTypes as CrystalPlateObjectThingTypes,
+)
+
 # Things xchembku provides.
 from xchembku_api.datafaces.context import Context as XchembkuDatafaceClientContext
 from xchembku_api.datafaces.datafaces import xchembku_datafaces_get_default
@@ -141,6 +146,7 @@ async def __run_part1(self, scrapable_image_count, constants, output_directory):
                 formulatrix__plate__id=10,
                 barcode="98ab",
                 visit=visit,
+                thing_type=CrystalPlateObjectThingTypes.SWISS3,
             )
         )