Modify allowed proposals for trigger_xchem

Author: rowanwalker96

src/dlstbx/services/trigger_xchem.py

@@ -236,18 +236,17 @@ def trigger_pandda_xchem(
 
         query = (session.query(Proposal)).filter(Proposal.proposalId == proposal_id)
         proposal = query.first()
+        proposal_code = proposal.proposalCode
+        proposal_number = proposal.proposalNumber
+        proposal_string = proposal_code + proposal_number
 
-        # 0. Check that this is an XChem expt & locate .SQLite database
-        if proposal.proposalCode not in {"lb"}:
-            self.log.debug(
-                f"Not triggering PanDDA2 pipeline for dcid={dcid} with proposal_code={proposal.proposalCode}"
-            )
-            return {"success": True}
+        # TEMPORARY FILTER
+        allowed_proposals = ["lb42888", "sw44107"]
 
-        # TEMPORARY, FILTER BY OPENBIND VISIT
-        if proposal.proposalNumber not in {"42888"}:
+        # 0. Check that this is an XChem expt & locate .SQLite database
+        if proposal_string not in allowed_proposals:
             self.log.debug(
-                f"Not triggering PanDDA2 pipeline for dcid={dcid}, only accepting data collections from lb42888 during test phase"
+                f"Not triggering PanDDA2 pipeline for dcid={dcid} proposal {proposal_string}"
             )
             return {"success": True}