feat(fc2c-i): artist overview + paged images endpoints

This commit is contained in:
2026-05-15 15:52:02 -04:00
parent 8484cb9aaa
commit e43c2a0dd0
5 changed files with 352 additions and 0 deletions
+2
View File
@@ -16,6 +16,7 @@ api_bp.add_url_rule("/health", view_func=health.get_health, methods=["GET"])
def all_blueprints() -> list[Blueprint]:
from .aliases import aliases_bp
from .allowlist import allowlist_bp
from .artist import artist_bp
from .gallery import gallery_bp
from .import_admin import import_admin_bp
from .ml_admin import ml_admin_bp
@@ -27,6 +28,7 @@ def all_blueprints() -> list[Blueprint]:
api_bp,
gallery_bp,
tags_bp,
artist_bp,
showcase_bp,
settings_bp,
import_admin_bp,
+36
View File
@@ -0,0 +1,36 @@
"""Artist API: overview aggregates + paged images."""
from quart import Blueprint, jsonify, request
from ..extensions import get_session
from ..services.artist_service import ArtistService
artist_bp = Blueprint("artist", __name__, url_prefix="/api/artist")
@artist_bp.route("/<slug>", methods=["GET"])
async def overview(slug: str):
async with get_session() as session:
svc = ArtistService(session)
data = await svc.overview(slug)
if data is None:
return jsonify({"error": "artist not found"}), 404
return jsonify(data)
@artist_bp.route("/<slug>/images", methods=["GET"])
async def images(slug: str):
cursor = request.args.get("cursor") or None
try:
limit = int(request.args.get("limit", "60"))
except ValueError:
return jsonify({"error": "limit must be an integer"}), 400
async with get_session() as session:
svc = ArtistService(session)
try:
page = await svc.images(slug, cursor=cursor, limit=limit)
except ValueError as exc:
return jsonify({"error": str(exc)}), 400
if page is None:
return jsonify({"error": "artist not found"}), 404
return jsonify({"images": page.images, "next_cursor": page.next_cursor})
+199
View File
@@ -0,0 +1,199 @@
"""Artist overview + paged images.
Images are linked to an artist through the provenance chain
Source(artist_id) -> ImageProvenance(source_id) -> ImageRecord. DISTINCT
guards against an image having several provenance rows for one artist.
Dates come from Post.post_date via ImageProvenance.post_id.
"""
from dataclasses import dataclass
from sqlalchemy import and_, func, or_, select
from sqlalchemy.ext.asyncio import AsyncSession
from ..models import (
Artist,
ImageProvenance,
ImageRecord,
Post,
Source,
Tag,
)
from ..models.tag import image_tag
from .gallery_service import decode_cursor, encode_cursor, thumbnail_url
@dataclass(frozen=True)
class ArtistImagesPage:
images: list[dict]
next_cursor: str | None
class ArtistService:
def __init__(self, session: AsyncSession):
self.session = session
async def _artist_by_slug(self, slug: str) -> Artist | None:
return (
await self.session.execute(
select(Artist).where(Artist.slug == slug)
)
).scalar_one_or_none()
async def overview(self, slug: str) -> dict | None:
artist = await self._artist_by_slug(slug)
if artist is None:
return None
aid = artist.id
img_join = (
select(ImageRecord.id)
.join(ImageProvenance, ImageProvenance.image_record_id == ImageRecord.id)
.join(Source, Source.id == ImageProvenance.source_id)
.where(Source.artist_id == aid)
)
image_count = (
await self.session.execute(
select(func.count(func.distinct(ImageRecord.id)))
.select_from(ImageRecord)
.join(ImageProvenance, ImageProvenance.image_record_id == ImageRecord.id)
.join(Source, Source.id == ImageProvenance.source_id)
.where(Source.artist_id == aid)
)
).scalar_one()
date_row = (
await self.session.execute(
select(func.min(Post.post_date), func.max(Post.post_date))
.select_from(Post)
.join(ImageProvenance, ImageProvenance.post_id == Post.id)
.join(Source, Source.id == ImageProvenance.source_id)
.where(Source.artist_id == aid)
)
).first()
dmin, dmax = date_row if date_row else (None, None)
cooccurring = (
await self.session.execute(
select(
Tag.id, Tag.name, Tag.kind,
func.count(image_tag.c.image_record_id).label("cnt"),
)
.select_from(Tag)
.join(image_tag, image_tag.c.tag_id == Tag.id)
.where(image_tag.c.image_record_id.in_(img_join))
.group_by(Tag.id, Tag.name, Tag.kind)
.order_by(func.count(image_tag.c.image_record_id).desc())
.limit(20)
)
).all()
sources = (
await self.session.execute(
select(
Source.id, Source.platform, Source.url,
func.count(func.distinct(ImageProvenance.image_record_id)).label("cnt"),
)
.select_from(Source)
.outerjoin(ImageProvenance, ImageProvenance.source_id == Source.id)
.where(Source.artist_id == aid)
.group_by(Source.id, Source.platform, Source.url)
.order_by(Source.id)
)
).all()
month = func.date_trunc("month", Post.post_date).label("m")
activity = (
await self.session.execute(
select(month, func.count(func.distinct(ImageProvenance.image_record_id)))
.select_from(Post)
.join(ImageProvenance, ImageProvenance.post_id == Post.id)
.join(Source, Source.id == ImageProvenance.source_id)
.where(and_(Source.artist_id == aid, Post.post_date.isnot(None)))
.group_by(month)
.order_by(month)
)
).all()
return {
"id": artist.id,
"name": artist.name,
"slug": artist.slug,
"image_count": int(image_count),
"date_range": {
"min": dmin.isoformat() if dmin else None,
"max": dmax.isoformat() if dmax else None,
},
"cooccurring_tags": [
{
"id": tid,
"name": name,
"kind": kind.value if hasattr(kind, "value") else kind,
"count": int(cnt),
}
for tid, name, kind, cnt in cooccurring
],
"sources": [
{
"id": sid,
"platform": platform,
"url": url,
"image_count": int(cnt),
}
for sid, platform, url, cnt in sources
],
"activity": [
{"month": m.isoformat(), "count": int(cnt)}
for m, cnt in activity
],
}
async def images(
self, slug: str, cursor: str | None, limit: int = 60
) -> ArtistImagesPage | None:
if limit < 1 or limit > 200:
raise ValueError("limit must be between 1 and 200")
artist = await self._artist_by_slug(slug)
if artist is None:
return None
stmt = (
select(ImageRecord)
.join(ImageProvenance, ImageProvenance.image_record_id == ImageRecord.id)
.join(Source, Source.id == ImageProvenance.source_id)
.where(Source.artist_id == artist.id)
.distinct()
)
if cursor:
cur_ts, cur_id = decode_cursor(cursor)
stmt = stmt.where(
or_(
ImageRecord.created_at < cur_ts,
and_(ImageRecord.created_at == cur_ts, ImageRecord.id < cur_id),
)
)
stmt = stmt.order_by(
ImageRecord.created_at.desc(), ImageRecord.id.desc()
).limit(limit + 1)
rows = (await self.session.execute(stmt)).scalars().all()
next_cursor = None
if len(rows) > limit:
last = rows[limit - 1]
next_cursor = encode_cursor(last.created_at, last.id)
rows = rows[:limit]
return ArtistImagesPage(
images=[
{
"id": r.id,
"sha256": r.sha256,
"mime": r.mime,
"width": r.width,
"height": r.height,
"thumbnail_url": thumbnail_url(r.sha256, r.mime),
}
for r in rows
],
next_cursor=next_cursor,
)
+37
View File
@@ -0,0 +1,37 @@
import pytest
from backend.app import create_app
from backend.app.models import Artist
pytestmark = pytest.mark.integration
@pytest.fixture
async def client():
app = create_app()
async with app.test_client() as c:
yield c
@pytest.mark.asyncio
async def test_artist_404(client):
resp = await client.get("/api/artist/nope")
assert resp.status_code == 404
@pytest.mark.asyncio
async def test_artist_overview_ok(client, db):
db.add(Artist(name="Mira", slug="mira"))
await db.flush()
await db.commit()
resp = await client.get("/api/artist/mira")
assert resp.status_code == 200
body = await resp.get_json()
assert body["name"] == "Mira"
assert body["image_count"] == 0
@pytest.mark.asyncio
async def test_artist_images_404(client):
resp = await client.get("/api/artist/nope/images")
assert resp.status_code == 404
+78
View File
@@ -0,0 +1,78 @@
from datetime import UTC, datetime
import pytest
from backend.app.models import (
Artist, ImageProvenance, ImageRecord, Post, Source, Tag, TagKind,
)
from backend.app.models.tag import image_tag
from backend.app.services.artist_service import ArtistService
pytestmark = pytest.mark.integration
async def _fixture(db):
artist = Artist(name="Nadia", slug="nadia")
db.add(artist)
await db.flush()
src = Source(artist_id=artist.id, platform="web", url="http://x")
db.add(src)
await db.flush()
post = Post(
source_id=src.id, external_post_id="p1",
post_date=datetime(2026, 3, 1, tzinfo=UTC),
)
db.add(post)
await db.flush()
img = ImageRecord(
path="/images/a/1.jpg", sha256="a" + "0" * 63,
size_bytes=1, mime="image/jpeg", width=4, height=8,
origin="downloaded", integrity_status="unknown",
)
db.add(img)
await db.flush()
db.add(ImageProvenance(
image_record_id=img.id, post_id=post.id, source_id=src.id))
tag = Tag(name="forest", kind=TagKind.general)
db.add(tag)
await db.flush()
await db.execute(image_tag.insert().values(
image_record_id=img.id, tag_id=tag.id, source="manual"))
await db.flush()
return artist, src, img
@pytest.mark.asyncio
async def test_overview_aggregates(db):
artist, src, img = await _fixture(db)
svc = ArtistService(db)
ov = await svc.overview("nadia")
assert ov["name"] == "Nadia"
assert ov["image_count"] == 1
assert ov["date_range"]["min"].startswith("2026-03-01")
assert ov["date_range"]["max"].startswith("2026-03-01")
assert any(t["name"] == "forest" for t in ov["cooccurring_tags"])
assert ov["sources"][0]["image_count"] == 1
assert ov["activity"][0]["count"] == 1
@pytest.mark.asyncio
async def test_overview_unknown_slug_returns_none(db):
svc = ArtistService(db)
assert await svc.overview("ghost") is None
@pytest.mark.asyncio
async def test_paged_images(db):
artist, src, img = await _fixture(db)
svc = ArtistService(db)
page = await svc.images("nadia", cursor=None, limit=10)
assert page is not None
assert len(page.images) == 1
assert page.images[0]["thumbnail_url"].startswith("/images/thumbs/")
@pytest.mark.asyncio
async def test_paged_images_unknown_slug_none(db):
svc = ArtistService(db)
assert await svc.images("ghost", cursor=None, limit=10) is None