sources: Update query to fetch a commit by ID

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/buildservice/sources.py b/src/buildservice/sources.py
index cac2a7a8a1576162714c68c9519af4b829e69f3d..98b9805ced92ca5c934fc3e2ed69b89f32aa62a6 100644 (file)
--- a/src/buildservice/sources.py
+++ b/src/buildservice/sources.py
@@ -106,23 +106,28 @@ class Sources(base.Object):
 
        # Commits
 
-       def _get_commit(self, query, *args):
-               res = self.db.get(query, *args)
-
-               if res:
-                       return Commit(self.backend, res.id, data=res)
-
        def _get_commits(self, query, *args):
                res = self.db.query(query, *args)
 
                for row in res:
                        yield Commit(self.backend, row.id, data=row)
 
-       def get_commit_by_id(self, commit_id):
-               commit = self.db.get("SELECT id FROM sources_commits WHERE id = %s", commit_id)
+       def _get_commit(self, query, *args):
+               res = self.db.get(query, *args)
 
-               if commit:
-                       return Commit(self.pakfire, commit.id)
+               if res:
+                       return Commit(self.backend, res.id, data=res)
+
+       def get_commit_by_id(self, id):
+               return self._get_commit("""
+                       SELECT
+                               *
+                       FROM
+                               source_commits
+                       WHERE
+                               id = %s
+                       """, id,
+               )
 
 
 class Source(base.DataObject):