scheduler: resolve git references into revisions on submission

Signed-off-by: Charles Baynham <[email protected]>

Author: charlesbaynham

RELEASE_NOTES.rst

@@ -45,6 +45,7 @@ Highlights:
 * Full Python 3.10 support.
 * MSYS2 packaging for Windows, which replaces Conda. Conda packages are still available to
   support legacy installations, but may be removed in a future release.
+* Experiments can now be submitted with revisions set to a branch / tag name instead of only git hashes.
 
 Breaking changes:
 

artiq/master/experiments.py

@@ -111,7 +111,7 @@ async def scan_repository(self, new_cur_rev=None):
         try:
             if new_cur_rev is None:
                 new_cur_rev = self.repo_backend.get_head_rev()
-            wd, _ = self.repo_backend.request_rev(new_cur_rev)
+            wd, _, _ = self.repo_backend.request_rev(new_cur_rev)
             self.repo_backend.release_rev(self.cur_rev)
             self.cur_rev = new_cur_rev
             self.status["cur_rev"] = new_cur_rev
@@ -132,7 +132,7 @@ async def examine(self, filename, use_repository=True, revision=None):
         if use_repository:
             if revision is None:
                 revision = self.cur_rev
-            wd, _ = self.repo_backend.request_rev(revision)
+            wd, _, revision = self.repo_backend.request_rev(revision)
             filename = os.path.join(wd, filename)
         worker = Worker(self.worker_handlers)
         try:
@@ -169,7 +169,7 @@ def get_head_rev(self):
         return "N/A"
 
     def request_rev(self, rev):
-        return self.root, None
+        return self.root, None, "N/A"
 
     def release_rev(self, rev):
         pass
@@ -198,16 +198,22 @@ def __init__(self, root):
         self.checkouts = dict()
 
     def get_head_rev(self):
-        return str(self.git.head.target)
+        return str(self.git.head.target)    
+
+    def _get_pinned_rev(self, rev):
+        'Resolve a git pointer (e.g. "HEAD", "master", "abcdef123456...") into a git hash'
+        commit, _ = self.git.resolve_refish(rev)
+        return commit.hex
 
     def request_rev(self, rev):
+        rev = self._get_pinned_rev(rev)
         if rev in self.checkouts:
             co = self.checkouts[rev]
             co.ref_count += 1
         else:
             co = _GitCheckout(self.git, rev)
             self.checkouts[rev] = co
-        return co.path, co.message
+        return co.path, co.message, rev
 
     def release_rev(self, rev):
         co = self.checkouts[rev]

artiq/master/scheduler.py

@@ -139,8 +139,12 @@ def submit(self, expid, priority, due_date, flush, pipeline_name):
         if "repo_rev" in expid:
             if expid["repo_rev"] is None:
                 expid["repo_rev"] = self.experiment_db.cur_rev
-            wd, repo_msg = self.experiment_db.repo_backend.request_rev(
-                expid["repo_rev"])
+            wd, repo_msg, pinned_rev = self.experiment_db.repo_backend.request_rev(
+                expid["repo_rev"]
+            )
+            # Mutate expid's repo_rev to that returned from request_rev, in case
+            # a branch was passed instead of a hash
+            expid["repo_rev"] = pinned_rev
         else:
             if "file" in expid:
                 expid["file"] = os.path.abspath(expid["file"])