advanced the review workflow

src/repo_registry/storage/sqlite.py

@@ -383,6 +383,64 @@ class RegistryStore:
                     (status, repository_id, analysis_run_id),
                 )
 
+    def reject_candidate_ability(
+        self,
+        repository_id: int,
+        analysis_run_id: int,
+        candidate_ability_id: int,
+    ) -> None:
+        with self.connect() as connection:
+            ability_cursor = connection.execute(
+                """
+                UPDATE candidate_abilities
+                SET status = 'rejected'
+                WHERE id = ?
+                  AND repository_id = ?
+                  AND analysis_run_id = ?
+                  AND status = 'candidate'
+                """,
+                (candidate_ability_id, repository_id, analysis_run_id),
+            )
+            if ability_cursor.rowcount == 0:
+                raise NotFoundError(
+                    "candidate ability "
+                    f"{candidate_ability_id} was not found for repository "
+                    f"{repository_id} analysis run {analysis_run_id}"
+                )
+            capability_rows = connection.execute(
+                """
+                SELECT id FROM candidate_capabilities
+                WHERE ability_id = ? AND repository_id = ? AND analysis_run_id = ?
+                """,
+                (candidate_ability_id, repository_id, analysis_run_id),
+            ).fetchall()
+            capability_ids = [row["id"] for row in capability_rows]
+            connection.execute(
+                """
+                UPDATE candidate_capabilities
+                SET status = 'rejected'
+                WHERE ability_id = ? AND repository_id = ? AND analysis_run_id = ?
+                """,
+                (candidate_ability_id, repository_id, analysis_run_id),
+            )
+            for capability_id in capability_ids:
+                connection.execute(
+                    """
+                    UPDATE candidate_features
+                    SET status = 'rejected'
+                    WHERE capability_id = ? AND repository_id = ? AND analysis_run_id = ?
+                    """,
+                    (capability_id, repository_id, analysis_run_id),
+                )
+                connection.execute(
+                    """
+                    UPDATE candidate_evidence
+                    SET status = 'rejected'
+                    WHERE capability_id = ? AND repository_id = ? AND analysis_run_id = ?
+                    """,
+                    (capability_id, repository_id, analysis_run_id),
+                )
+
     def create_review_decision(
         self,
         repository_id: int,