Merge remote-tracking branch 'origin/main' into pr/7

README.md

@@ -51,7 +51,7 @@ The goal is to improve developer experience by making the review process as lean
 
 ```py
 # requirements.txt
-gradio-space-ci @ git+https://huggingface.co/spaces/Wauplin/gradio-space-ci@0.2.2
+gradio-space-ci @ git+https://huggingface.co/spaces/Wauplin/gradio-space-ci@0.2.3
 ```
 
 2. Set `HF_TOKEN` as a Space secret.
@@ -110,7 +110,7 @@ Add the following line to it:
 
 ```bash
 # requirements.txt
-gradio-space-ci @ git+https://huggingface.co/spaces/Wauplin/gradio-space-ci@0.2.2
+gradio-space-ci @ git+https://huggingface.co/spaces/Wauplin/gradio-space-ci@0.2.3
 ```
 
 ### 2. Add a user token as `HF_TOKEN` secret

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.

open_pr.py

@@ -61,7 +61,7 @@ def open_pr(space_id_or_url: str, oauth_token: gr.OAuthToken | None) -> str:
     else:
         requirements = ""
     if "gradio-space-ci" not in requirements:
-        requirements += "\ngradio-space-ci @ git+https://huggingface.co/spaces/Wauplin/gradio-space-ci@0.2.2\n"
+        requirements += "\ngradio-space-ci @ git+https://huggingface.co/spaces/Wauplin/gradio-space-ci@0.2.3\n"
 
     # 2. Configure CI in README.md
     card = SpaceCard.load(api.hf_hub_download(repo_id=space_id, repo_type="space", filename="README.md"))

src/gradio_space_ci/__init__.py

@@ -38,4 +38,4 @@ else:
     from .webhook import enable_space_ci  # noqa: F401
 
 
-__version__ = "0.2.2"
+__version__ = "0.2.3"

src/gradio_space_ci/webhook.py

@@ -57,6 +57,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.
@@ -201,7 +204,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.")
@@ -254,7 +257,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
     ):
         # A comment, is it by a command ?
         if payload.event.scope == "discussion.comment":
@@ -285,9 +288,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 untrusted author made a modification we unset the config
-                handle_modification(space_id=space_id, discussion=discussion)
+            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)