Compare commits
No commits in common. "main" and "chore/pre-commit-autofix" have entirely different histories.
main
...
chore/pre-
@ -31,7 +31,7 @@ docker build -t paper-dynasty-db . # Build image
|
|||||||
| **URL** | pddev.manticorum.com | pd.manticorum.com |
|
| **URL** | pddev.manticorum.com | pd.manticorum.com |
|
||||||
| **Host** | `ssh pd-database` | `ssh akamai` → `/root/container-data/paper-dynasty` |
|
| **Host** | `ssh pd-database` | `ssh akamai` → `/root/container-data/paper-dynasty` |
|
||||||
| **API container** | `dev_pd_database` | `pd_api` |
|
| **API container** | `dev_pd_database` | `pd_api` |
|
||||||
| **PostgreSQL** | `sba_postgres` / `paperdynasty_dev` / `sba_admin` | `pd_postgres` / `pd_master` |
|
| **PostgreSQL** | `pd_postgres` (port 5432) | `pd_postgres` |
|
||||||
| **Adminer** | port 8081 | — |
|
| **Adminer** | port 8081 | — |
|
||||||
| **API port** | 816 | 815 |
|
| **API port** | 816 | 815 |
|
||||||
| **Image** | `manticorum67/paper-dynasty-database` | `manticorum67/paper-dynasty-database` |
|
| **Image** | `manticorum67/paper-dynasty-database` | `manticorum67/paper-dynasty-database` |
|
||||||
|
|||||||
@ -2,15 +2,7 @@ import datetime
|
|||||||
import os.path
|
import os.path
|
||||||
|
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
from fastapi import (
|
from fastapi import APIRouter, Depends, HTTPException, Request, Response, Query
|
||||||
APIRouter,
|
|
||||||
BackgroundTasks,
|
|
||||||
Depends,
|
|
||||||
HTTPException,
|
|
||||||
Request,
|
|
||||||
Response,
|
|
||||||
Query,
|
|
||||||
)
|
|
||||||
from fastapi.responses import FileResponse
|
from fastapi.responses import FileResponse
|
||||||
from fastapi.templating import Jinja2Templates
|
from fastapi.templating import Jinja2Templates
|
||||||
from typing import Optional, List, Literal
|
from typing import Optional, List, Literal
|
||||||
@ -40,9 +32,7 @@ from ..db_engine import (
|
|||||||
)
|
)
|
||||||
from ..db_helpers import upsert_players
|
from ..db_helpers import upsert_players
|
||||||
from ..dependencies import oauth2_scheme, valid_token
|
from ..dependencies import oauth2_scheme, valid_token
|
||||||
from ..services.card_storage import backfill_variant_image_url
|
|
||||||
from ..services.refractor_boost import compute_variant_hash
|
from ..services.refractor_boost import compute_variant_hash
|
||||||
from ..services.apng_generator import apng_cache_path, generate_animated_card
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
# Persistent browser instance (WP-02)
|
# Persistent browser instance (WP-02)
|
||||||
@ -737,138 +727,11 @@ async def get_one_player(player_id: int, csv: Optional[bool] = False):
|
|||||||
return return_val
|
return return_val
|
||||||
|
|
||||||
|
|
||||||
@router.get("/{player_id}/{card_type}card/{d}/{variant}/animated")
|
|
||||||
async def get_animated_card(
|
|
||||||
request: Request,
|
|
||||||
player_id: int,
|
|
||||||
card_type: Literal["batting", "pitching"],
|
|
||||||
variant: int,
|
|
||||||
d: str,
|
|
||||||
tier: Optional[int] = Query(
|
|
||||||
None, ge=0, le=4, description="Override refractor tier for preview (dev only)"
|
|
||||||
),
|
|
||||||
):
|
|
||||||
try:
|
|
||||||
this_player = Player.get_by_id(player_id)
|
|
||||||
except DoesNotExist:
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=404, detail=f"No player found with id {player_id}"
|
|
||||||
)
|
|
||||||
|
|
||||||
refractor_tier = (
|
|
||||||
tier if tier is not None else resolve_refractor_tier(player_id, variant)
|
|
||||||
)
|
|
||||||
if refractor_tier < 3:
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=404,
|
|
||||||
detail=f"No animation for tier {refractor_tier}; animated cards require T3 or T4",
|
|
||||||
)
|
|
||||||
|
|
||||||
cache_path = apng_cache_path(
|
|
||||||
this_player.cardset.id, card_type, player_id, d, variant
|
|
||||||
)
|
|
||||||
headers = {"Cache-Control": "public, max-age=86400"}
|
|
||||||
|
|
||||||
if os.path.isfile(cache_path) and tier is None:
|
|
||||||
return FileResponse(path=cache_path, media_type="image/png", headers=headers)
|
|
||||||
|
|
||||||
all_pos = (
|
|
||||||
CardPosition.select()
|
|
||||||
.where(CardPosition.player == this_player)
|
|
||||||
.order_by(CardPosition.innings.desc())
|
|
||||||
)
|
|
||||||
|
|
||||||
if card_type == "batting":
|
|
||||||
this_bc = BattingCard.get_or_none(
|
|
||||||
BattingCard.player == this_player, BattingCard.variant == variant
|
|
||||||
)
|
|
||||||
if this_bc is None:
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=404,
|
|
||||||
detail=f"Batting card not found for id {player_id}, variant {variant}",
|
|
||||||
)
|
|
||||||
rating_vl = BattingCardRatings.get_or_none(
|
|
||||||
BattingCardRatings.battingcard == this_bc, BattingCardRatings.vs_hand == "L"
|
|
||||||
)
|
|
||||||
rating_vr = BattingCardRatings.get_or_none(
|
|
||||||
BattingCardRatings.battingcard == this_bc, BattingCardRatings.vs_hand == "R"
|
|
||||||
)
|
|
||||||
if None in [rating_vr, rating_vl]:
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=404,
|
|
||||||
detail=f"Ratings not found for batting card {this_bc.id}",
|
|
||||||
)
|
|
||||||
card_data = get_batter_card_data(
|
|
||||||
this_player, this_bc, rating_vl, rating_vr, all_pos
|
|
||||||
)
|
|
||||||
if (
|
|
||||||
this_player.description in this_player.cardset.name
|
|
||||||
and this_player.cardset.id not in [23]
|
|
||||||
):
|
|
||||||
card_data["cardset_name"] = this_player.cardset.name
|
|
||||||
else:
|
|
||||||
card_data["cardset_name"] = this_player.description
|
|
||||||
card_data["refractor_tier"] = refractor_tier
|
|
||||||
card_data["request"] = request
|
|
||||||
html_response = templates.TemplateResponse("player_card.html", card_data)
|
|
||||||
|
|
||||||
else:
|
|
||||||
this_pc = PitchingCard.get_or_none(
|
|
||||||
PitchingCard.player == this_player, PitchingCard.variant == variant
|
|
||||||
)
|
|
||||||
if this_pc is None:
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=404,
|
|
||||||
detail=f"Pitching card not found for id {player_id}, variant {variant}",
|
|
||||||
)
|
|
||||||
rating_vl = PitchingCardRatings.get_or_none(
|
|
||||||
PitchingCardRatings.pitchingcard == this_pc,
|
|
||||||
PitchingCardRatings.vs_hand == "L",
|
|
||||||
)
|
|
||||||
rating_vr = PitchingCardRatings.get_or_none(
|
|
||||||
PitchingCardRatings.pitchingcard == this_pc,
|
|
||||||
PitchingCardRatings.vs_hand == "R",
|
|
||||||
)
|
|
||||||
if None in [rating_vr, rating_vl]:
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=404,
|
|
||||||
detail=f"Ratings not found for pitching card {this_pc.id}",
|
|
||||||
)
|
|
||||||
card_data = get_pitcher_card_data(
|
|
||||||
this_player, this_pc, rating_vl, rating_vr, all_pos
|
|
||||||
)
|
|
||||||
if (
|
|
||||||
this_player.description in this_player.cardset.name
|
|
||||||
and this_player.cardset.id not in [23]
|
|
||||||
):
|
|
||||||
card_data["cardset_name"] = this_player.cardset.name
|
|
||||||
else:
|
|
||||||
card_data["cardset_name"] = this_player.description
|
|
||||||
card_data["refractor_tier"] = refractor_tier
|
|
||||||
card_data["request"] = request
|
|
||||||
html_response = templates.TemplateResponse("player_card.html", card_data)
|
|
||||||
|
|
||||||
browser = await get_browser()
|
|
||||||
page = await browser.new_page(viewport={"width": 1280, "height": 720})
|
|
||||||
try:
|
|
||||||
await generate_animated_card(
|
|
||||||
page,
|
|
||||||
html_response.body.decode("UTF-8"),
|
|
||||||
cache_path,
|
|
||||||
refractor_tier,
|
|
||||||
)
|
|
||||||
finally:
|
|
||||||
await page.close()
|
|
||||||
|
|
||||||
return FileResponse(path=cache_path, media_type="image/png", headers=headers)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/{player_id}/{card_type}card")
|
@router.get("/{player_id}/{card_type}card")
|
||||||
@router.get("/{player_id}/{card_type}card/{d}")
|
@router.get("/{player_id}/{card_type}card/{d}")
|
||||||
@router.get("/{player_id}/{card_type}card/{d}/{variant}")
|
@router.get("/{player_id}/{card_type}card/{d}/{variant}")
|
||||||
async def get_batter_card(
|
async def get_batter_card(
|
||||||
request: Request,
|
request: Request,
|
||||||
background_tasks: BackgroundTasks,
|
|
||||||
player_id: int,
|
player_id: int,
|
||||||
card_type: Literal["batting", "pitching"],
|
card_type: Literal["batting", "pitching"],
|
||||||
variant: int = 0,
|
variant: int = 0,
|
||||||
@ -1043,27 +906,6 @@ async def get_batter_card(
|
|||||||
# save_as=f'{player_id}-{d}-v{variant}.png'
|
# save_as=f'{player_id}-{d}-v{variant}.png'
|
||||||
# )
|
# )
|
||||||
|
|
||||||
# Schedule S3 upload for variant cards that don't have an image_url yet.
|
|
||||||
# Skip when tier is overridden (?tier= dev preview) — those renders don't
|
|
||||||
# correspond to real variant card rows.
|
|
||||||
if variant > 0 and tier is None:
|
|
||||||
CardModel = BattingCard if card_type == "batting" else PitchingCard
|
|
||||||
try:
|
|
||||||
card_row = CardModel.get(
|
|
||||||
(CardModel.player_id == player_id) & (CardModel.variant == variant)
|
|
||||||
)
|
|
||||||
if card_row.image_url is None:
|
|
||||||
background_tasks.add_task(
|
|
||||||
backfill_variant_image_url,
|
|
||||||
player_id=player_id,
|
|
||||||
variant=variant,
|
|
||||||
card_type=card_type,
|
|
||||||
cardset_id=this_player.cardset.id,
|
|
||||||
png_path=file_path,
|
|
||||||
)
|
|
||||||
except CardModel.DoesNotExist:
|
|
||||||
pass
|
|
||||||
|
|
||||||
return FileResponse(path=file_path, media_type="image/png", headers=headers)
|
return FileResponse(path=file_path, media_type="image/png", headers=headers)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@ -4,7 +4,7 @@ from fastapi import APIRouter, Depends, HTTPException, Query
|
|||||||
import logging
|
import logging
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
|
|
||||||
from ..db_engine import model_to_dict, BattingCard, PitchingCard
|
from ..db_engine import model_to_dict
|
||||||
from ..dependencies import oauth2_scheme, valid_token
|
from ..dependencies import oauth2_scheme, valid_token
|
||||||
from ..services.refractor_init import initialize_card_refractor, _determine_card_type
|
from ..services.refractor_init import initialize_card_refractor, _determine_card_type
|
||||||
|
|
||||||
@ -67,24 +67,6 @@ def _build_card_state_response(state, player_name=None) -> dict:
|
|||||||
if player_name is not None:
|
if player_name is not None:
|
||||||
result["player_name"] = player_name
|
result["player_name"] = player_name
|
||||||
|
|
||||||
# Resolve image_url from the variant card row
|
|
||||||
image_url = None
|
|
||||||
if state.variant and state.variant > 0:
|
|
||||||
card_type = (
|
|
||||||
state.track.card_type if hasattr(state, "track") and state.track else None
|
|
||||||
)
|
|
||||||
if card_type:
|
|
||||||
CardModel = BattingCard if card_type == "batter" else PitchingCard
|
|
||||||
try:
|
|
||||||
variant_card = CardModel.get(
|
|
||||||
(CardModel.player_id == state.player_id)
|
|
||||||
& (CardModel.variant == state.variant)
|
|
||||||
)
|
|
||||||
image_url = variant_card.image_url
|
|
||||||
except CardModel.DoesNotExist:
|
|
||||||
pass
|
|
||||||
result["image_url"] = image_url
|
|
||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
|
||||||
@ -229,7 +211,7 @@ async def list_card_states(
|
|||||||
if evaluated_only:
|
if evaluated_only:
|
||||||
query = query.where(RefractorCardState.last_evaluated_at.is_null(False))
|
query = query.where(RefractorCardState.last_evaluated_at.is_null(False))
|
||||||
|
|
||||||
total = query.count() or 0
|
total = query.count()
|
||||||
items = []
|
items = []
|
||||||
for state in query.offset(offset).limit(limit):
|
for state in query.offset(offset).limit(limit):
|
||||||
player_name = None
|
player_name = None
|
||||||
|
|||||||
@ -1,125 +0,0 @@
|
|||||||
"""
|
|
||||||
APNG animated card generation for T3 and T4 refractor tiers.
|
|
||||||
|
|
||||||
Captures animation frames by scrubbing CSS animations via Playwright — each
|
|
||||||
frame is rendered with a negative animation-delay that freezes the render at a
|
|
||||||
specific point in the animation cycle. The captured PNGs are then assembled
|
|
||||||
into a looping APNG using the apng library.
|
|
||||||
|
|
||||||
Cache / S3 path convention:
|
|
||||||
Local: storage/cards/cardset-{id}/{card_type}/{player_id}-{date}-v{variant}.apng
|
|
||||||
S3: cards/cardset-{id}/{card_type}/{player_id}-{date}-v{variant}.apng
|
|
||||||
"""
|
|
||||||
|
|
||||||
import os
|
|
||||||
import tempfile
|
|
||||||
|
|
||||||
from apng import APNG
|
|
||||||
from playwright.async_api import Page
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# Animation specs per tier
|
|
||||||
# Each entry: list of (css_selector, animation_duration_seconds) pairs that
|
|
||||||
# need to be scrubbed, plus the frame count and per-frame display time.
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
_T3_SPEC = {
|
|
||||||
"selectors_and_durations": [("#header::after", 2.5)],
|
|
||||||
"num_frames": 12,
|
|
||||||
"frame_delay_ms": 200,
|
|
||||||
}
|
|
||||||
|
|
||||||
_T4_SPEC = {
|
|
||||||
"selectors_and_durations": [
|
|
||||||
("#header::after", 6.0),
|
|
||||||
(".tier-diamond.diamond-glow", 2.0),
|
|
||||||
],
|
|
||||||
"num_frames": 24,
|
|
||||||
"frame_delay_ms": 250,
|
|
||||||
}
|
|
||||||
|
|
||||||
ANIM_SPECS = {3: _T3_SPEC, 4: _T4_SPEC}
|
|
||||||
|
|
||||||
|
|
||||||
def apng_cache_path(
|
|
||||||
cardset_id: int, card_type: str, player_id: int, d: str, variant: int
|
|
||||||
) -> str:
|
|
||||||
"""Return the local filesystem cache path for an animated card APNG."""
|
|
||||||
return f"storage/cards/cardset-{cardset_id}/{card_type}/{player_id}-{d}-v{variant}.apng"
|
|
||||||
|
|
||||||
|
|
||||||
async def generate_animated_card(
|
|
||||||
page: Page,
|
|
||||||
html_content: str,
|
|
||||||
output_path: str,
|
|
||||||
tier: int,
|
|
||||||
) -> None:
|
|
||||||
"""Generate an animated APNG for a T3 or T4 refractor card.
|
|
||||||
|
|
||||||
Scrubs each CSS animation by injecting an override <style> tag that sets
|
|
||||||
animation-play-state: running and a negative animation-delay, freezing the
|
|
||||||
render at evenly-spaced intervals across one animation cycle. The captured
|
|
||||||
frames are assembled into a looping APNG at output_path.
|
|
||||||
|
|
||||||
Args:
|
|
||||||
page: An open Playwright page (caller is responsible for lifecycle).
|
|
||||||
html_content: Rendered card HTML string (from TemplateResponse.body).
|
|
||||||
output_path: Destination path for the .apng file.
|
|
||||||
tier: Refractor tier — must be 3 or 4.
|
|
||||||
|
|
||||||
Raises:
|
|
||||||
ValueError: If tier is not 3 or 4.
|
|
||||||
"""
|
|
||||||
spec = ANIM_SPECS.get(tier)
|
|
||||||
if spec is None:
|
|
||||||
raise ValueError(
|
|
||||||
f"No animation spec for tier {tier}; animated cards are T3 and T4 only"
|
|
||||||
)
|
|
||||||
|
|
||||||
num_frames = spec["num_frames"]
|
|
||||||
frame_delay_ms = spec["frame_delay_ms"]
|
|
||||||
selectors_and_durations = spec["selectors_and_durations"]
|
|
||||||
|
|
||||||
frame_paths: list[str] = []
|
|
||||||
try:
|
|
||||||
for i in range(num_frames):
|
|
||||||
progress = i / num_frames # 0.0 .. (N-1)/N, seamless loop
|
|
||||||
await page.set_content(html_content)
|
|
||||||
|
|
||||||
# Inject override CSS: unpauses animation and seeks to frame offset
|
|
||||||
css_parts = []
|
|
||||||
for selector, duration in selectors_and_durations:
|
|
||||||
delay_s = -progress * duration
|
|
||||||
css_parts.append(
|
|
||||||
f"{selector} {{"
|
|
||||||
f" animation-play-state: running !important;"
|
|
||||||
f" animation-delay: {delay_s:.4f}s !important;"
|
|
||||||
f" }}"
|
|
||||||
)
|
|
||||||
await page.add_style_tag(content="\n".join(css_parts))
|
|
||||||
|
|
||||||
tmp = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
|
|
||||||
tmp.close()
|
|
||||||
await page.screenshot(
|
|
||||||
path=tmp.name,
|
|
||||||
type="png",
|
|
||||||
clip={"x": 0.0, "y": 0, "width": 1200, "height": 600},
|
|
||||||
)
|
|
||||||
frame_paths.append(tmp.name)
|
|
||||||
|
|
||||||
dir_path = os.path.dirname(output_path)
|
|
||||||
if dir_path:
|
|
||||||
os.makedirs(dir_path, exist_ok=True)
|
|
||||||
|
|
||||||
apng_obj = APNG()
|
|
||||||
for frame_path in frame_paths:
|
|
||||||
# delay/delay_den is the frame display time in seconds as a fraction
|
|
||||||
apng_obj.append_file(frame_path, delay=frame_delay_ms, delay_den=1000)
|
|
||||||
apng_obj.save(output_path)
|
|
||||||
|
|
||||||
finally:
|
|
||||||
for path in frame_paths:
|
|
||||||
try:
|
|
||||||
os.unlink(path)
|
|
||||||
except OSError:
|
|
||||||
pass
|
|
||||||
@ -1,198 +0,0 @@
|
|||||||
"""
|
|
||||||
card_storage.py — S3 upload utility for variant card images.
|
|
||||||
|
|
||||||
Public API
|
|
||||||
----------
|
|
||||||
get_s3_client()
|
|
||||||
Create and return a boto3 S3 client using ambient AWS credentials
|
|
||||||
(environment variables or instance profile).
|
|
||||||
|
|
||||||
build_s3_key(cardset_id, player_id, variant, card_type)
|
|
||||||
Construct the S3 object key for a variant card image.
|
|
||||||
|
|
||||||
build_s3_url(s3_key, render_date)
|
|
||||||
Return the full HTTPS S3 URL with a cache-busting date query param.
|
|
||||||
|
|
||||||
upload_card_to_s3(s3_client, png_bytes, s3_key)
|
|
||||||
Upload raw PNG bytes to S3 with correct ContentType and CacheControl headers.
|
|
||||||
|
|
||||||
backfill_variant_image_url(player_id, variant, card_type, cardset_id, png_path)
|
|
||||||
End-to-end: read PNG from disk, upload to S3, update BattingCard or
|
|
||||||
PitchingCard.image_url in the database. All exceptions are caught and
|
|
||||||
logged; this function never raises (safe to call as a background task).
|
|
||||||
|
|
||||||
Design notes
|
|
||||||
------------
|
|
||||||
- S3 credentials are resolved from the environment by boto3 at call time;
|
|
||||||
no credentials are hard-coded here.
|
|
||||||
- The cache-bust ?d= param matches the card-creation pipeline convention so
|
|
||||||
that clients can compare URLs across pipelines.
|
|
||||||
"""
|
|
||||||
|
|
||||||
import logging
|
|
||||||
import os
|
|
||||||
from datetime import date
|
|
||||||
|
|
||||||
import boto3
|
|
||||||
|
|
||||||
from app.db_engine import BattingCard, PitchingCard
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
S3_BUCKET = os.environ.get("S3_BUCKET", "paper-dynasty")
|
|
||||||
S3_REGION = os.environ.get("S3_REGION", "us-east-1")
|
|
||||||
|
|
||||||
|
|
||||||
def get_s3_client():
|
|
||||||
"""Create and return a boto3 S3 client for the configured region.
|
|
||||||
|
|
||||||
Credentials are resolved by boto3 from the standard chain:
|
|
||||||
environment variables → ~/.aws/credentials → instance profile.
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
A boto3 S3 client instance.
|
|
||||||
"""
|
|
||||||
return boto3.client("s3", region_name=S3_REGION)
|
|
||||||
|
|
||||||
|
|
||||||
def build_s3_key(cardset_id: int, player_id: int, variant: int, card_type: str) -> str:
|
|
||||||
"""Construct the S3 object key for a variant card image.
|
|
||||||
|
|
||||||
Key format:
|
|
||||||
cards/cardset-{csid:03d}/player-{pid}/v{variant}/{card_type}card.png
|
|
||||||
|
|
||||||
Args:
|
|
||||||
cardset_id: Numeric cardset ID (zero-padded to 3 digits).
|
|
||||||
player_id: Player ID.
|
|
||||||
variant: Variant number (0 = base, 1-4 = refractor tiers).
|
|
||||||
card_type: Either "batting" or "pitching".
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
The S3 object key string.
|
|
||||||
"""
|
|
||||||
return (
|
|
||||||
f"cards/cardset-{cardset_id:03d}/player-{player_id}"
|
|
||||||
f"/v{variant}/{card_type}card.png"
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def build_s3_url(s3_key: str, render_date: date) -> str:
|
|
||||||
"""Return the full HTTPS S3 URL for a card image with a cache-bust param.
|
|
||||||
|
|
||||||
URL format:
|
|
||||||
https://{bucket}.s3.{region}.amazonaws.com/{key}?d={date}
|
|
||||||
|
|
||||||
The ?d= query param matches the card-creation pipeline convention so that
|
|
||||||
clients invalidate their cache after each re-render.
|
|
||||||
|
|
||||||
Args:
|
|
||||||
s3_key: S3 object key (from build_s3_key).
|
|
||||||
render_date: The date the card was rendered, used for cache-busting.
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
Full HTTPS URL string.
|
|
||||||
"""
|
|
||||||
base_url = f"https://{S3_BUCKET}.s3.{S3_REGION}.amazonaws.com"
|
|
||||||
date_str = render_date.strftime("%Y-%m-%d")
|
|
||||||
return f"{base_url}/{s3_key}?d={date_str}"
|
|
||||||
|
|
||||||
|
|
||||||
def upload_card_to_s3(s3_client, png_bytes: bytes, s3_key: str) -> None:
|
|
||||||
"""Upload raw PNG bytes to S3 with the standard card image headers.
|
|
||||||
|
|
||||||
Sets ContentType=image/png and CacheControl=public, max-age=300 (5 min)
|
|
||||||
so that CDN and browser caches are refreshed within a short window after
|
|
||||||
a re-render.
|
|
||||||
|
|
||||||
Args:
|
|
||||||
s3_client: A boto3 S3 client (from get_s3_client).
|
|
||||||
png_bytes: Raw PNG image bytes.
|
|
||||||
s3_key: S3 object key (from build_s3_key).
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
None
|
|
||||||
"""
|
|
||||||
s3_client.put_object(
|
|
||||||
Bucket=S3_BUCKET,
|
|
||||||
Key=s3_key,
|
|
||||||
Body=png_bytes,
|
|
||||||
ContentType="image/png",
|
|
||||||
CacheControl="public, max-age=300",
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def backfill_variant_image_url(
|
|
||||||
player_id: int,
|
|
||||||
variant: int,
|
|
||||||
card_type: str,
|
|
||||||
cardset_id: int,
|
|
||||||
png_path: str,
|
|
||||||
) -> None:
|
|
||||||
"""Read a rendered PNG from disk, upload it to S3, and update the DB row.
|
|
||||||
|
|
||||||
Determines the correct card model (BattingCard or PitchingCard) from
|
|
||||||
card_type, then:
|
|
||||||
1. Reads PNG bytes from png_path.
|
|
||||||
2. Uploads to S3 via upload_card_to_s3.
|
|
||||||
3. Fetches the card row by (player_id, variant).
|
|
||||||
4. Sets image_url to the new S3 URL and calls save().
|
|
||||||
|
|
||||||
All exceptions are caught and logged — this function is intended to be
|
|
||||||
called as a background task and must never propagate exceptions.
|
|
||||||
|
|
||||||
Args:
|
|
||||||
player_id: Player ID used to locate the card row.
|
|
||||||
variant: Variant number (matches the card row's variant field).
|
|
||||||
card_type: "batting" or "pitching" — selects the model.
|
|
||||||
cardset_id: Cardset ID used for the S3 key.
|
|
||||||
png_path: Absolute path to the rendered PNG file on disk.
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
None
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
# 1. Read PNG from disk
|
|
||||||
with open(png_path, "rb") as f:
|
|
||||||
png_bytes = f.read()
|
|
||||||
|
|
||||||
# 2. Build key and upload
|
|
||||||
s3_key = build_s3_key(
|
|
||||||
cardset_id=cardset_id,
|
|
||||||
player_id=player_id,
|
|
||||||
variant=variant,
|
|
||||||
card_type=card_type,
|
|
||||||
)
|
|
||||||
s3_client = get_s3_client()
|
|
||||||
upload_card_to_s3(s3_client, png_bytes, s3_key)
|
|
||||||
|
|
||||||
# 3. Build URL with today's date for cache-busting
|
|
||||||
image_url = build_s3_url(s3_key, render_date=date.today())
|
|
||||||
|
|
||||||
# 4. Locate the card row and update image_url
|
|
||||||
if card_type == "batting":
|
|
||||||
card = BattingCard.get(
|
|
||||||
BattingCard.player_id == player_id, BattingCard.variant == variant
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
card = PitchingCard.get(
|
|
||||||
PitchingCard.player_id == player_id, PitchingCard.variant == variant
|
|
||||||
)
|
|
||||||
|
|
||||||
card.image_url = image_url
|
|
||||||
card.save()
|
|
||||||
|
|
||||||
logger.info(
|
|
||||||
"backfill_variant_image_url: updated %s card player=%s variant=%s url=%s",
|
|
||||||
card_type,
|
|
||||||
player_id,
|
|
||||||
variant,
|
|
||||||
image_url,
|
|
||||||
)
|
|
||||||
|
|
||||||
except Exception:
|
|
||||||
logger.exception(
|
|
||||||
"backfill_variant_image_url: failed for player=%s variant=%s card_type=%s",
|
|
||||||
player_id,
|
|
||||||
variant,
|
|
||||||
card_type,
|
|
||||||
)
|
|
||||||
@ -12,5 +12,3 @@ requests==2.32.3
|
|||||||
html2image==2.0.6
|
html2image==2.0.6
|
||||||
jinja2==3.1.4
|
jinja2==3.1.4
|
||||||
playwright==1.45.1
|
playwright==1.45.1
|
||||||
apng==3.1.1
|
|
||||||
boto3==1.42.65
|
|
||||||
|
|||||||
@ -1,350 +0,0 @@
|
|||||||
"""
|
|
||||||
Unit tests for app/services/card_storage.py — S3 upload utility.
|
|
||||||
|
|
||||||
This module covers:
|
|
||||||
- S3 key construction for variant cards (batting, pitching, zero-padded cardset)
|
|
||||||
- Full S3 URL construction with cache-busting date param
|
|
||||||
- put_object call validation (correct params, return value)
|
|
||||||
- End-to-end backfill: read PNG from disk, upload to S3, update DB row
|
|
||||||
|
|
||||||
Why we test S3 key construction separately:
|
|
||||||
The key format is a contract used by both the renderer and the URL builder.
|
|
||||||
Validating it in isolation catches regressions before they corrupt stored URLs.
|
|
||||||
|
|
||||||
Why we test URL construction separately:
|
|
||||||
The cache-bust param (?d=...) must be appended consistently so that clients
|
|
||||||
invalidate cached images after a re-render. Testing it independently prevents
|
|
||||||
the formatter from silently changing.
|
|
||||||
|
|
||||||
Why we test upload params:
|
|
||||||
ContentType and CacheControl must be set exactly so that S3 serves images
|
|
||||||
with the correct headers. A missing header is a silent misconfiguration.
|
|
||||||
|
|
||||||
Why we test backfill error swallowing:
|
|
||||||
The backfill function is called as a background task — it must never raise
|
|
||||||
exceptions that would abort a card render response. We verify that S3 failures
|
|
||||||
and missing files are both silently logged, not propagated.
|
|
||||||
|
|
||||||
Test isolation:
|
|
||||||
All tests use unittest.mock; no real S3 calls or DB connections are made.
|
|
||||||
The `backfill_variant_image_url` tests patch `get_s3_client` and the DB
|
|
||||||
model classes at the card_storage module level so lazy imports work correctly.
|
|
||||||
"""
|
|
||||||
|
|
||||||
import os
|
|
||||||
from datetime import date
|
|
||||||
from unittest.mock import MagicMock, patch
|
|
||||||
|
|
||||||
# Set env before importing module so db_engine doesn't try to connect
|
|
||||||
os.environ.setdefault("DATABASE_TYPE", "postgresql")
|
|
||||||
os.environ.setdefault("POSTGRES_PASSWORD", "test-dummy")
|
|
||||||
|
|
||||||
from app.services.card_storage import (
|
|
||||||
build_s3_key,
|
|
||||||
build_s3_url,
|
|
||||||
upload_card_to_s3,
|
|
||||||
backfill_variant_image_url,
|
|
||||||
S3_BUCKET,
|
|
||||||
S3_REGION,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# TestBuildS3Key
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestBuildS3Key:
|
|
||||||
"""Tests for build_s3_key — S3 object key construction.
|
|
||||||
|
|
||||||
The key format must match the existing card-creation pipeline so that
|
|
||||||
the database API and card-creation tool write to the same S3 paths.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def test_batting_card_key(self):
|
|
||||||
"""batting card type produces 'battingcard.png' in the key."""
|
|
||||||
key = build_s3_key(cardset_id=27, player_id=42, variant=1, card_type="batting")
|
|
||||||
assert key == "cards/cardset-027/player-42/v1/battingcard.png"
|
|
||||||
|
|
||||||
def test_pitching_card_key(self):
|
|
||||||
"""pitching card type produces 'pitchingcard.png' in the key."""
|
|
||||||
key = build_s3_key(cardset_id=27, player_id=99, variant=2, card_type="pitching")
|
|
||||||
assert key == "cards/cardset-027/player-99/v2/pitchingcard.png"
|
|
||||||
|
|
||||||
def test_cardset_zero_padded_to_three_digits(self):
|
|
||||||
"""Single-digit cardset IDs are zero-padded to three characters."""
|
|
||||||
key = build_s3_key(cardset_id=5, player_id=1, variant=0, card_type="batting")
|
|
||||||
assert "cardset-005" in key
|
|
||||||
|
|
||||||
def test_cardset_two_digit_zero_padded(self):
|
|
||||||
"""Two-digit cardset IDs are zero-padded correctly."""
|
|
||||||
key = build_s3_key(cardset_id=27, player_id=1, variant=0, card_type="batting")
|
|
||||||
assert "cardset-027" in key
|
|
||||||
|
|
||||||
def test_cardset_three_digit_no_padding(self):
|
|
||||||
"""Three-digit cardset IDs are not altered."""
|
|
||||||
key = build_s3_key(cardset_id=100, player_id=1, variant=0, card_type="batting")
|
|
||||||
assert "cardset-100" in key
|
|
||||||
|
|
||||||
def test_variant_included_in_key(self):
|
|
||||||
"""Variant number is included in the path so variants have distinct keys."""
|
|
||||||
key_v0 = build_s3_key(
|
|
||||||
cardset_id=27, player_id=1, variant=0, card_type="batting"
|
|
||||||
)
|
|
||||||
key_v3 = build_s3_key(
|
|
||||||
cardset_id=27, player_id=1, variant=3, card_type="batting"
|
|
||||||
)
|
|
||||||
assert "/v0/" in key_v0
|
|
||||||
assert "/v3/" in key_v3
|
|
||||||
assert key_v0 != key_v3
|
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# TestBuildS3Url
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestBuildS3Url:
|
|
||||||
"""Tests for build_s3_url — full URL construction with cache-bust param.
|
|
||||||
|
|
||||||
The URL format must be predictable so clients can construct and verify
|
|
||||||
image URLs without querying the database.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def test_url_contains_bucket_and_region(self):
|
|
||||||
"""URL includes bucket name and region in the S3 hostname."""
|
|
||||||
key = "cards/cardset-027/player-42/v1/battingcard.png"
|
|
||||||
render_date = date(2026, 4, 6)
|
|
||||||
url = build_s3_url(key, render_date)
|
|
||||||
assert S3_BUCKET in url
|
|
||||||
assert S3_REGION in url
|
|
||||||
|
|
||||||
def test_url_contains_s3_key(self):
|
|
||||||
"""URL path includes the full S3 key."""
|
|
||||||
key = "cards/cardset-027/player-42/v1/battingcard.png"
|
|
||||||
render_date = date(2026, 4, 6)
|
|
||||||
url = build_s3_url(key, render_date)
|
|
||||||
assert key in url
|
|
||||||
|
|
||||||
def test_url_has_cache_bust_param(self):
|
|
||||||
"""URL ends with ?d=<render_date> for cache invalidation."""
|
|
||||||
key = "cards/cardset-027/player-42/v1/battingcard.png"
|
|
||||||
render_date = date(2026, 4, 6)
|
|
||||||
url = build_s3_url(key, render_date)
|
|
||||||
assert "?d=2026-04-06" in url
|
|
||||||
|
|
||||||
def test_url_format_full(self):
|
|
||||||
"""Full URL matches expected S3 pattern exactly."""
|
|
||||||
key = "cards/cardset-027/player-1/v0/battingcard.png"
|
|
||||||
render_date = date(2025, 11, 8)
|
|
||||||
url = build_s3_url(key, render_date)
|
|
||||||
expected = (
|
|
||||||
f"https://{S3_BUCKET}.s3.{S3_REGION}.amazonaws.com/{key}?d=2025-11-08"
|
|
||||||
)
|
|
||||||
assert url == expected
|
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# TestUploadCardToS3
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestUploadCardToS3:
|
|
||||||
"""Tests for upload_card_to_s3 — S3 put_object call validation.
|
|
||||||
|
|
||||||
We verify the exact parameters passed to put_object so that S3 serves
|
|
||||||
images with the correct Content-Type and Cache-Control headers.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def test_put_object_called_with_correct_params(self):
|
|
||||||
"""put_object is called once with bucket, key, body, ContentType, CacheControl."""
|
|
||||||
mock_client = MagicMock()
|
|
||||||
png_bytes = b"\x89PNG\r\n\x1a\n"
|
|
||||||
s3_key = "cards/cardset-027/player-42/v1/battingcard.png"
|
|
||||||
|
|
||||||
upload_card_to_s3(mock_client, png_bytes, s3_key)
|
|
||||||
|
|
||||||
mock_client.put_object.assert_called_once_with(
|
|
||||||
Bucket=S3_BUCKET,
|
|
||||||
Key=s3_key,
|
|
||||||
Body=png_bytes,
|
|
||||||
ContentType="image/png",
|
|
||||||
CacheControl="public, max-age=300",
|
|
||||||
)
|
|
||||||
|
|
||||||
def test_upload_returns_none(self):
|
|
||||||
"""upload_card_to_s3 returns None (callers should not rely on a return value)."""
|
|
||||||
mock_client = MagicMock()
|
|
||||||
result = upload_card_to_s3(mock_client, b"PNG", "some/key.png")
|
|
||||||
assert result is None
|
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# TestBackfillVariantImageUrl
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
class TestBackfillVariantImageUrl:
|
|
||||||
"""Tests for backfill_variant_image_url — end-to-end disk→S3→DB path.
|
|
||||||
|
|
||||||
The function is fire-and-forget: it reads a PNG from disk, uploads to S3,
|
|
||||||
then updates the appropriate card model's image_url. All errors are caught
|
|
||||||
and logged; the function must never raise.
|
|
||||||
|
|
||||||
Test strategy:
|
|
||||||
- Use tmp_path for temporary PNG files so no filesystem state leaks.
|
|
||||||
- Patch get_s3_client at the module level to intercept the S3 call.
|
|
||||||
- Patch BattingCard/PitchingCard at the module level (lazy import target).
|
|
||||||
"""
|
|
||||||
|
|
||||||
def test_batting_card_image_url_updated(self, tmp_path):
|
|
||||||
"""BattingCard.image_url is updated after a successful upload."""
|
|
||||||
png_path = tmp_path / "card.png"
|
|
||||||
png_path.write_bytes(b"\x89PNG\r\n\x1a\n fake png data")
|
|
||||||
|
|
||||||
mock_s3 = MagicMock()
|
|
||||||
mock_card = MagicMock()
|
|
||||||
|
|
||||||
with (
|
|
||||||
patch("app.services.card_storage.get_s3_client", return_value=mock_s3),
|
|
||||||
patch("app.services.card_storage.BattingCard") as MockBatting,
|
|
||||||
):
|
|
||||||
MockBatting.get.return_value = mock_card
|
|
||||||
|
|
||||||
backfill_variant_image_url(
|
|
||||||
player_id=42,
|
|
||||||
variant=1,
|
|
||||||
card_type="batting",
|
|
||||||
cardset_id=27,
|
|
||||||
png_path=str(png_path),
|
|
||||||
)
|
|
||||||
|
|
||||||
MockBatting.get.assert_called_once_with(
|
|
||||||
MockBatting.player_id == 42, MockBatting.variant == 1
|
|
||||||
)
|
|
||||||
assert mock_card.image_url is not None
|
|
||||||
mock_card.save.assert_called_once()
|
|
||||||
|
|
||||||
def test_pitching_card_image_url_updated(self, tmp_path):
|
|
||||||
"""PitchingCard.image_url is updated after a successful upload."""
|
|
||||||
png_path = tmp_path / "card.png"
|
|
||||||
png_path.write_bytes(b"\x89PNG\r\n\x1a\n fake png data")
|
|
||||||
|
|
||||||
mock_s3 = MagicMock()
|
|
||||||
mock_card = MagicMock()
|
|
||||||
|
|
||||||
with (
|
|
||||||
patch("app.services.card_storage.get_s3_client", return_value=mock_s3),
|
|
||||||
patch("app.services.card_storage.PitchingCard") as MockPitching,
|
|
||||||
):
|
|
||||||
MockPitching.get.return_value = mock_card
|
|
||||||
|
|
||||||
backfill_variant_image_url(
|
|
||||||
player_id=99,
|
|
||||||
variant=2,
|
|
||||||
card_type="pitching",
|
|
||||||
cardset_id=27,
|
|
||||||
png_path=str(png_path),
|
|
||||||
)
|
|
||||||
|
|
||||||
MockPitching.get.assert_called_once_with(
|
|
||||||
MockPitching.player_id == 99, MockPitching.variant == 2
|
|
||||||
)
|
|
||||||
assert mock_card.image_url is not None
|
|
||||||
mock_card.save.assert_called_once()
|
|
||||||
|
|
||||||
def test_s3_upload_called_with_png_bytes(self, tmp_path):
|
|
||||||
"""The PNG bytes read from disk are passed to put_object."""
|
|
||||||
png_bytes = b"\x89PNG\r\n\x1a\n real png content"
|
|
||||||
png_path = tmp_path / "card.png"
|
|
||||||
png_path.write_bytes(png_bytes)
|
|
||||||
|
|
||||||
mock_s3 = MagicMock()
|
|
||||||
|
|
||||||
with (
|
|
||||||
patch("app.services.card_storage.get_s3_client", return_value=mock_s3),
|
|
||||||
patch("app.services.card_storage.BattingCard") as MockBatting,
|
|
||||||
):
|
|
||||||
MockBatting.get.return_value = MagicMock()
|
|
||||||
|
|
||||||
backfill_variant_image_url(
|
|
||||||
player_id=1,
|
|
||||||
variant=0,
|
|
||||||
card_type="batting",
|
|
||||||
cardset_id=5,
|
|
||||||
png_path=str(png_path),
|
|
||||||
)
|
|
||||||
|
|
||||||
mock_s3.put_object.assert_called_once()
|
|
||||||
call_kwargs = mock_s3.put_object.call_args.kwargs
|
|
||||||
assert call_kwargs["Body"] == png_bytes
|
|
||||||
|
|
||||||
def test_s3_error_is_swallowed(self, tmp_path):
|
|
||||||
"""If S3 raises an exception, backfill swallows it and returns normally.
|
|
||||||
|
|
||||||
The function is called as a background task — it must never propagate
|
|
||||||
exceptions that would abort the calling request handler.
|
|
||||||
"""
|
|
||||||
png_path = tmp_path / "card.png"
|
|
||||||
png_path.write_bytes(b"PNG data")
|
|
||||||
|
|
||||||
mock_s3 = MagicMock()
|
|
||||||
mock_s3.put_object.side_effect = Exception("S3 connection refused")
|
|
||||||
|
|
||||||
with (
|
|
||||||
patch("app.services.card_storage.get_s3_client", return_value=mock_s3),
|
|
||||||
patch("app.services.card_storage.BattingCard"),
|
|
||||||
):
|
|
||||||
# Must not raise
|
|
||||||
backfill_variant_image_url(
|
|
||||||
player_id=1,
|
|
||||||
variant=0,
|
|
||||||
card_type="batting",
|
|
||||||
cardset_id=27,
|
|
||||||
png_path=str(png_path),
|
|
||||||
)
|
|
||||||
|
|
||||||
def test_missing_file_is_swallowed(self, tmp_path):
|
|
||||||
"""If the PNG file does not exist, backfill swallows the error and returns.
|
|
||||||
|
|
||||||
Render failures may leave no file on disk; the background task must
|
|
||||||
handle this gracefully rather than crashing the request.
|
|
||||||
"""
|
|
||||||
missing_path = str(tmp_path / "nonexistent.png")
|
|
||||||
|
|
||||||
with (
|
|
||||||
patch("app.services.card_storage.get_s3_client"),
|
|
||||||
patch("app.services.card_storage.BattingCard"),
|
|
||||||
):
|
|
||||||
# Must not raise
|
|
||||||
backfill_variant_image_url(
|
|
||||||
player_id=1,
|
|
||||||
variant=0,
|
|
||||||
card_type="batting",
|
|
||||||
cardset_id=27,
|
|
||||||
png_path=missing_path,
|
|
||||||
)
|
|
||||||
|
|
||||||
def test_db_error_is_swallowed(self, tmp_path):
|
|
||||||
"""If the DB save raises, backfill swallows it and returns normally."""
|
|
||||||
png_path = tmp_path / "card.png"
|
|
||||||
png_path.write_bytes(b"PNG data")
|
|
||||||
|
|
||||||
mock_s3 = MagicMock()
|
|
||||||
mock_card = MagicMock()
|
|
||||||
mock_card.save.side_effect = Exception("DB connection lost")
|
|
||||||
|
|
||||||
with (
|
|
||||||
patch("app.services.card_storage.get_s3_client", return_value=mock_s3),
|
|
||||||
patch("app.services.card_storage.BattingCard") as MockBatting,
|
|
||||||
):
|
|
||||||
MockBatting.get.return_value = mock_card
|
|
||||||
|
|
||||||
# Must not raise
|
|
||||||
backfill_variant_image_url(
|
|
||||||
player_id=1,
|
|
||||||
variant=0,
|
|
||||||
card_type="batting",
|
|
||||||
cardset_id=27,
|
|
||||||
png_path=str(png_path),
|
|
||||||
)
|
|
||||||
@ -1,311 +0,0 @@
|
|||||||
"""Tests for image_url field in refractor cards API response.
|
|
||||||
|
|
||||||
What: Verifies that GET /api/v2/refractor/cards includes image_url in each card state
|
|
||||||
item, pulling the URL from the variant BattingCard or PitchingCard row.
|
|
||||||
|
|
||||||
Why: The refractor card art pipeline stores rendered card image URLs in the
|
|
||||||
BattingCard/PitchingCard rows. The Discord bot and website need image_url in
|
|
||||||
the /refractor/cards response so they can display variant art without a separate
|
|
||||||
lookup. These tests guard against regressions where image_url is accidentally
|
|
||||||
dropped from the response serialization.
|
|
||||||
|
|
||||||
Test cases:
|
|
||||||
test_cards_response_includes_image_url -- BattingCard with image_url set; verify
|
|
||||||
the value appears in the /cards response.
|
|
||||||
test_cards_response_image_url_null_when_not_set -- BattingCard with image_url=None;
|
|
||||||
verify null is returned (not omitted).
|
|
||||||
|
|
||||||
Uses the shared-memory SQLite TestClient pattern from test_refractor_state_api.py
|
|
||||||
so no PostgreSQL connection is required.
|
|
||||||
"""
|
|
||||||
|
|
||||||
import os
|
|
||||||
|
|
||||||
os.environ.setdefault("API_TOKEN", "test")
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
from fastapi import FastAPI, Request
|
|
||||||
from fastapi.testclient import TestClient
|
|
||||||
from peewee import SqliteDatabase
|
|
||||||
|
|
||||||
from app.db_engine import (
|
|
||||||
BattingCard,
|
|
||||||
BattingSeasonStats,
|
|
||||||
Card,
|
|
||||||
Cardset,
|
|
||||||
Decision,
|
|
||||||
Event,
|
|
||||||
MlbPlayer,
|
|
||||||
Pack,
|
|
||||||
PackType,
|
|
||||||
PitchingCard,
|
|
||||||
PitchingSeasonStats,
|
|
||||||
Player,
|
|
||||||
ProcessedGame,
|
|
||||||
Rarity,
|
|
||||||
RefractorCardState,
|
|
||||||
RefractorCosmetic,
|
|
||||||
RefractorTierBoost,
|
|
||||||
RefractorTrack,
|
|
||||||
Roster,
|
|
||||||
RosterSlot,
|
|
||||||
ScoutClaim,
|
|
||||||
ScoutOpportunity,
|
|
||||||
StratGame,
|
|
||||||
StratPlay,
|
|
||||||
Team,
|
|
||||||
)
|
|
||||||
|
|
||||||
AUTH_HEADER = {"Authorization": "Bearer test"}
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# SQLite database + model list
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
_img_url_db = SqliteDatabase(
|
|
||||||
"file:imgurlapitest?mode=memory&cache=shared",
|
|
||||||
uri=True,
|
|
||||||
pragmas={"foreign_keys": 1},
|
|
||||||
)
|
|
||||||
|
|
||||||
# Full model list matching the existing state API tests — needed so all FK
|
|
||||||
# constraints resolve in SQLite.
|
|
||||||
_IMG_URL_MODELS = [
|
|
||||||
Rarity,
|
|
||||||
Event,
|
|
||||||
Cardset,
|
|
||||||
MlbPlayer,
|
|
||||||
Player,
|
|
||||||
BattingCard,
|
|
||||||
PitchingCard,
|
|
||||||
Team,
|
|
||||||
PackType,
|
|
||||||
Pack,
|
|
||||||
Card,
|
|
||||||
Roster,
|
|
||||||
RosterSlot,
|
|
||||||
StratGame,
|
|
||||||
StratPlay,
|
|
||||||
Decision,
|
|
||||||
ScoutOpportunity,
|
|
||||||
ScoutClaim,
|
|
||||||
BattingSeasonStats,
|
|
||||||
PitchingSeasonStats,
|
|
||||||
ProcessedGame,
|
|
||||||
RefractorTrack,
|
|
||||||
RefractorCardState,
|
|
||||||
RefractorTierBoost,
|
|
||||||
RefractorCosmetic,
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(autouse=False)
|
|
||||||
def setup_img_url_db():
|
|
||||||
"""Bind image-url test models to shared-memory SQLite and create tables.
|
|
||||||
|
|
||||||
What: Initialises the in-process SQLite database before each test and drops
|
|
||||||
all tables afterwards to ensure test isolation.
|
|
||||||
|
|
||||||
Why: SQLite shared-memory databases persist between tests in the same
|
|
||||||
process unless tables are dropped. Creating and dropping around each test
|
|
||||||
guarantees a clean state without requiring a real PostgreSQL instance.
|
|
||||||
"""
|
|
||||||
_img_url_db.bind(_IMG_URL_MODELS)
|
|
||||||
_img_url_db.connect(reuse_if_open=True)
|
|
||||||
_img_url_db.create_tables(_IMG_URL_MODELS)
|
|
||||||
yield _img_url_db
|
|
||||||
_img_url_db.drop_tables(list(reversed(_IMG_URL_MODELS)), safe=True)
|
|
||||||
|
|
||||||
|
|
||||||
def _build_image_url_app() -> FastAPI:
|
|
||||||
"""Minimal FastAPI app with refractor router for image_url tests."""
|
|
||||||
from app.routers_v2.refractor import router as refractor_router
|
|
||||||
|
|
||||||
app = FastAPI()
|
|
||||||
|
|
||||||
@app.middleware("http")
|
|
||||||
async def db_middleware(request: Request, call_next):
|
|
||||||
_img_url_db.connect(reuse_if_open=True)
|
|
||||||
return await call_next(request)
|
|
||||||
|
|
||||||
app.include_router(refractor_router)
|
|
||||||
return app
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
|
||||||
def img_url_client(setup_img_url_db):
|
|
||||||
"""FastAPI TestClient backed by shared-memory SQLite for image_url tests."""
|
|
||||||
with TestClient(_build_image_url_app()) as c:
|
|
||||||
yield c
|
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# Seed helpers
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
def _make_rarity():
|
|
||||||
r, _ = Rarity.get_or_create(
|
|
||||||
value=10, name="IU_Common", defaults={"color": "#ffffff"}
|
|
||||||
)
|
|
||||||
return r
|
|
||||||
|
|
||||||
|
|
||||||
def _make_cardset():
|
|
||||||
cs, _ = Cardset.get_or_create(
|
|
||||||
name="IU Test Set",
|
|
||||||
defaults={"description": "image url test cardset", "total_cards": 1},
|
|
||||||
)
|
|
||||||
return cs
|
|
||||||
|
|
||||||
|
|
||||||
def _make_player(name: str = "Test Player") -> Player:
|
|
||||||
return Player.create(
|
|
||||||
p_name=name,
|
|
||||||
rarity=_make_rarity(),
|
|
||||||
cardset=_make_cardset(),
|
|
||||||
set_num=1,
|
|
||||||
pos_1="CF",
|
|
||||||
image="https://example.com/img.png",
|
|
||||||
mlbclub="TST",
|
|
||||||
franchise="TST",
|
|
||||||
description="image url test",
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def _make_team(suffix: str = "IU") -> Team:
|
|
||||||
return Team.create(
|
|
||||||
abbrev=suffix,
|
|
||||||
sname=suffix,
|
|
||||||
lname=f"Team {suffix}",
|
|
||||||
gmid=99900 + len(suffix),
|
|
||||||
gmname=f"gm_{suffix.lower()}",
|
|
||||||
gsheet="https://docs.google.com/iu_test",
|
|
||||||
wallet=500,
|
|
||||||
team_value=1000,
|
|
||||||
collection_value=1000,
|
|
||||||
season=11,
|
|
||||||
is_ai=False,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def _make_track(card_type: str = "batter") -> RefractorTrack:
|
|
||||||
track, _ = RefractorTrack.get_or_create(
|
|
||||||
name=f"IU {card_type} Track",
|
|
||||||
defaults=dict(
|
|
||||||
card_type=card_type,
|
|
||||||
formula="pa",
|
|
||||||
t1_threshold=100,
|
|
||||||
t2_threshold=300,
|
|
||||||
t3_threshold=700,
|
|
||||||
t4_threshold=1200,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
return track
|
|
||||||
|
|
||||||
|
|
||||||
def _make_batting_card(player: Player, variant: int, image_url=None) -> BattingCard:
|
|
||||||
return BattingCard.create(
|
|
||||||
player=player,
|
|
||||||
variant=variant,
|
|
||||||
steal_low=1,
|
|
||||||
steal_high=3,
|
|
||||||
steal_auto=False,
|
|
||||||
steal_jump=1.0,
|
|
||||||
bunting="N",
|
|
||||||
hit_and_run="N",
|
|
||||||
running=5,
|
|
||||||
offense_col=1,
|
|
||||||
hand="R",
|
|
||||||
image_url=image_url,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def _make_card_state(
|
|
||||||
player: Player,
|
|
||||||
team: Team,
|
|
||||||
track: RefractorTrack,
|
|
||||||
variant: int,
|
|
||||||
current_tier: int = 1,
|
|
||||||
current_value: float = 150.0,
|
|
||||||
) -> RefractorCardState:
|
|
||||||
import datetime
|
|
||||||
|
|
||||||
return RefractorCardState.create(
|
|
||||||
player=player,
|
|
||||||
team=team,
|
|
||||||
track=track,
|
|
||||||
current_tier=current_tier,
|
|
||||||
current_value=current_value,
|
|
||||||
fully_evolved=False,
|
|
||||||
last_evaluated_at=datetime.datetime(2026, 4, 1, 12, 0, 0),
|
|
||||||
variant=variant,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# Tests
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
def test_cards_response_includes_image_url(setup_img_url_db, img_url_client):
|
|
||||||
"""GET /api/v2/refractor/cards includes image_url when the variant BattingCard has one.
|
|
||||||
|
|
||||||
What: Seeds a RefractorCardState at variant=1 and a matching BattingCard with
|
|
||||||
image_url set. Calls the /cards endpoint and asserts that image_url in the
|
|
||||||
response matches the seeded URL.
|
|
||||||
|
|
||||||
Why: This is the primary happy-path test for the image_url feature. If the
|
|
||||||
DB lookup in _build_card_state_response fails or the field is accidentally
|
|
||||||
omitted from the response dict, this test will catch it.
|
|
||||||
"""
|
|
||||||
player = _make_player("Homer Simpson")
|
|
||||||
team = _make_team("IU1")
|
|
||||||
track = _make_track("batter")
|
|
||||||
|
|
||||||
expected_url = (
|
|
||||||
"https://s3.example.com/cards/cardset-001/player-1/v1/battingcard.png"
|
|
||||||
)
|
|
||||||
_make_batting_card(player, variant=1, image_url=expected_url)
|
|
||||||
_make_card_state(player, team, track, variant=1)
|
|
||||||
|
|
||||||
resp = img_url_client.get(
|
|
||||||
f"/api/v2/refractor/cards?team_id={team.id}&evaluated_only=false",
|
|
||||||
headers=AUTH_HEADER,
|
|
||||||
)
|
|
||||||
assert resp.status_code == 200, resp.text
|
|
||||||
data = resp.json()
|
|
||||||
assert data["count"] == 1
|
|
||||||
item = data["items"][0]
|
|
||||||
assert "image_url" in item, "image_url key missing from response"
|
|
||||||
assert item["image_url"] == expected_url
|
|
||||||
|
|
||||||
|
|
||||||
def test_cards_response_image_url_null_when_not_set(setup_img_url_db, img_url_client):
|
|
||||||
"""GET /api/v2/refractor/cards returns image_url: null when BattingCard.image_url is None.
|
|
||||||
|
|
||||||
What: Seeds a BattingCard with image_url=None and a RefractorCardState at
|
|
||||||
variant=1. Verifies the response contains image_url with a null value.
|
|
||||||
|
|
||||||
Why: The image_url field must always be present in the response (even when
|
|
||||||
null) so API consumers can rely on its presence. Returning null rather than
|
|
||||||
omitting the key is the correct contract — omitting it would break consumers
|
|
||||||
that check for the key's presence to determine upload status.
|
|
||||||
"""
|
|
||||||
player = _make_player("Bart Simpson")
|
|
||||||
team = _make_team("IU2")
|
|
||||||
track = _make_track("batter")
|
|
||||||
|
|
||||||
_make_batting_card(player, variant=1, image_url=None)
|
|
||||||
_make_card_state(player, team, track, variant=1)
|
|
||||||
|
|
||||||
resp = img_url_client.get(
|
|
||||||
f"/api/v2/refractor/cards?team_id={team.id}&evaluated_only=false",
|
|
||||||
headers=AUTH_HEADER,
|
|
||||||
)
|
|
||||||
assert resp.status_code == 200, resp.text
|
|
||||||
data = resp.json()
|
|
||||||
assert data["count"] == 1
|
|
||||||
item = data["items"][0]
|
|
||||||
assert "image_url" in item, "image_url key missing from response"
|
|
||||||
assert item["image_url"] is None
|
|
||||||
Loading…
Reference in New Issue
Block a user