Change primary key in tomogram picks

Author: stephen-riggs

src/murfey/util/db.py

@@ -681,9 +681,8 @@ class CtfParameters(SQLModel, table=True):  # type: ignore
 
 
 class TomogramPicks(SQLModel, table=True):  # type: ignore
-    id: Optional[int] = Field(default=None, primary_key=True)
+    tomogram: str = Field(primary_key=True)
     pj_id: int = Field(foreign_key="processingjob.id")
-    tomogram: str
     cbox_3d: str
     particle_count: int
     tomogram_pixel_size: float

src/murfey/workflows/tomo/picking.py

@@ -63,7 +63,6 @@ def _register_picked_tomogram_use_diameter(message: dict, _db: Session):
     )
     _db.add(pick_params)
     _db.commit()
-    _db.close()
 
     picking_db_len = _db.exec(
         select(func.count(ParticleSizes.id)).where(ParticleSizes.pj_id == pj_id)
@@ -150,6 +149,7 @@ def _register_picked_tomogram_use_diameter(message: dict, _db: Session):
                         "processing_recipe", zocalo_message, new_connection=True
                     )
                 feedback_params.next_job += 2
+                _db.delete(saved_message)
         else:
             # If the diameter is known then just send the new message
             particle_diameter = tomo_params.particle_diameter