api.views: update pr description on landing trigger (bug 2031664)

Author: zzzeid

src/lando/api/views.py

@@ -90,6 +90,38 @@ def generate_warnings_and_blockers(
     return {"warnings": warnings, "blockers": blockers}
 
 
+def generate_enhanced_pr_description(
+    pull_request: PullRequest,
+    target_repo: Repo,
+    request: WSGIRequest,
+    template: str = "pr_description.md",
+) -> str:
+    context = {}
+    context.update(generate_warnings_and_blockers(target_repo, pull_request, request))
+
+    path = reverse(
+        "pull-request",
+        kwargs={
+            "repo_name": target_repo.name,
+            "number": pull_request.number,
+        },
+    )
+
+    context["lando_url"] = f"{settings.SITE_URL}{path}"
+    context["special_delimiter"] = SPECIAL_DELIMITER
+    bugs = parse_bugs(pull_request.title)
+    context["bugs"] = []
+
+    for bug in bugs:
+        context["bugs"].append((bug, f"{settings.BUGZILLA_URL}/{bug}"))
+
+    context["title"] = pull_request.title
+    context["body"] = pull_request.parsed_body
+
+    rendered = render_to_string(template, context)
+    return rendered
+
+
 @method_decorator(csrf_exempt, name="dispatch")
 class LegacyDiffWarningView(View):
     """
@@ -309,6 +341,14 @@ class Form(forms.Form):
         job.status = JobStatus.SUBMITTED
         job.save()
 
+        description = generate_enhanced_pr_description(
+            self.pull_request,
+            self.target_repo,
+            request,
+            template="pr_description_landing.md",
+        )
+        self.client.update_pull_request_body(pull_number, description)
+
         return JsonResponse({"id": job.id}, status=201)
 
 
@@ -385,30 +425,10 @@ def post(
         if not request.user.has_perm("main.can_change_landing_job"):
             raise PermissionError()
 
-        context = {}
-        context.update(
-            generate_warnings_and_blockers(self.target_repo, self.pull_request, request)
+        description = generate_enhanced_pr_description(
+            self.pull_request, self.target_repo, request
         )
-
-        path = reverse(
-            "pull-request",
-            kwargs={
-                "repo_name": self.target_repo.name,
-                "number": self.pull_request.number,
-            },
+        self.client.update_pull_request_body(pull_number, description)
+        return JsonResponse(
+            {"status": f"Pull request {pull_number} description updated."}
         )
-
-        context["lando_url"] = f"{settings.SITE_URL}{path}"
-        context["special_delimiter"] = SPECIAL_DELIMITER
-        bugs = parse_bugs(self.pull_request.title)
-        context["bugs"] = []
-
-        for bug in bugs:
-            context["bugs"].append((bug, f"{settings.BUGZILLA_URL}/{bug}"))
-
-        context["title"] = self.pull_request.title
-        context["body"] = self.pull_request.parsed_body
-
-        rendered = render_to_string("pr_description.md", context)
-        self.client.update_pull_request_body(pull_number, rendered)
-        return JsonResponse({"context": context, "md": rendered})

src/lando/ui/jinja2/pr_description.md

@@ -1,7 +1,7 @@
-{% block upper %}
 Lando: [link]({{ lando_url }})
 {% if bugs %}Bugzilla: {% for bug in bugs %}[bug {{ bug.0 }}]({{ bug.1 }}){% endfor %}{% endif %}
 
+{% block upper %}
 {% if warnings %}
 |Warnings|
 |---------|

src/lando/ui/jinja2/pr_description_landing.md

@@ -0,0 +1,4 @@
+{% extends "pr_description.md" %}
+{% block upper %}
+**Landing was requested.**
+{% endblock %}