# План миграции: ВМ → Flask (полный перенос, без ВМ)

**Дата:** 2026-07-14
**Цель:** Убрать `deploy/convert_server.py` и весь ВМ-слой. Вся логика — во Flask.
**Принцип:** НИКАКОГО монолита. Каждый слой — отдельный модуль/blueprint.

---

## 0. Аудит: насколько код уже decoupled

### ✅ УЖЕ ГОТОВО (можно брать как есть)

| Модуль | Строк | Статус | Почему |
|--------|-------|--------|--------|
| `db/connection.py` | ~60 | ✅ Идеально | Connection pool, ноль зависимостей |
| `db/documents.py` | ~100 | ✅ Идеально | Чистый CRUD, только `query()/execute()` |
| `db/contracts.py` | ~25 | ✅ Идеально | Чистый CRUD |
| `db/supplements.py` | ~60 | ✅ Идеально | Чистый CRUD |
| `db/spec_current.py` | ~20 | ✅ Идеально | Чистые запросы |
| `db/spec_events.py` | ~30 | ✅ Идеально | Чистый CRUD |
| `db/prompts.py` | ~80 | ✅ Идеально | Чистый CRUD + seed |
| `compare/parse.py` | ~100 | ✅ Идеально | Чистая функция: `(filename, bytes) → dict` |
| `compare/llm_client.py` | ~80 | ✅ Идеально | Protocol + Httpx + Fake, DI-ready |
| `compare/grouping.py` | ~160 | ✅ Идеально | Чистая логика: `batch_id → groups`, нет HTTP |
| `compare/llm.py` | ~40 | ✅ Хорошо | Уже принимает `llm_client` опционально (DI) |
| `llm_prompt.py` | ~80 | ✅ Идеально | Чистые функции сборки промптов |
| `repository.py` | ~120 | ✅ Хорошо | Protocol есть, PgRepository частично реализован |

### 🔧 НУЖНА АДАПТАЦИЯ (логика готова, интерфейс — нет)

| Модуль | Проблема | Что сделать |
|--------|----------|-------------|
| `compare/upload.py` | Использует `cgi.FieldStorage` | Заменить на `request.files` из Flask |
| `compare/unzip.py` | Читает `rfile.read()` сырые байты | Заменить на `request.files` / `request.data` |
| `compare/classify.py` | `ThreadPoolExecutor` ок, но запускается из HTTP-метода | Обернуть в Flask background task (см. ниже) |
| `compare/process.py` | SSE через `self.wfile.write()` | Заменить на `Response(stream_with_context(...))` |

### ❌ НУЖНО ПЕРЕПИСАТЬ

| Модуль | Проблема | Что сделать |
|--------|----------|-------------|
| `convert_server.py` | Монолит ~500 строк, все роуты в одном классе | Разобрать на Flask blueprints |
| `classify_worker.py` | subprocess.Popen — не нужно во Flask | Убрать, classify в отдельном потоке/процессе через `@app.route` |

---

## 1. Целевая архитектура Flask

```
contracts-flask/
├── site/
│   ├── app.py                  # create_app(), регистрация blueprints
│   ├── config.py               # Настройки (DB, LLM, лимиты)
│   ├── routes/
│   │   ├── __init__.py          # register_routes(app)
│   │   ├── upload_bp.py         # POST /upload, /convert-doc, /unzip-upload
│   │   ├── pipeline_bp.py       # GET /process-v2 (SSE), POST /classify-batch
│   │   ├── api_bp.py            # GET/POST /api/* (groups, documents, supplements, sync, cleanup)
│   │   ├── prompts_bp.py        # GET/POST /api/prompts/*
│   │   ├── health_bp.py         # GET /health
│   │   └── pages_bp.py          # GET /, /architect (HTML-страницы)
│   ├── services/                # Бизнес-логика (перенос из deploy/compare/)
│   │   ├── __init__.py
│   │   ├── parse.py             # ← deploy/compare/parse.py
│   │   ├── classify.py          # ← deploy/compare/classify.py
│   │   ├── grouping.py          # ← deploy/compare/grouping.py
│   │   ├── llm.py               # ← deploy/compare/llm.py
│   │   ├── llm_client.py        # ← deploy/compare/llm_client.py
│   │   └── metrics.py           # ← deploy/compare/metrics.py
│   ├── db/                      # Перенос из deploy/db/
│   │   ├── __init__.py
│   │   ├── connection.py        # ← deploy/db/connection.py
│   │   ├── documents.py         # ← deploy/db/documents.py
│   │   ├── contracts.py         # ← deploy/db/contracts.py
│   │   ├── supplements.py       # ← deploy/db/supplements.py
│   │   ├── spec_current.py      # ← deploy/db/spec_current.py
│   │   ├── spec_events.py       # ← deploy/db/spec_events.py
│   │   └── prompts.py           # ← deploy/db/prompts.py
│   ├── repository.py            # ← deploy/repository.py (расширить)
│   ├── llm_prompt.py            # ← deploy/llm_prompt.py
│   ├── templates/
│   │   └── index.html           # УЖЕ ЕСТЬ, почти не меняется
│   └── static/                  # JS-файлы: state.js, app_utils.js, files.js, groups.js, compare.js, app.js
├── requirements.txt             # Flask + psycopg2 + httpx + pdfplumber + python-docx
├── Dockerfile
└── deploy/                      # ОСТАЁТСЯ только:
    ├── nginx-contracts.conf
    ├── sync.sh
    └── convert_doc.py           # libreoffice-конвертер (stdin→stdout, НЕ HTTP)
```

### Ключевое правило: ZERO монолита

- **Blueprints** — каждый на ≤100 строк, одна ответственность
- **services/** — чистые функции, никакого `request`/`Response`
- **db/** — чистый CRUD, никакого Flask
- **repository.py** — единая точка доступа к данным (DI во все services)

---

## 2. Пошаговый план (8 шагов)

### Шаг 1: Перенести `db/` как есть

**Файлы:** `deploy/db/*.py` → `site/db/*.py`

**Что делать:** Копировать. Менять НИЧЕГО не надо.
- `connection.py` уже использует `psycopg2.pool.ThreadedConnectionPool` — идеально для Flask (каждый request — своё соединение из пула)
- Все модули зависят только от `connection.query()` и `connection.execute()`

**Проверка:** `python3 -c "from site.db import documents; print(documents.list_by_batch('test'))"`

---

### Шаг 2: Перенести `services/` (бывший `compare/`)

**Файлы:** `deploy/compare/{parse,classify,grouping,llm,llm_client,metrics}.py` → `site/services/`

**Что делать:** Копировать, исправить импорты:
- `from db import ...` → `from site.db import ...`
- `from .parse import ...` → `from site.services.parse import ...`
- `from llm_prompt import ...` → `from site.llm_prompt import ...`

**НЕ переносить:** `upload.py`, `unzip.py`, `process.py` — они привязаны к HTTP и будут переписаны в blueprints.

**Проверка:** `python3 -c "from site.services.classify import classify_batch; print('ok')"`

---

### Шаг 3: Создать `config.py`

```python
# site/config.py
import os

DB_CONFIG = {
    "host": os.getenv("DB_HOST", "127.0.0.1"),
    "port": int(os.getenv("DB_PORT", "5432")),
    "dbname": os.getenv("DB_NAME", "baza"),
    "user": os.getenv("DB_USER", "super"),
    "password": os.getenv("DB_PASS", ""),
}

LLM_URL = os.getenv("LLM_API_URL", "https://api.aillm.ru/v1/chat/completions")
LLM_KEY = os.getenv("LLM_API_KEY", "")
LLM_MODEL = os.getenv("LLM_MODEL", "gpt-oss-120b")

MAX_CONTENT_LENGTH = 200 * 1024 * 1024  # 200 MB
VERSION = "2.0.0"
```

---

### Шаг 4: Blueprint `upload_bp.py` — загрузка файлов

Самый критичный blueprint. Замена `deploy/compare/upload.py` + `deploy/compare/unzip.py`.

```python
# site/routes/upload_bp.py
from flask import Blueprint, request, jsonify
from site.services.parse import parse_file
from site.db import documents, contracts
from site.config import MAX_CONTENT_LENGTH
import hashlib, base64, zipfile, io, os

upload_bp = Blueprint("upload", __name__)

ALLOWED = {"pdf", "docx", "doc", "zip"}

@upload_bp.route("/upload", methods=["POST"])
def upload():
    """Загрузка одного файла → парсинг → БД."""
    f = request.files.get("files")
    if not f:
        return jsonify(ok=False, error="no file"), 400

    ext = f.filename.rsplit(".", 1)[-1].lower() if "." in f.filename else ""
    if ext not in ALLOWED:
        return jsonify(ok=False, error=f"unsupported: .{ext}"), 400

    data = f.read()
    content_hash = hashlib.sha256(data).hexdigest()[:16]

    batch_id = request.form.get("batch_id")
    zip_source = request.form.get("zip_source")

    # Проверка дубликата по хешу
    if batch_id:
        existing = documents.get_by_hash(batch_id, content_hash)
        if existing:
            return jsonify(ok=False, error="duplicate", doc_id=existing["id"])

    doc = documents.insert(
        filename=f.filename,
        mime_type=f.content_type or "application/octet-stream",
        original_bytes=base64.b64encode(data).decode(),
        batch_id=batch_id,
        zip_source=zip_source,
        content_hash=content_hash,
    )

    # Парсинг
    try:
        result = parse_file(f.filename, data)
        if result["status"] == "parsed":
            documents.set_parsed(doc["id"], result["elements"])
        else:
            documents.set_error(doc["id"], result.get("error", "parse failed"))
    except Exception as e:
        documents.set_error(doc["id"], str(e))
        result = {"status": "error", "error": str(e)}

    contract_id = request.form.get("contract_id")
    return jsonify(ok=True, doc_id=doc["id"], contract_id=contract_id,
                   parsed={"status": result["status"], "element_count": result.get("element_count", 0)})


@upload_bp.route("/convert-doc", methods=["POST"])
def convert_doc():
    """.doc → .docx через libreoffice."""
    import subprocess, tempfile
    f = request.files.get("files")
    if not f:
        return jsonify(ok=False, error="no file"), 400
    data = f.read()
    with tempfile.NamedTemporaryFile(suffix=".doc", delete=False) as tmp:
        tmp.write(data)
        doc_path = tmp.name
    tmpdir = tempfile.mkdtemp()
    try:
        subprocess.run(["libreoffice", "--headless", "--convert-to", "docx", "--outdir", tmpdir, doc_path],
                       timeout=30, capture_output=True)
        docx_files = [x for x in os.listdir(tmpdir) if x.endswith(".docx")]
        if docx_files:
            with open(os.path.join(tmpdir, docx_files[0]), "rb") as out:
                from flask import send_file
                return send_file(io.BytesIO(out.read()), mimetype="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
        return jsonify(ok=False, error="conversion produced no output"), 500
    finally:
        os.unlink(doc_path)
        for x in os.listdir(tmpdir):
            os.unlink(os.path.join(tmpdir, x))
        os.rmdir(tmpdir)


@upload_bp.route("/unzip-upload", methods=["POST"])
def unzip_upload():
    """Распаковать ZIP → список файлов (base64)."""
    f = request.files.get("files")
    if not f:
        return jsonify(ok=False, error="no file"), 400
    data = f.read()
    MAX_FILES = 500
    MAX_UNCOMPRESSED = 500 * 1024 * 1024
    files = []
    total = 0
    with zipfile.ZipFile(io.BytesIO(data)) as zf:
        if len(zf.namelist()) > MAX_FILES:
            return jsonify(ok=False, error=f"too many files (max {MAX_FILES})"), 400
        for info in zf.infolist():
            if info.is_dir():
                continue
            name = os.path.basename(info.filename)
            if not name or ".." in name:
                continue
            raw = zf.read(info)
            total += len(raw)
            if total > MAX_UNCOMPRESSED:
                return jsonify(ok=False, error="total uncompressed > 500MB"), 400
            ext = name.rsplit(".", 1)[-1].lower() if "." in name else ""
            files.append({
                "filename": name,
                "ext": ext,
                "size": len(raw),
                "data_b64": base64.b64encode(raw).decode(),
            })
    return jsonify(ok=True, files=files)
```

**Критично:**
- `request.files` вместо `cgi.FieldStorage` — файл уже в памяти
- `base64` всё ещё нужен (JS на фронте делает `atob()`)
- Таймауты: Flask не режет сам — поставить `timeout` на libreoffice

---

### Шаг 5: Blueprint `pipeline_bp.py` — SSE + classify

```python
# site/routes/pipeline_bp.py
from flask import Blueprint, request, jsonify, Response, stream_with_context
from site.services.process import run_pipeline  # адаптированный process.py
from site.services.classify import classify_batch
from site.llm_prompt import build_prompt
from site.db import documents
import json, re, os, threading, sys

pipeline_bp = Blueprint("pipeline", __name__)

@pipeline_bp.route("/process-v2", methods=["GET"])
def process_v2():
    """SSE-стриминг сравнения."""
    cid = request.args.get("contract_id")
    if not cid or not re.fullmatch(r'[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}', cid, re.I):
        return jsonify(ok=False, error="invalid contract_id"), 400

    order_ids = request.args.get("order", "")

    def generate():
        yield ": ok\n\n"
        try:
            for event in run_pipeline(cid, order_ids, build_prompt):
                yield f"data: {json.dumps(event, ensure_ascii=False)}\n\n"
        except GeneratorExit:
            return
        except Exception as e:
            yield f"data: {json.dumps({'type': 'error', 'message': str(e)}, ensure_ascii=False)}\n\n"

    return Response(
        stream_with_context(generate()),
        content_type="text/event-stream; charset=utf-8",
        headers={
            "Cache-Control": "no-cache",
            "X-Accel-Buffering": "no",  # ← nginx не буферизует
        }
    )


@pipeline_bp.route("/api/classify-batch", methods=["POST"])
def classify_batch_route():
    """Запустить классификацию. Синхронно для малых батчей, 202 для больших."""
    body = request.get_json()
    batch_id = body.get("batch_id")
    if not batch_id:
        return jsonify(ok=False, error="batch_id required"), 400

    pending = documents.list_pending(batch_id)
    total = len(pending)
    if total == 0:
        return jsonify(ok=False, error="no pending documents"), 400

    # Для ≤10 файлов — синхронно (быстрее, проще)
    if total <= 10:
        result = classify_batch(batch_id)
        return jsonify(result)

    # Для >10 файлов — в отдельном потоке, сразу 202
    lock_path = f"/tmp/classify_{batch_id}.lock"
    if os.path.exists(lock_path):
        return jsonify(ok=False, error="classify already running"), 409

    with open(lock_path, "w") as lf:
        lf.write(str(os.getpid()))

    def _run():
        try:
            classify_batch(batch_id)
        finally:
            if os.path.exists(lock_path):
                os.remove(lock_path)

    threading.Thread(target=_run, daemon=True).start()
    return jsonify(ok=True, total=total), 202
```

**Критично:**
- `compare/process.py` нужно адаптировать: `run_pipeline` должен стать **генератором** (yield события), а не принимать `sse_send` callback
- `GeneratorExit` в генераторе — обязательно
- `X-Accel-Buffering: no` — иначе nginx буферизует SSE
- classify: синхронно для ≤10 файлов, Thread для >10 (вместо subprocess)

---

### Шаг 6: Blueprint `api_bp.py` — всё остальное API

```python
# site/routes/api_bp.py
from flask import Blueprint, request, jsonify
from site.db import documents, supplements, contracts, spec_current
from site.services.grouping import group_documents, apply_groups
from site.db.connection import execute, query

api_bp = Blueprint("api", __name__)

@api_bp.route("/api/supplements")
def api_supplements():
    cid = request.args.get("contract_id")
    if not cid:
        return jsonify(ok=False, error="contract_id required"), 400
    rows = supplements.list_by_contract(cid)
    return jsonify(ok=True, supplements=rows)

@api_bp.route("/api/documents/<doc_id>")
def api_document(doc_id):
    doc = documents.get(doc_id)
    if not doc:
        return jsonify(ok=False, error="not found"), 404
    return jsonify(ok=True, **{k: doc.get(k) for k in [
        "id", "filename", "status", "elements_json", "doc_type",
        "own_number", "parent_number", "doc_date", "counterparty",
        "classify_status", "classify_raw", "classify_input"
    ]})

@api_bp.route("/api/documents/<doc_id>", methods=["DELETE"])
def api_document_delete(doc_id):
    supps = query("SELECT id, contract_id FROM supplements WHERE document_id = %s", (doc_id,))
    for s in (supps or []):
        execute("DELETE FROM spec_current WHERE contract_id = %s AND last_event_id IN (SELECT id FROM spec_events WHERE supplement_id = %s)", (s["contract_id"], s["id"]))
        execute("DELETE FROM spec_events WHERE supplement_id = %s", (s["id"],))
        execute("DELETE FROM supplements WHERE id = %s", (s["id"],))
    execute("DELETE FROM documents WHERE id = %s", (doc_id,))
    return jsonify(ok=True)

@api_bp.route("/api/sync", methods=["POST"])
def api_sync():
    body = request.get_json()
    keep_ids = set(body.get("keep_ids", []))
    if len(keep_ids) > 1000:
        return jsonify(ok=False, error="too many keep_ids"), 400
    docs = query("SELECT id FROM documents", ())
    deleted = 0
    for d in (docs or []):
        if d["id"] in keep_ids:
            continue
        supps = query("SELECT id, contract_id FROM supplements WHERE document_id = %s", (d["id"],))
        for s in (supps or []):
            execute("DELETE FROM spec_current WHERE contract_id = %s AND last_event_id IN (SELECT id FROM spec_events WHERE supplement_id = %s)", (s["contract_id"], s["id"]))
            execute("DELETE FROM spec_events WHERE supplement_id = %s", (s["id"],))
            execute("DELETE FROM supplements WHERE id = %s", (s["id"],))
        execute("DELETE FROM documents WHERE id = %s", (d["id"],))
        deleted += 1
    execute("DELETE FROM contracts WHERE id NOT IN (SELECT DISTINCT contract_id FROM supplements)")
    return jsonify(ok=True, deleted=deleted)

@api_bp.route("/api/groups")
def api_groups():
    batch_id = request.args.get("batch")
    if not batch_id:
        return jsonify(ok=False, error="batch required"), 400
    result = group_documents(batch_id)
    return jsonify(result)

@api_bp.route("/api/batch-progress")
def api_batch_progress():
    batch_id = request.args.get("batch")
    if not batch_id:
        return jsonify(ok=False, error="batch required"), 400
    counts = documents.count_by_status(batch_id)
    docs = documents.list_by_batch(batch_id)
    return jsonify(ok=True, counts=counts, total=len(docs))

@api_bp.route("/api/apply-groups", methods=["POST"])
def api_apply_groups():
    body = request.get_json()
    batch_id = body.get("batch_id")
    groups = body.get("groups", [])
    if not batch_id:
        return jsonify(ok=False, error="batch_id required"), 400
    result = apply_groups(batch_id, groups)
    return jsonify(result)

@api_bp.route("/api/spec-current")
def api_spec_current():
    cid = request.args.get("contract_id")
    if not cid:
        return jsonify(ok=False, error="contract_id required"), 400
    rows = spec_current.list_by_contract(cid)
    return jsonify(ok=True, rows=rows)

@api_bp.route("/api/cleanup", methods=["POST"])
def api_cleanup():
    execute("DELETE FROM spec_current")
    execute("DELETE FROM spec_events")
    execute("DELETE FROM supplements")
    execute("DELETE FROM upload_chunks")
    execute("DELETE FROM documents")
    execute("DELETE FROM contracts")
    return jsonify(ok=True, message="all data cleaned")
```

---

### Шаг 7: `app.py` — точка входа

```python
# site/app.py
from flask import Flask, render_template
from site.config import VERSION, MAX_CONTENT_LENGTH

def create_app():
    app = Flask(__name__)
    app.config["VERSION"] = VERSION
    app.config["MAX_CONTENT_LENGTH"] = MAX_CONTENT_LENGTH

    # Blueprints
    from site.routes.upload_bp import upload_bp
    from site.routes.pipeline_bp import pipeline_bp
    from site.routes.api_bp import api_bp
    from site.routes.prompts_bp import prompts_bp
    from site.routes.health_bp import health_bp
    from site.routes.pages_bp import pages_bp

    app.register_blueprint(upload_bp)
    app.register_blueprint(pipeline_bp)
    app.register_blueprint(api_bp)
    app.register_blueprint(prompts_bp)
    app.register_blueprint(health_bp)
    app.register_blueprint(pages_bp)

    @app.after_request
    def no_cache(response):
        response.headers["Cache-Control"] = "no-cache, no-store, must-revalidate"
        response.headers["Pragma"] = "no-cache"
        response.headers["Expires"] = "0"
        return response

    return app

app = create_app()
```

---

### Шаг 8: Адаптировать `compare/process.py` в генератор

Текущий `run_pipeline(cid, order_ids, sse_send, build_prompt_fn)` принимает callback `sse_send`. Нужно сделать генератором:

```python
# site/services/process.py
def run_pipeline(contract_id, order_ids, build_prompt_fn):
    """Generator: yield SSE events."""
    # ... та же логика, но вместо sse_send(event) → yield event
    # ... GeneratorExit уже обрабатывается во view
```

---

## 3. Что НЕ трогаем (остаётся как есть)

| Что | Почему |
|-----|--------|
| `templates/index.html` | Уже работает, меняются только URL (с ВМ на свои) |
| `static/*.js` (6 файлов) | Меняется только `VM_API` → `""` (свои же endpoints) |
| `convert_doc.py` | Остаётся как утилита (libreoffice), вызывается из upload_bp |
| CSS, иконки, модалки | Без изменений |

---

## 4. Изменения во фронтенде (минимальные)

В `deploy/app.js` (статика) поменять ОДНУ строку:

```javascript
// Было:
var VM_API = 'https://contracts.kube5s.ru';
// Стало:
var VM_API = '';  // все API на том же домене
```

ВСЁ. Больше ничего не меняется — XHR, SSE, fetch работают с теми же путями.

---

## 5. Последовательность выполнения

| # | Шаг | Сложность | Риск |
|---|-----|-----------|------|
| 1 | `db/` → `site/db/` | Низкая | Низкий — просто копирование |
| 2 | `compare/` → `site/services/` | Низкая | Низкий — правим импорты |
| 3 | `config.py` | Низкая | Низкий |
| 4 | `upload_bp.py` | Средняя | **Высокий** — ключевой функционал |
| 5 | `pipeline_bp.py` + адаптация `process.py` | Высокая | **Высокий** — SSE критичен |
| 6 | `api_bp.py` | Средняя | Средний — много ручек |
| 7 | `prompts_bp.py`, `health_bp.py`, `pages_bp.py` | Низкая | Низкий |
| 8 | `app.py` + тесты | Средняя | Средний |

---

## 6. Чек-лист перед деплоем

- [ ] `python3 -c "from site.app import app"` — приложение создаётся без ошибок
- [ ] `/health` → `{"ok": true}`
- [ ] `POST /upload` с реальным PDF → `{"ok": true, "doc_id": "..."}`
- [ ] `POST /unzip-upload` с ZIP → список файлов
- [ ] `GET /process-v2?contract_id=...` → SSE-поток (curl test)
- [ ] `POST /api/classify-batch` → классификация работает
- [ ] `GET /api/groups?batch=...` → группы
- [ ] `POST /api/apply-groups` → создаются supplements
- [ ] `GET /api/spec-current?contract_id=...` → спецификация
- [ ] `GET /` → HTML с таблицей файлов
- [ ] `no_cache` after_request — есть
- [ ] `X-Accel-Buffering: no` на SSE
- [ ] `GeneratorExit` в SSE-генераторе
- [ ] `stream_with_context` на SSE
- [ ] `MAX_CONTENT_LENGTH = 200 MB`
- [ ] `VM_API = ''` в app.js (фронтенд)
- [ ] Все старые тесты проходят

---

## 7. Риски и mitigation

| Риск | Mitigation |
|------|-----------|
| SSE зависает под нагрузкой | `stream_with_context` + `X-Accel-Buffering: no` |
| classify блокирует HTTP | Thread для >10 файлов, sync для ≤10 |
| libreoffice падает | timeout=30, отдельный процесс |
| Коннекты к БД исчерпываются | ThreadedConnectionPool уже есть, minconn=1, maxconn=10 |
| ZIP-бомба | Проверка MAX_UNCOMPRESSED = 500MB, MAX_RATIO |
| Загрузка больших PDF (>100MB) | MAX_CONTENT_LENGTH = 200MB, XHR timeout = 180s |