Фаза 2: site/app.py → blueprint'ы (main, health, api), старый drhider.py удалён
Deploy drhider / validate (push) Waiting to run
Deploy drhider / validate (push) Waiting to run
This commit is contained in:
@@ -0,0 +1,20 @@
|
||||
"""
|
||||
Регистрация всех blueprint'ов приложения.
|
||||
|
||||
Импортируется из site/app.py при создании Flask-приложения.
|
||||
"""
|
||||
|
||||
from .main_bp import main_bp
|
||||
from .health_bp import health_bp
|
||||
from .api_bp import api_bp
|
||||
|
||||
|
||||
def register_routes(app):
|
||||
"""Зарегистрировать все blueprint'ы на Flask-приложении.
|
||||
|
||||
Args:
|
||||
app: Экземпляр Flask-приложения
|
||||
"""
|
||||
app.register_blueprint(main_bp)
|
||||
app.register_blueprint(health_bp)
|
||||
app.register_blueprint(api_bp)
|
||||
@@ -0,0 +1,73 @@
|
||||
"""
|
||||
Blueprint: API обфускации (POST /api/drhider).
|
||||
|
||||
Принимает multipart/form-data с файлами, возвращает ZIP-архив
|
||||
с обфусцированными документами.
|
||||
"""
|
||||
|
||||
import io
|
||||
import traceback
|
||||
from flask import Blueprint, request, send_file, jsonify
|
||||
|
||||
from drhider import obfuscate_files, LLMClient
|
||||
|
||||
# ═══════════════════════════════════════════════════════════════════════════
|
||||
# Blueprint: API DrHider
|
||||
# ═══════════════════════════════════════════════════════════════════════════
|
||||
|
||||
api_bp = Blueprint("api", __name__, url_prefix="/api")
|
||||
|
||||
# Максимальный размер тела запроса: 200 MB
|
||||
MAX_BODY = 200 * 1024 * 1024
|
||||
|
||||
|
||||
@api_bp.route("/drhider", methods=["POST"])
|
||||
def drhider():
|
||||
"""Обфускация документов.
|
||||
|
||||
Принимает multipart/form-data с полем 'files' (один или несколько файлов).
|
||||
Возвращает ZIP-архив с обфусцированными документами + mapping.csv.
|
||||
|
||||
Поддерживаемые форматы:
|
||||
.docx, .pdf, .txt, .doc (бинарный — без изменений)
|
||||
.zip (распаковывается, содержимое обфусцируется)
|
||||
|
||||
Returns:
|
||||
200: ZIP-архив (application/zip)
|
||||
400: {"ok": false, "error": "..."}
|
||||
500: {"ok": false, "error": "..."}
|
||||
"""
|
||||
# ── Получаем файлы из запроса ──
|
||||
uploaded = request.files.getlist("files")
|
||||
if not uploaded:
|
||||
return jsonify({"ok": False, "error": "Нет файлов"}), 400
|
||||
|
||||
# Формируем список для obfuscate_files: [(name, bytes, mimetype), ...]
|
||||
files = []
|
||||
for f in uploaded:
|
||||
if f.filename:
|
||||
files.append((f.filename, f.read(), f.mimetype or ""))
|
||||
|
||||
if not files:
|
||||
return jsonify({"ok": False, "error": "Нет файлов"}), 400
|
||||
|
||||
# ── Обфускация ──
|
||||
try:
|
||||
# Создаём LLM-клиент для NER-сканирования
|
||||
llm = LLMClient()
|
||||
|
||||
# Вызываем ядро обфускации
|
||||
zip_data, _csv = obfuscate_files(files, llm_client=llm)
|
||||
|
||||
# Отдаём ZIP-архив
|
||||
return send_file(
|
||||
io.BytesIO(zip_data),
|
||||
mimetype="application/zip",
|
||||
as_attachment=True,
|
||||
download_name="drhider_output.zip",
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
# Логируем полный трейсбек на сервере
|
||||
traceback.print_exc()
|
||||
return jsonify({"ok": False, "error": str(e)}), 500
|
||||
@@ -0,0 +1,27 @@
|
||||
"""
|
||||
Blueprint: health-check (GET /health).
|
||||
|
||||
Используется платформой Штурвал для проверки живости приложения.
|
||||
"""
|
||||
|
||||
from flask import Blueprint, jsonify, current_app
|
||||
|
||||
# ═══════════════════════════════════════════════════════════════════════════
|
||||
# Blueprint: health check
|
||||
# ═══════════════════════════════════════════════════════════════════════════
|
||||
|
||||
health_bp = Blueprint("health", __name__)
|
||||
|
||||
|
||||
@health_bp.route("/health")
|
||||
def health():
|
||||
"""Эндпоинт проверки живости.
|
||||
|
||||
Возвращает JSON с версией приложения.
|
||||
Используется платформой для readiness/liveness probes.
|
||||
|
||||
Returns:
|
||||
{"ok": true, "version": "X.Y.Z"}
|
||||
"""
|
||||
version = current_app.config.get("VERSION", "0.0.0")
|
||||
return jsonify({"ok": True, "version": version})
|
||||
@@ -0,0 +1,23 @@
|
||||
"""
|
||||
Blueprint: главная страница (GET /).
|
||||
|
||||
Отдаёт HTML-интерфейс DrHider.
|
||||
"""
|
||||
|
||||
from flask import Blueprint, render_template, current_app
|
||||
|
||||
# ═══════════════════════════════════════════════════════════════════════════
|
||||
# Blueprint: главная страница
|
||||
# ═══════════════════════════════════════════════════════════════════════════
|
||||
|
||||
main_bp = Blueprint("main", __name__)
|
||||
|
||||
|
||||
@main_bp.route("/")
|
||||
def index():
|
||||
"""Главная страница — веб-интерфейс DrHider.
|
||||
|
||||
Передаёт в шаблон текущую версию приложения.
|
||||
"""
|
||||
version = current_app.config.get("VERSION", "0.0.0")
|
||||
return render_template("index.html", version=version)
|
||||
Reference in New Issue
Block a user