Refactor VersionedAttributes, ProcessSteps, and Conditions retrieval.

[plomtask] / plomtask / db.py

diff --git a/plomtask/db.py b/plomtask/db.py
index bf633e2c9826f464c690b03b75dfda4a33df7234..f45d2b6adac0e109938d9f8018653cf1caa73fcf 100644 (file)
--- a/plomtask/db.py
+++ b/plomtask/db.py
@@ -76,6 +76,11 @@ class DatabaseConnection:
             q_marks = self.__class__.q_marks_from_values(values)
             self.exec(f'INSERT INTO {table_name} VALUES {q_marks}', values)
 
+    def all_where(self, table_name: str, key: str, target: int) -> list[Row]:
+        """Return list of Rows at table where key == target."""
+        return list(self.exec(f'SELECT * FROM {table_name} WHERE {key} = ?',
+                              (target,)))
+
     @staticmethod
     def q_marks_from_values(values: tuple[Any]) -> str:
         """Return placeholder to insert values into SQL code."""