Fix dashboard review speed: DB-only decisions (0.03s vs 11.6s); add process-staging flow for moves
This commit is contained in:
@@ -5,10 +5,13 @@ Reads photo_pipeline.db + staging dirs on .13. Serves:
|
|||||||
/review — thumbnail review grid with keep/reject/reset (htmx)
|
/review — thumbnail review grid with keep/reject/reset (htmx)
|
||||||
/file/{sha} — full-size original image
|
/file/{sha} — full-size original image
|
||||||
/thumbs/... — generated thumbnails
|
/thumbs/... — generated thumbnails
|
||||||
/stats — JSON stats (for future dashboard widgets)
|
/stats — JSON stats
|
||||||
|
|
||||||
|
Review actions update the DB ONLY (instant). File moves happen at import time
|
||||||
|
(via the immich-import / process-staging flow) — this keeps review responsive
|
||||||
|
even for large batches; cross-filesystem moves are slow.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import shutil
|
|
||||||
import sqlite3
|
import sqlite3
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
@@ -82,7 +85,6 @@ def index(request: Request):
|
|||||||
).fetchall()
|
).fetchall()
|
||||||
total = conn.execute("SELECT COUNT(*) FROM image_hashes").fetchone()[0]
|
total = conn.execute("SELECT COUNT(*) FROM image_hashes").fetchone()[0]
|
||||||
conn.close()
|
conn.close()
|
||||||
# folder targets (staging dirs)
|
|
||||||
folders = {}
|
folders = {}
|
||||||
for name in ("01_keep", "02_review", "03_delete"):
|
for name in ("01_keep", "02_review", "03_delete"):
|
||||||
d = STAGING / name
|
d = STAGING / name
|
||||||
@@ -104,6 +106,8 @@ def review(request: Request, source: str = None, status: str = None):
|
|||||||
if status:
|
if status:
|
||||||
q += " AND status=?"
|
q += " AND status=?"
|
||||||
params.append(status)
|
params.append(status)
|
||||||
|
else:
|
||||||
|
q += " AND status IN ('scanned','review')"
|
||||||
q += " ORDER BY added_at DESC LIMIT 300"
|
q += " ORDER BY added_at DESC LIMIT 300"
|
||||||
rows = conn.execute(q, params).fetchall()
|
rows = conn.execute(q, params).fetchall()
|
||||||
conn.close()
|
conn.close()
|
||||||
@@ -114,9 +118,8 @@ def review(request: Request, source: str = None, status: str = None):
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@app.get("/stats", response_class=FileResponse)
|
@app.get("/stats")
|
||||||
def stats():
|
def stats():
|
||||||
"""JSON stats for widgets."""
|
|
||||||
import json
|
import json
|
||||||
|
|
||||||
conn = _conn()
|
conn = _conn()
|
||||||
@@ -126,64 +129,43 @@ def stats():
|
|||||||
by_source = dict(conn.execute(
|
by_source = dict(conn.execute(
|
||||||
"SELECT source, COUNT(*) FROM image_hashes GROUP BY source").fetchall())
|
"SELECT source, COUNT(*) FROM image_hashes GROUP BY source").fetchall())
|
||||||
conn.close()
|
conn.close()
|
||||||
return FileResponse(path=None, content=json.dumps(
|
return json.dumps({"total": total, "by_status": by_status, "by_source": by_source})
|
||||||
{"total": total, "by_status": by_status, "by_source": by_source}))
|
|
||||||
|
|
||||||
|
|
||||||
def _set_status(sha: str, status: str, move_to: str | None, request: Request):
|
def _set_status(sha: str, status: str, request: Request) -> HTMLResponse:
|
||||||
"""Update DB status AND (optionally) move the file into a staging folder."""
|
"""Update DB status ONLY — instant. File moves happen at import time."""
|
||||||
conn = _conn()
|
conn = _conn()
|
||||||
row = conn.execute("SELECT * FROM image_hashes WHERE sha256=?", (sha,)).fetchone()
|
row = conn.execute("SELECT * FROM image_hashes WHERE sha256=?", (sha,)).fetchone()
|
||||||
if not row:
|
if not row:
|
||||||
conn.close()
|
conn.close()
|
||||||
raise HTTPException(404)
|
raise HTTPException(404)
|
||||||
src = Path(row["path"])
|
|
||||||
# move file if requested and source exists
|
|
||||||
if move_to and src.exists():
|
|
||||||
dest_dir = STAGING / move_to
|
|
||||||
dest_dir.mkdir(parents=True, exist_ok=True)
|
|
||||||
dest = dest_dir / src.name
|
|
||||||
try:
|
|
||||||
shutil.move(str(src), str(dest))
|
|
||||||
conn.execute(
|
|
||||||
"UPDATE image_hashes SET path=?, status=? WHERE sha256=?",
|
|
||||||
(str(dest), status, sha))
|
|
||||||
except Exception as e:
|
|
||||||
conn.close()
|
|
||||||
raise HTTPException(500, detail=f"move failed: {e}")
|
|
||||||
else:
|
|
||||||
conn.execute("UPDATE image_hashes SET status=? WHERE sha256=?", (status, sha))
|
conn.execute("UPDATE image_hashes SET status=? WHERE sha256=?", (status, sha))
|
||||||
conn.commit()
|
conn.commit()
|
||||||
|
row = conn.execute("SELECT * FROM image_hashes WHERE sha256=?", (sha,)).fetchone()
|
||||||
conn.close()
|
conn.close()
|
||||||
return templates.TemplateResponse(
|
return templates.TemplateResponse(
|
||||||
request, "_card.html", {"item": _load_item(row)})
|
request, "_card.html", {"item": _load_item(row)}
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@app.post("/review/{sha}/approve")
|
@app.post("/review/{sha}/approve")
|
||||||
def approve(sha: str, request: Request):
|
def approve(sha: str, request: Request):
|
||||||
# approved → move to 01_keep (ready for Immich import)
|
return _set_status(sha, "approved", request)
|
||||||
return _set_status(sha, "approved", "01_keep", request)
|
|
||||||
|
|
||||||
|
|
||||||
@app.post("/review/{sha}/reject")
|
@app.post("/review/{sha}/reject")
|
||||||
def reject(sha: str, request: Request):
|
def reject(sha: str, request: Request):
|
||||||
# rejected → move to 03_delete (holding, never auto-deleted)
|
return _set_status(sha, "rejected", request)
|
||||||
return _set_status(sha, "rejected", "03_delete", request)
|
|
||||||
|
|
||||||
|
|
||||||
@app.post("/review/{sha}/reset")
|
@app.post("/review/{sha}/reset")
|
||||||
def reset_status(sha: str, request: Request):
|
def reset_status(sha: str, request: Request):
|
||||||
# undo: move back to a neutral status without moving the file
|
return _set_status(sha, "scanned", request)
|
||||||
return _set_status(sha, "scanned", None, request)
|
|
||||||
|
|
||||||
|
|
||||||
@app.post("/review/bulk")
|
@app.post("/review/bulk")
|
||||||
async def bulk_action(request: Request):
|
async def bulk_action(request: Request):
|
||||||
"""Bulk keep/reject/reset for a list of sha256 values.
|
"""Bulk status update — DB only, instant. Body: {"action", "shas": []}"""
|
||||||
|
|
||||||
Body: JSON {"action": "approve"|"reject"|"reset", "shas": ["..."]}
|
|
||||||
Moves files accordingly; returns counts.
|
|
||||||
"""
|
|
||||||
import json
|
import json
|
||||||
|
|
||||||
body = json.loads(await request.body())
|
body = json.loads(await request.body())
|
||||||
@@ -191,39 +173,21 @@ async def bulk_action(request: Request):
|
|||||||
shas = body.get("shas", [])
|
shas = body.get("shas", [])
|
||||||
if action not in ("approve", "reject", "reset"):
|
if action not in ("approve", "reject", "reset"):
|
||||||
raise HTTPException(400, "action must be approve/reject/reset")
|
raise HTTPException(400, "action must be approve/reject/reset")
|
||||||
|
|
||||||
move_map = {"approve": "01_keep", "reject": "03_delete", "reset": None}
|
|
||||||
status_map = {"approve": "approved", "reject": "rejected", "reset": "scanned"}
|
status_map = {"approve": "approved", "reject": "rejected", "reset": "scanned"}
|
||||||
move_to = move_map[action]
|
|
||||||
status = status_map[action]
|
status = status_map[action]
|
||||||
|
|
||||||
conn = _conn()
|
conn = _conn()
|
||||||
moved = 0
|
updated = 0
|
||||||
missing = 0
|
missing = 0
|
||||||
for sha in shas:
|
for sha in shas:
|
||||||
row = conn.execute("SELECT path FROM image_hashes WHERE sha256=?", (sha,)).fetchone()
|
cur = conn.execute("UPDATE image_hashes SET status=? WHERE sha256=?", (status, sha))
|
||||||
if not row:
|
if cur.rowcount:
|
||||||
|
updated += 1
|
||||||
|
else:
|
||||||
missing += 1
|
missing += 1
|
||||||
continue
|
|
||||||
src = Path(row["path"])
|
|
||||||
if move_to and src.exists():
|
|
||||||
dest_dir = STAGING / move_to
|
|
||||||
dest_dir.mkdir(parents=True, exist_ok=True)
|
|
||||||
dest = dest_dir / src.name
|
|
||||||
try:
|
|
||||||
shutil.move(str(src), str(dest))
|
|
||||||
conn.execute(
|
|
||||||
"UPDATE image_hashes SET path=?, status=? WHERE sha256=?",
|
|
||||||
(str(dest), status, sha))
|
|
||||||
moved += 1
|
|
||||||
continue
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
conn.execute("UPDATE image_hashes SET status=? WHERE sha256=?", (status, sha))
|
|
||||||
moved += 1
|
|
||||||
conn.commit()
|
conn.commit()
|
||||||
conn.close()
|
conn.close()
|
||||||
return {"action": action, "moved": moved, "missing": missing, "total": len(shas)}
|
return {"action": action, "updated": updated, "missing": missing, "total": len(shas)}
|
||||||
|
|
||||||
|
|
||||||
@app.get("/file/{sha}")
|
@app.get("/file/{sha}")
|
||||||
|
|||||||
@@ -17,8 +17,8 @@
|
|||||||
{% if item["size_mb"] %}<span style="color:#666"> · {{ item["size_mb"] }}MB</span>{% endif %}
|
{% if item["size_mb"] %}<span style="color:#666"> · {{ item["size_mb"] }}MB</span>{% endif %}
|
||||||
</div>
|
</div>
|
||||||
<div class="actions">
|
<div class="actions">
|
||||||
<button class="approve" hx-post="/review/{{ item["sha256"] }}/approve" hx-target="#card-{{ sha8 }}" hx-swap="outerHTML">✅ Keep</button>
|
<button class="approve" hx-post="/review/{{ item["sha256"] }}/approve" hx-target="#card-{{ sha8 }}" hx-swap="outerHTML" hx-on::after-request="if(event.detail.successful){this.closest('.card').remove();updateCount();showToast('Approved');}">✅ Keep</button>
|
||||||
<button class="reject" hx-post="/review/{{ item["sha256"] }}/reject" hx-target="#card-{{ sha8 }}" hx-swap="outerHTML">🗑 Reject</button>
|
<button class="reject" hx-post="/review/{{ item["sha256"] }}/reject" hx-target="#card-{{ sha8 }}" hx-swap="outerHTML" hx-on::after-request="if(event.detail.successful){this.closest('.card').remove();updateCount();showToast('Rejected');}">🗑 Reject</button>
|
||||||
<button class="reset" hx-post="/review/{{ item["sha256"] }}/reset" hx-target="#card-{{ sha8 }}" hx-swap="outerHTML">↺</button>
|
<button class="reset" hx-post="/review/{{ item["sha256"] }}/reset" hx-target="#card-{{ sha8 }}" hx-swap="outerHTML">↺</button>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -40,6 +40,7 @@
|
|||||||
.reset { background: #444; color: #eee; }
|
.reset { background: #444; color: #eee; }
|
||||||
.none { color: #666; font-style: italic; padding: 2rem; text-align: center; }
|
.none { color: #666; font-style: italic; padding: 2rem; text-align: center; }
|
||||||
.selall { display: inline-flex; align-items: center; gap: .35rem; background: #222; border: 1px solid #444; border-radius: 6px; padding: .4rem .8rem; cursor: pointer; }
|
.selall { display: inline-flex; align-items: center; gap: .35rem; background: #222; border: 1px solid #444; border-radius: 6px; padding: .4rem .8rem; cursor: pointer; }
|
||||||
|
.toast { position: fixed; bottom: 1.2rem; right: 1.2rem; background: #1d4; color: #031; padding: .7rem 1rem; border-radius: 8px; display: none; z-index: 50; font-size: .9rem; box-shadow: 0 2px 12px rgba(0,0,0,.5); }
|
||||||
</style>
|
</style>
|
||||||
</head>
|
</head>
|
||||||
<body>
|
<body>
|
||||||
@@ -70,6 +71,7 @@
|
|||||||
<button class="btn-reject" id="bulk-reject">🗑 Bulk Reject</button>
|
<button class="btn-reject" id="bulk-reject">🗑 Bulk Reject</button>
|
||||||
<button class="btn-reset" id="bulk-reset">↺ Reset</button>
|
<button class="btn-reset" id="bulk-reset">↺ Reset</button>
|
||||||
</div>
|
</div>
|
||||||
|
<div class="toast" id="toast"></div>
|
||||||
<p class="count">{{ items|length }} images</p>
|
<p class="count">{{ items|length }} images</p>
|
||||||
<main>
|
<main>
|
||||||
<div class="grid" id="review-grid">
|
<div class="grid" id="review-grid">
|
||||||
@@ -120,22 +122,56 @@
|
|||||||
async function bulk(action) {
|
async function bulk(action) {
|
||||||
if (selected.size === 0) return;
|
if (selected.size === 0) return;
|
||||||
const shas = [...selected];
|
const shas = [...selected];
|
||||||
|
// disable buttons during the operation (cross-filesystem moves can be slow)
|
||||||
|
setBulkBusy(true);
|
||||||
|
try {
|
||||||
const resp = await fetch('/review/bulk', {
|
const resp = await fetch('/review/bulk', {
|
||||||
method: 'POST',
|
method: 'POST',
|
||||||
headers: { 'Content-Type': 'application/json' },
|
headers: { 'Content-Type': 'application/json' },
|
||||||
body: JSON.stringify({ action, shas })
|
body: JSON.stringify({ action, shas })
|
||||||
});
|
});
|
||||||
const data = await resp.json();
|
const data = await resp.json();
|
||||||
alert(`Bulk ${action}: ${data.moved} processed, ${data.missing} missing`);
|
// remove processed cards from DOM immediately (match by sha in value attr)
|
||||||
// refresh: remove done cards
|
|
||||||
shas.forEach(s => {
|
shas.forEach(s => {
|
||||||
const card = document.querySelector(`[data-name="${s}"]`)?.closest('.card');
|
const cb = document.querySelector(`.item-check[value="${s}"]`);
|
||||||
if (card) card.remove();
|
if (cb) cb.closest('.card')?.remove();
|
||||||
});
|
});
|
||||||
selected.clear();
|
selected.clear();
|
||||||
refresh();
|
refresh();
|
||||||
location.reload();
|
updateCount();
|
||||||
|
showToast(`Bulk ${action}: ${data.moved} done, ${data.missing} missing`);
|
||||||
|
} catch (e) {
|
||||||
|
showToast('Bulk action failed: ' + e, true);
|
||||||
|
} finally {
|
||||||
|
setBulkBusy(false);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function setBulkBusy(busy) {
|
||||||
|
['bulk-keep','bulk-reject','bulk-reset'].forEach(id => {
|
||||||
|
const b = document.getElementById(id);
|
||||||
|
if (b) { b.disabled = busy; b.textContent = busy ? 'Working…' : b.dataset.orig; }
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function updateCount() {
|
||||||
|
const el = document.querySelector('.count');
|
||||||
|
const n = document.querySelectorAll('.card').length;
|
||||||
|
if (el) el.textContent = n + ' images';
|
||||||
|
}
|
||||||
|
|
||||||
|
function showToast(msg, isErr) {
|
||||||
|
const t = document.getElementById('toast');
|
||||||
|
t.textContent = msg;
|
||||||
|
t.style.background = isErr ? '#d43' : '#1d4';
|
||||||
|
t.style.color = isErr ? '#fff' : '#031';
|
||||||
|
t.style.display = 'block';
|
||||||
|
setTimeout(() => { t.style.display = 'none'; }, 3000);
|
||||||
|
}
|
||||||
|
|
||||||
|
// stash original labels for busy-state restore
|
||||||
|
document.querySelectorAll('.bulk-bar button').forEach(b => { b.dataset.orig = b.textContent; });
|
||||||
|
|
||||||
document.getElementById('bulk-keep').addEventListener('click', () => bulk('approve'));
|
document.getElementById('bulk-keep').addEventListener('click', () => bulk('approve'));
|
||||||
document.getElementById('bulk-reject').addEventListener('click', () => bulk('reject'));
|
document.getElementById('bulk-reject').addEventListener('click', () => bulk('reject'));
|
||||||
document.getElementById('bulk-reset').addEventListener('click', () => bulk('reset'));
|
document.getElementById('bulk-reset').addEventListener('click', () => bulk('reset'));
|
||||||
|
|||||||
85
process_staging.py
Normal file
85
process_staging.py
Normal file
@@ -0,0 +1,85 @@
|
|||||||
|
"""photo-pipeline: process-staging flow.
|
||||||
|
|
||||||
|
After review (dashboard = DB-only decisions), this flow materializes the
|
||||||
|
decisions: moves approved files to /mnt/data/01_keep, rejected to 03_delete.
|
||||||
|
|
||||||
|
Separating decisions (instant, in dashboard) from moves (here) keeps review
|
||||||
|
fast — cross-filesystem moves are the slow part and happen in one efficient pass.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import shutil
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from prefect import flow, task
|
||||||
|
|
||||||
|
import photo_db as db
|
||||||
|
|
||||||
|
STAGING = Path("/mnt/data")
|
||||||
|
STATUS_TO_DIR = {
|
||||||
|
"approved": "01_keep",
|
||||||
|
"rejected": "03_delete",
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@task
|
||||||
|
def materialize_decisions(status: str = None) -> dict:
|
||||||
|
"""Move files according to their DB status. Idempotent (skips already-moved)."""
|
||||||
|
db.init_db()
|
||||||
|
conn = db.get_db()
|
||||||
|
if status:
|
||||||
|
rows = conn.execute(
|
||||||
|
"SELECT sha256, path, status FROM image_hashes WHERE status=?", (status,)
|
||||||
|
).fetchall()
|
||||||
|
else:
|
||||||
|
rows = conn.execute(
|
||||||
|
"SELECT sha256, path, status FROM image_hashes WHERE status IN ('approved','rejected')"
|
||||||
|
).fetchall()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
moved = 0
|
||||||
|
already = 0
|
||||||
|
errors = []
|
||||||
|
for sha, path, st in rows:
|
||||||
|
target_dir = STATUS_TO_DIR.get(st)
|
||||||
|
if not target_dir:
|
||||||
|
continue
|
||||||
|
src = Path(path)
|
||||||
|
dest_dir = STAGING / target_dir
|
||||||
|
dest_dir.mkdir(parents=True, exist_ok=True)
|
||||||
|
dest = dest_dir / src.name
|
||||||
|
|
||||||
|
# already in the right place?
|
||||||
|
if src.parent == dest_dir:
|
||||||
|
already += 1
|
||||||
|
continue
|
||||||
|
if not src.exists():
|
||||||
|
errors.append(f"{path}: missing source")
|
||||||
|
continue
|
||||||
|
# avoid collision: append suffix if dest exists
|
||||||
|
if dest.exists():
|
||||||
|
dest = dest_dir / f"{src.stem}_{sha[:8]}{src.suffix}"
|
||||||
|
try:
|
||||||
|
shutil.move(str(src), str(dest))
|
||||||
|
conn = db.get_db()
|
||||||
|
conn.execute(
|
||||||
|
"UPDATE image_hashes SET path=?, status=? WHERE sha256=?", (str(dest), st, sha)
|
||||||
|
)
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
moved += 1
|
||||||
|
except Exception as e:
|
||||||
|
errors.append(f"{path}: {e}")
|
||||||
|
|
||||||
|
return {"moved": moved, "already": already, "errors": errors}
|
||||||
|
|
||||||
|
|
||||||
|
@flow(name="process-staging")
|
||||||
|
def process_staging(status: str = None):
|
||||||
|
"""Materialize review decisions: move approved/rejected files to staging."""
|
||||||
|
result = materialize_decisions(status)
|
||||||
|
print(f"process-staging: {result}")
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
process_staging()
|
||||||
Reference in New Issue
Block a user