From 885d962c0622a86cb217d0a0326b181a6a0f3b55 Mon Sep 17 00:00:00 2001 From: huni Date: Fri, 24 Jul 2026 07:55:48 +0800 Subject: [PATCH] add main.py --- main.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 main.py diff --git a/main.py b/main.py new file mode 100644 index 0000000..c42eaff --- /dev/null +++ b/main.py @@ -0,0 +1,27 @@ +import json, os, signal +from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer +class H(BaseHTTPRequestHandler): + def _j(self,o,s=200): + p=json.dumps(o).encode(); self.send_response(s) + self.send_header("Content-Type","application/json"); self.send_header("Content-Length",str(len(p))); self.end_headers(); self.wfile.write(p) + def do_GET(self): + if self.path=="/health": self._j({"status":"ok"}) + elif self.path=="/": self._j({"name":"light-adapt-test","status":"running"}) + else: self._j({"error":"not found"},404) + def do_POST(self): + if self.path=="/task": + l=int(self.headers.get("Content-Length",0)); b=self.rfile.read(l) if l else b"{}" + try: t=json.loads(b) + except: t={} + print("task",t.get("model_address",""),flush=True) + self._j({"status":"accepted","model":t.get("model_address","")}) + else: self._j({"error":"not found"},404) + def log_message(self,*a): pass +run=True +def stop(*a): + global run; run=False +signal.signal(signal.SIGTERM,stop); signal.signal(signal.SIGINT,stop) +srv=ThreadingHTTPServer(("0.0.0.0",8080),H); srv.timeout=1 +print("listening 8080",flush=True) +while run: srv.handle_request() +srv.server_close()