Implement retrieval of repo name

This commit is contained in:
George Lacey 2021-05-05 05:50:53 +01:00
parent 28c51e5154
commit e512fb1e23
2 changed files with 15 additions and 0 deletions

View File

@ -47,6 +47,9 @@ class BorgDatabase(object):
def get_repos(self): def get_repos(self):
return self.repo_conn.get_all() return self.repo_conn.get_all()
def get_repo_name(self, repo):
return self.label_conn.get_repo_name(repo.primary_key)
def get_cache(self, repo): def get_cache(self, repo):
archive = Archive.from_sql(self.get_latest_archive(repo)) archive = Archive.from_sql(self.get_latest_archive(repo))
return self.cache_conn.get(archive.primary_key) return self.cache_conn.get(archive.primary_key)

View File

@ -56,3 +56,15 @@ class LabelConn(DatabaseConnection):
self.sql_commit() self.sql_commit()
# endregion # endregion
# region QUERIES
def get_repo_name(self, repo_id):
result = self.sql_execute_one(f"SELECT label FROM {self._sql_table} WHERE repo_id = ?",
(repo_id,))
if result is None:
return None
else:
return result[0]
# endregion