better

src/gradio_space_ci/webhook.py

@@ -205,6 +205,8 @@ def recover_after_restart(space_id: str) -> None:
     print("Looping through PRs to check if any needs to be synced.")
     for discussion in get_repo_discussions(repo_id=space_id, repo_type="space", discussion_type="pull_request"):
         if discussion.status in ACTIVE_PR_STATUS:
+            # check pr trust status
+            handle_modification(space_id=space_id, discussion=discussion)
             if not is_pr_synced(space_id=space_id, pr_num=discussion.num):
                 # Found a PR that is not yet synced
                 print(f"Recovery. Found an open PR that is not synced: {discussion.url}. Syncing it.")
@@ -438,7 +440,9 @@ def delete_ci_space(space_id: str, pr_num: int) -> None:
 def notify_pr(
     space_id: str,
     pr_num: int,
-    action: Literal["created_not_configured", "created_and_configured", "updated", "deleted","trusted_pr","untrusted_pr"],
+    action: Literal[
+        "created_not_configured", "created_and_configured", "updated", "deleted", "trusted_pr", "untrusted_pr"
+    ],
 ) -> None:
     ci_space_id = _get_ci_space_id(space_id=space_id, pr_num=pr_num)
     if action == "created_not_configured":
@@ -449,7 +453,7 @@ def notify_pr(
         comment = NOTIFICATION_TEMPLATE_UPDATED.format(ci_space_id=ci_space_id)
     elif action == "deleted":
         comment = NOTIFICATION_TEMPLATE_DELETED
-    elif action == "trusted_pr": 
+    elif action == "trusted_pr":
         comment = NOTIFICATION_TEMPLATE_TRUSTED_PR
     elif action == "untrusted_pr":
         comment = NOTIFICATION_TEMPLATE_UNTRUSTED_PR
@@ -517,11 +521,10 @@ def handle_command(space_id: str, payload: WebhookPayload) -> None:
     if event_author in EPHEMERAL_SPACES_CONFIG["trusted_authors"]:
         if payload.comment.content == "/trust_pr":
             set_config(space_id=space_id, pr_num=pr_num)
-            notify_pr(space_id=space_id,pr_num=pr_num,action="trusted_pr")
+            notify_pr(space_id=space_id, pr_num=pr_num, action="trusted_pr")
         elif payload.comment.content == "/untrust_pr":
             unset_config(space_id=space_id, pr_num=pr_num)
-            notify_pr(space_id=space_id,pr_num=pr_num,action="untrusted_pr")
-            
+            notify_pr(space_id=space_id, pr_num=pr_num, action="untrusted_pr")
 
 
 NOTIFICATION_TEMPLATE_CREATED_AND_CONFIGURED = """\