magicciv/tools/sprite-generation/test_rank_simple.py

24 lines
720 B
Python

#!/usr/bin/env python3
import sys
from pathlib import Path
sys.path.insert(0, 'engine')
from registry import SpriteRegistry
def test():
reg = SpriteRegistry(db_path=Path('spritegen.db'))
sprite_id = 'units/bowmen_dwarves_f'
# Manually run the same query as the ranker
variants = reg.conn.execute(
"SELECT * FROM variants WHERE sprite_id=? AND job_status='completed' "
"AND raw_path IS NOT NULL AND (rating IS NULL OR rating != -1) "
"AND COALESCE(review_tier, 0) = ?",
(sprite_id, 0),
).fetchall()
print(f"Found {len(variants)} variants needing tier 0 scoring")
for v in variants:
print(f" Variant {v['id']}: raw={v['raw_path']}")
test()