From c94d40d54d196f58b71805e4f4664d43bf9dee72 Mon Sep 17 00:00:00 2001 From: matt-codecov <137832199+matt-codecov@users.noreply.github.com> Date: Mon, 7 Apr 2025 15:14:09 -0700 Subject: [PATCH] badges: fetch branch head and associated commit in same query --- graphs/views.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/graphs/views.py b/graphs/views.py index 68f0779aeb..73267a8375 100644 --- a/graphs/views.py +++ b/graphs/views.py @@ -112,17 +112,11 @@ def get_coverage(self): return None, coverage_range branch_name = self.kwargs.get("branch") or repo.branch - branch = Branch.objects.filter( + branch_qs = Branch.objects.filter( name=branch_name, repository_id=repo.repoid - ).first() - - if branch is None: - log.warning( - "Branch not found", extra=dict(branch_name=branch_name, repo=repo) - ) - return None, coverage_range + ) try: - commit = repo.commits.filter(commitid=branch.head).first() + commit = repo.commits.filter(commitid__in=branch_qs.values_list("head", flat=True)).first() except ObjectDoesNotExist: # if commit does not exist return None coverage log.warning("Commit not found", extra=dict(commit=branch.head))