diff --git a/git.py b/git.py index bf30e83..8402c8c 100644 --- a/git.py +++ b/git.py @@ -223,6 +223,17 @@ class Repo: refs = self._for_each_ref(pattern="refs/heads/", sort="-authordate") return [ref[len("refs/heads/") :] for _, _, ref in refs] + @functools.cache + def main_branch(self): + """Get the name of the main branch.""" + bs = self.branch_names() + for branch in ["master", "main"]: + if branch in bs: + return branch + if bs: + return bs[0] + return None + @functools.cache def tags(self, sort="-taggerdate"): """Get the (name, obj_id) of the tags.""" diff --git a/views/summary.html b/views/summary.html index d3388e6..9b180b8 100644 --- a/views/summary.html +++ b/views/summary.html @@ -35,20 +35,22 @@