scripts/mobile_status.py - master - Max_Cherep/super-resolution

scripts/mobile_status.py
109 строк4 KB
maxim.cherepanov
add timezone-aware job stats for mobile clients
15 июл 2026, 15:09
15 июл 2026, 15:09ffcf55e
"""Сводка для мобильного клиента (Android): очередь и worker."""

from __future__ import annotations

import time
from pathlib import Path

import pika

from db import get_conn
from messaging import QUEUE_NAME, rabbit_connection_params
from process_options import job_options_summary
from stats import build_stats
from version import get_version_info


def _rabbit_queue_stats() -> tuple[int, int]:
    """(сообщений в очереди ready, число consumer)."""
    try:
        conn = pika.BlockingConnection(rabbit_connection_params(heartbeat=None))
        try:
            ch = conn.channel()
            q = ch.queue_declare(queue=QUEUE_NAME, passive=True)
            return int(q.method.message_count), int(q.method.consumer_count)
        finally:
            conn.close()
    except (pika.exceptions.AMQPConnectionError, pika.exceptions.ChannelClosedByBroker):
        return 0, 0


def _jobs_queued() -> int:
    with get_conn() as conn:
        row = conn.execute(
            "SELECT COUNT(*) AS n FROM jobs WHERE status = 'queued'",
        ).fetchone()
        return int(row["n"]) if row else 0


def _jobs_processing() -> int:
    with get_conn() as conn:
        row = conn.execute(
            "SELECT COUNT(*) AS n FROM jobs WHERE status = 'processing'",
        ).fetchone()
        return int(row["n"]) if row else 0


def _input_format(filename: str) -> str | None:
    ext = Path(filename).suffix.lstrip(".").lower()
    return ext or None


def _current_job() -> dict | None:
    with get_conn() as conn:
        row = conn.execute(
            """
            SELECT * FROM jobs
            WHERE status = 'processing'
            ORDER BY started_at ASC
            LIMIT 1
            """,
        ).fetchone()
    if not row:
        return None
    job = dict(row)
    filename = job.get("filename") or ""
    pct = job.get("progress_pct")
    return {
        "id": job["id"],
        "filename": filename,
        "input_format": _input_format(filename),
        "job_type": job.get("job_type") or "process",
        "progress_pct": float(pct) if pct is not None else None,
        "progress_detail": job.get("progress_detail"),
        "output_format": job.get("output_format") or "wav",
        "options_summary": job_options_summary(job),
    }


def build_mobile_status(tz_name: str | None = None) -> dict:
    """
    Плоский JSON для GET /api/mobile-status.
    queue_size — max(queued в SQLite, ready в RabbitMQ).
    workers_total — consumer_count из RabbitMQ (0 если worker не подключён).
    workers_busy — jobs в processing (0 или 1 при одном GPU).
    current_job — активная задача или null.
    tz_name — IANA timezone клиента; без параметра — UTC.
    """
    rmq_ready, consumers = _rabbit_queue_stats()
    queued = _jobs_queued()
    processing = _jobs_processing()
    queue_size = max(queued, rmq_ready)
    ver = get_version_info()
    stats = build_stats(tz_name)

    return {
        "status": "ok",
        "timestamp": int(time.time()),
        "app_version": ver["label"],
        "app_build": ver["build"],
        "queue_size": queue_size,
        "workers_total": consumers,
        "workers_busy": processing,
        "timezone": stats["timezone"],
        "local_date": stats["local_date"],
        "tasks_completed_today": stats["completed_today"],
        "tasks_completed_yesterday": stats["completed_yesterday"],
        "ready_downloads": stats["ready_downloads"],
        "current_job": _current_job(),
    }