feat: async DrHider — POST job + poll + download, v0.5/v1.0.188

site/app.py

@@ -3,11 +3,17 @@
 Все тяжёлые операции — на ВМ (contracts.kube5s.ru).
 """
 import os
+import uuid
+import threading
 import httpx
 from flask import Flask, render_template, request, jsonify
 
 app = Flask(__name__)
 
+# DrHider job store (in-memory, no persistence)
+_drhider_jobs = {}
+_drhider_lock = threading.Lock()
+
 VM_API = os.environ.get("VM_API", "https://contracts.kube5s.ru")
 LLM_URL = os.environ.get("LLM_API_URL", "https://api.aillm.ru/v1/chat/completions")
 LLM_KEY = os.environ.get("LLM_API_KEY", "")
@@ -110,9 +116,7 @@ def drhider():
 
 @app.route("/api/drhider", methods=["POST"])
 def api_drhider():
-    """Обфускация: файлы → LLM-NER → ZIP."""
-    import io
-    from flask import send_file
+    """Обфускация: старт фоновой задачи."""
     from services.drhider import obfuscate_files
 
     uploaded = request.files.getlist("files")
@@ -123,13 +127,50 @@ def api_drhider():
     if not files:
         return jsonify({"ok": False, "error": "Нет файлов"}), 400
 
-    try:
-        zip_data, _csv = obfuscate_files(files)
-        buf = io.BytesIO(zip_data)
-        buf.seek(0)
-        return send_file(buf, mimetype="application/zip", as_attachment=True, download_name="drhider_output.zip")
-    except Exception as e:
-        return jsonify({"ok": False, "error": str(e)}), 500
+    job_id = uuid.uuid4().hex
+    with _drhider_lock:
+        _drhider_jobs[job_id] = {"done": False, "data": None, "error": None}
+
+    def _process():
+        try:
+            zip_data, _csv = obfuscate_files(files)
+            with _drhider_lock:
+                _drhider_jobs[job_id] = {"done": True, "data": zip_data, "error": None}
+        except Exception as e:
+            with _drhider_lock:
+                _drhider_jobs[job_id] = {"done": True, "data": None, "error": str(e)}
+
+    threading.Thread(target=_process, daemon=True).start()
+    return jsonify({"ok": True, "job_id": job_id})
+
+
+@app.route("/api/drhider/<job_id>/status")
+def api_drhider_status(job_id):
+    """Статус задачи."""
+    with _drhider_lock:
+        job = _drhider_jobs.get(job_id)
+    if not job:
+        return jsonify({"ok": False, "error": "Неизвестный job_id"}), 404
+    return jsonify({"ok": True, "done": job["done"], "error": job["error"]})
+
+
+@app.route("/api/drhider/<job_id>/download")
+def api_drhider_download(job_id):
+    """Скачать результат."""
+    from flask import send_file
+    import io
+    with _drhider_lock:
+        job = _drhider_jobs.get(job_id)
+    if not job or not job["done"]:
+        return jsonify({"ok": False, "error": "Не готов"}), 404
+    if job["error"]:
+        return jsonify({"ok": False, "error": job["error"]}), 500
+    buf = io.BytesIO(job["data"])
+    buf.seek(0)
+    # Очистка
+    with _drhider_lock:
+        _drhider_jobs.pop(job_id, None)
+    return send_file(buf, mimetype="application/zip", as_attachment=True, download_name="drhider_output.zip")
 
 
 @app.route("/health")