Deploy on draft PRs

RELEASE.md

@@ -1,3 +1,7 @@
+## 0.2.3
+
+- Deploy on "draft" PRs as well, not just "open" PRs.
+
 ## 0.2.2
 
 - Install `huggingface_hub>=0.21.1` now that 0.21 have been released.

src/gradio_space_ci/webhook.py

@@ -54,6 +54,9 @@ if SPACE_ID is not None:  # If running in a Space (i.e. not locally)
 
 EPHEMERAL_SPACES_CONFIG: Dict[str, Any] = {}
 
+# Draft and open PRs are considered as active (in opposition to closed and merged PRs)
+ACTIVE_PR_STATUS = ("draft", "open")
+
 
 def enable_space_ci() -> None:
     """Enable Space CI for the current Space based on config from the README.md file.
@@ -198,7 +201,7 @@ background_pool = ThreadPoolExecutor(max_workers=1)
 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 == "open":
+        if discussion.status in ACTIVE_PR_STATUS:
             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.")
@@ -251,7 +254,7 @@ async def trigger_ci_on_pr(payload: WebhookPayload, task_queue: BackgroundTasks)
         and payload.event.action == "create"
         and payload.discussion is not None
         and payload.discussion.isPullRequest
-        and payload.discussion.status == "open"
+        and payload.discussion.status in ACTIVE_PR_STATUS
     ):
         if not is_pr_synced(space_id=space_id, pr_num=payload.discussion.num):
             # New PR! Sync task scheduled
@@ -278,7 +281,7 @@ async def trigger_ci_on_pr(payload: WebhookPayload, task_queue: BackgroundTasks)
         # New repo change. Is it a commit on a PR?
         # => loop through all PRs and check if new changes happened
         for discussion in get_repo_discussions(repo_id=space_id, repo_type="space"):
-            if discussion.is_pull_request and discussion.status == "open":
+            if discussion.is_pull_request and discussion.status in ACTIVE_PR_STATUS:
                 if not is_pr_synced(space_id=space_id, pr_num=discussion.num):
                     # Found a PR that is not yet synced
                     task_queue.add_task(sync_ci_space, space_id=space_id, pr_num=discussion.num)