From 2471d3e23f67cefab3d1c2cf2baf548d84c1264d Mon Sep 17 00:00:00 2001 From: George Lacey Date: Wed, 5 May 2021 18:00:20 +0100 Subject: [PATCH] Implement label id retrieval --- src/borgmanager/database/connection/labelconn.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/borgmanager/database/connection/labelconn.py b/src/borgmanager/database/connection/labelconn.py index bfd0ee6..9da1ef5 100644 --- 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