tarekziade HF Staff
Publish unified PR diff prototype with complete review threads and filters
bd0f095 verified | """Standalone server for the PR diff prototype and its public GitHub snapshot.""" | |
| from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer | |
| from pathlib import Path | |
| from urllib.parse import parse_qs, urlparse | |
| import os | |
| ROOT = Path(__file__).resolve().parent | |
| ASSETS = { | |
| "/": ("patch_view.html", "text/html; charset=utf-8", None), | |
| "/patch-view": ("patch_view.html", "text/html; charset=utf-8", None), | |
| "/patch-view/script.js": ("patch_view.js", "text/javascript; charset=utf-8", None), | |
| "/patch-view/data": ("data/pr.diff", "text/plain; charset=utf-8", None), | |
| "/patch-view/comments": ( | |
| "data/comments.json.gz", | |
| "application/json; charset=utf-8", | |
| "gzip", | |
| ), | |
| } | |
| class Handler(BaseHTTPRequestHandler): | |
| def do_GET(self): # noqa: N802 | |
| parsed = urlparse(self.path) | |
| if parsed.path == "/health": | |
| self.reply(200, "text/plain", b"ok") | |
| return | |
| asset = ASSETS.get(parsed.path) | |
| if asset is None: | |
| self.reply(404, "text/plain", b"Not found") | |
| return | |
| if parsed.path in {"/patch-view/data", "/patch-view/comments"}: | |
| params = parse_qs(parsed.query) | |
| if ( | |
| params.get("pr", ["43451"])[0] != "43451" | |
| or params.get("repository", ["huggingface/transformers"])[0] | |
| != "huggingface/transformers" | |
| ): | |
| self.reply( | |
| 400, | |
| "text/plain", | |
| b"This prototype contains the saved huggingface/transformers PR #43451 snapshot.", | |
| ) | |
| return | |
| name, content_type, encoding = asset | |
| self.reply(200, content_type, (ROOT / name).read_bytes(), encoding) | |
| def reply(self, status, content_type, body, encoding=None): | |
| self.send_response(status) | |
| self.send_header("Content-Type", content_type) | |
| self.send_header("Content-Length", str(len(body))) | |
| self.send_header("Cache-Control", "no-cache") | |
| if encoding: | |
| self.send_header("Content-Encoding", encoding) | |
| self.end_headers() | |
| self.wfile.write(body) | |
| if __name__ == "__main__": | |
| port = int(os.environ.get("PORT", "7860")) | |
| print(f"Serving prototype on :{port}", flush=True) | |
| ThreadingHTTPServer(("0.0.0.0", port), Handler).serve_forever() | |