new web ract

api/route/admin_vk.py

@@ -0,0 +1,113 @@
+from fastapi import APIRouter, BackgroundTasks, Depends, HTTPException
+from pydantic import BaseModel
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from bd.database import get_db
+from bd.models import VkSource
+from route.deps import require_admin
+from vk_parser import run_import, run_history_import, VK_TOKEN
+
+router = APIRouter()
+
+
+class SourceIn(BaseModel):
+    group_id: str
+    group_name: str
+    enabled: bool = True
+
+
+def _source_dict(s: VkSource) -> dict:
+    return {
+        "id":         str(s.id),
+        "group_id":   s.group_id,
+        "group_name": s.group_name,
+        "enabled":    s.enabled,
+        "last_run":   s.last_run.isoformat() if s.last_run else None,
+        "created_at": s.created_at.isoformat(),
+    }
+
+
+@router.get("/status")
+async def vk_status(_: str = Depends(require_admin)):
+    return {"configured": bool(VK_TOKEN)}
+
+
+# ── CRUD источников ───────────────────────────────────────────────────────────
+
+@router.get("/sources")
+async def list_sources(db: AsyncSession = Depends(get_db), _: str = Depends(require_admin)):
+    rows = (await db.execute(select(VkSource).order_by(VkSource.created_at))).scalars().all()
+    return [_source_dict(r) for r in rows]
+
+
+def _clean_group_id(raw: str) -> str:
+    """Из строки вида '-95503797', '95503797_6512', 'wall-95503797_12' вытащить только числовой ID группы."""
+    import re
+    raw = raw.strip().lstrip("-")
+    # Берём только первое число (до '_' если есть)
+    m = re.search(r"(\d+)", raw)
+    return m.group(1) if m else raw
+
+
+@router.post("/sources", status_code=201)
+async def add_source(data: SourceIn, db: AsyncSession = Depends(get_db), _: str = Depends(require_admin)):
+    group_id = _clean_group_id(data.group_id)
+    exists = (await db.execute(select(VkSource).where(VkSource.group_id == group_id))).scalar_one_or_none()
+    if exists:
+        raise HTTPException(status_code=409, detail="Источник с таким group_id уже есть")
+    source = VkSource(group_id=group_id, group_name=data.group_name, enabled=data.enabled)
+    db.add(source)
+    await db.commit()
+    await db.refresh(source)
+    return _source_dict(source)
+
+
+@router.patch("/sources/{source_id}")
+async def update_source(source_id: str, data: SourceIn, db: AsyncSession = Depends(get_db), _: str = Depends(require_admin)):
+    source = (await db.execute(select(VkSource).where(VkSource.id == source_id))).scalar_one_or_none()
+    if not source:
+        raise HTTPException(status_code=404, detail="Источник не найден")
+    source.group_id   = _clean_group_id(data.group_id)
+    source.group_name = data.group_name
+    source.enabled    = data.enabled
+    await db.commit()
+    await db.refresh(source)
+    return _source_dict(source)
+
+
+@router.delete("/sources/{source_id}", status_code=204)
+async def delete_source(source_id: str, db: AsyncSession = Depends(get_db), _: str = Depends(require_admin)):
+    source = (await db.execute(select(VkSource).where(VkSource.id == source_id))).scalar_one_or_none()
+    if not source:
+        raise HTTPException(status_code=404, detail="Источник не найден")
+    await db.delete(source)
+    await db.commit()
+
+
+# ── Ручной запуск импорта ─────────────────────────────────────────────────────
+
+@router.post("/import")
+async def vk_manual_import(background_tasks: BackgroundTasks, _: str = Depends(require_admin)):
+    background_tasks.add_task(run_import)
+    return {"ok": True, "message": "Импорт последних постов запущен в фоне"}
+
+
+@router.post("/import/history")
+async def vk_history_import_all(background_tasks: BackgroundTasks, _: str = Depends(require_admin)):
+    background_tasks.add_task(run_history_import)
+    return {"ok": True, "message": "Исторический импорт всех групп запущен в фоне (до 5 мин)"}
+
+
+@router.post("/import/history/{source_id}")
+async def vk_history_import_one(
+    source_id: str,
+    background_tasks: BackgroundTasks,
+    db: AsyncSession = Depends(get_db),
+    _: str = Depends(require_admin),
+):
+    source = (await db.execute(select(VkSource).where(VkSource.id == source_id))).scalar_one_or_none()
+    if not source:
+        raise HTTPException(status_code=404, detail="Источник не найден")
+    background_tasks.add_task(run_history_import, group_id=source.group_id)
+    return {"ok": True, "message": f"Исторический импорт группы «{source.group_name}» запущен в фоне"}