Georges Haddad
3b3bcf2cf2
- Add stage/process-staged/remove-staged endpoints for batch upload workflow - Add staging gallery UI with thumbnail previews and per-image removal - Inherit person info & doc fields from page 1 for multi-page PDFs - Auto-link page 2+ to same person on confirmation - Show info banner on review page for multi-page documents - Exclude staged docs from stats counters
91 lines
3.1 KiB
Python
91 lines
3.1 KiB
Python
import os
|
|
from pathlib import Path
|
|
|
|
from fastapi import APIRouter, Request
|
|
from fastapi.responses import JSONResponse
|
|
from fastapi.templating import Jinja2Templates
|
|
|
|
from config import UPLOAD_DIR
|
|
from database.connection import get_db
|
|
|
|
router = APIRouter()
|
|
templates = Jinja2Templates(directory="templates")
|
|
|
|
|
|
@router.get("/documents")
|
|
async def document_queue(request: Request, status: str = "", uploaded: int = 0):
|
|
with get_db() as conn:
|
|
if status:
|
|
rows = conn.execute(
|
|
"""SELECT d.*, p.first_name, p.family_name
|
|
FROM documents d
|
|
LEFT JOIN persons p ON p.id = d.person_id
|
|
WHERE d.status=?
|
|
ORDER BY d.id DESC""",
|
|
(status,),
|
|
).fetchall()
|
|
else:
|
|
rows = conn.execute(
|
|
"""SELECT d.*, p.first_name, p.family_name
|
|
FROM documents d
|
|
LEFT JOIN persons p ON p.id = d.person_id
|
|
ORDER BY d.id DESC"""
|
|
).fetchall()
|
|
|
|
stats = conn.execute(
|
|
"""SELECT
|
|
COUNT(*) AS total,
|
|
SUM(CASE WHEN status='confirmed' THEN 1 ELSE 0 END) AS confirmed,
|
|
SUM(CASE WHEN status='extracted' THEN 1 ELSE 0 END) AS pending_review,
|
|
SUM(CASE WHEN status='pending' THEN 1 ELSE 0 END) AS processing,
|
|
SUM(CASE WHEN status='error' THEN 1 ELSE 0 END) AS errors
|
|
FROM documents WHERE status != 'staged'"""
|
|
).fetchone()
|
|
|
|
return templates.TemplateResponse(
|
|
request,
|
|
"documents.html",
|
|
{
|
|
"documents": [dict(r) for r in rows],
|
|
"stats": dict(stats) if stats else {},
|
|
"current_status": status,
|
|
"uploaded": uploaded,
|
|
},
|
|
)
|
|
|
|
|
|
@router.delete("/documents/{doc_id}")
|
|
async def delete_document(doc_id: int):
|
|
"""Delete a document and its properties. Also removes the image file from disk."""
|
|
with get_db() as conn:
|
|
doc = conn.execute(
|
|
"SELECT image_path, person_id FROM documents WHERE id=?", (doc_id,)
|
|
).fetchone()
|
|
if not doc:
|
|
return JSONResponse({"error": "not found"}, status_code=404)
|
|
|
|
# Delete properties for this document
|
|
conn.execute("DELETE FROM properties WHERE document_id=?", (doc_id,))
|
|
|
|
# Delete the document record
|
|
conn.execute("DELETE FROM documents WHERE id=?", (doc_id,))
|
|
|
|
# If the person has no remaining documents, remove them too
|
|
if doc["person_id"]:
|
|
remaining = conn.execute(
|
|
"SELECT COUNT(*) AS n FROM documents WHERE person_id=?",
|
|
(doc["person_id"],),
|
|
).fetchone()["n"]
|
|
if remaining == 0:
|
|
conn.execute("DELETE FROM properties WHERE person_id=?", (doc["person_id"],))
|
|
conn.execute("DELETE FROM persons WHERE id=?", (doc["person_id"],))
|
|
|
|
# Remove the image file from disk
|
|
image_file = Path(UPLOAD_DIR) / doc["image_path"]
|
|
try:
|
|
image_file.unlink(missing_ok=True)
|
|
except OSError:
|
|
pass
|
|
|
|
return JSONResponse({"ok": True})
|