Implement label id retrieval

2021-05-05 18:00:20 +01:00 · 2021-05-05 18:00:20 +01:00 · 2471d3e23f
commit 2471d3e23f
parent 165fd70acc
1 changed files with 8 additions and 0 deletions
--- a/src/borgmanager/database/connection/labelconn.py
+++ b/src/borgmanager/database/connection/labelconn.py
@ -68,4 +68,12 @@ class LabelConn(DatabaseConnection):
        else:
            return result[0]

+    def get_label_id(self, repo_id):
+        result = self.sql_execute_one(f"SELECT id FROM {self._sql_table} WHERE repo_id = ?",
+                                      (repo_id,))
+        if result is None:
+            return None
+        else:
+            return result[0]
+
    # endregion