_ostree.py: Add set_ref()

1 files changed, 22 insertions, 0 deletions

diff --git a/buildstream/_ostree.py b/buildstream/_ostree.py
index c165d5224..3bcc0e828 100644
--- a/buildstream/_ostree.py
+++ b/buildstream/_ostree.py
@@ -158,6 +158,28 @@ def commit(repo, dir, ref, branch=None):
         raise
 
 
+# set_ref():
+#
+# Set symbolic reference to specified revision.
+#
+# Args:
+#    repo (OSTree.Repo): The repo
+#    ref (str): A symbolic reference (tag) for the commit
+#    rev (str): Commit checksum
+#
+def set_ref(repo, ref, rev):
+
+    repo.prepare_transaction()
+    try:
+        repo.transaction_set_ref(None, ref, rev)
+
+        # complete repo transaction
+        repo.commit_transaction(None)
+    except:
+        repo.abort_transaction()
+        raise
+
+
 # exists():
 #
 # Checks wether a given commit or symbolic ref exists and