paper-dynasty-card-creation/analyze_cardset_rarity.py

import asyncio
from collections import defaultdict
from db_calls import db_get


async def analyze_cardset_rarity(cardset_id: int = 27):
    """Analyze players by franchise and rarity for a given cardset."""

    print(f"Fetching players from cardset {cardset_id}...\n")

    # Fetch all players from the cardset
    p_query = await db_get(
        "players", params=[("cardset_id", cardset_id), ("inc_dex", False)]
    )

    if not p_query or p_query["count"] == 0:
        print(f"No players found for cardset {cardset_id}")
        return

    players = p_query["players"]
    print(f"Found {len(players)} players\n")

    # First pass: collect all unique rarity names
    all_rarities = set()
    for player in players:
        rarity_obj = player.get("rarity", {})
        if isinstance(rarity_obj, dict):
            rarity_name = rarity_obj.get("name")
            if rarity_name:
                all_rarities.add(rarity_name)

    sorted_rarities = sorted(
        all_rarities,
        key=lambda x: (
            {
                "Common": 0,
                "Uncommon": 1,
                "Rare": 2,
                "Epic": 3,
                "Legend": 4,
                "Starter": 0,
                "Bench": 1,
                "All-Star": 2,
                "MVP": 3,
                "Hall of Fame": 4,
            }.get(x, 99),
            x,
        ),
    )

    print(f"Found rarities: {sorted_rarities}\n")

    # Group players by franchise and rarity
    franchise_data = defaultdict(
        lambda: {
            "batters": defaultdict(int),
            "pitchers": defaultdict(int),
            "combined": defaultdict(int),
        }
    )

    for player in players:
        franchise = player.get("franchise", "Unknown")
        rarity_obj = player.get("rarity", {})

        # Extract rarity name from rarity object
        if isinstance(rarity_obj, dict):
            rarity = rarity_obj.get("name", "Unknown")
        else:
            rarity = str(rarity_obj) if rarity_obj else "Unknown"

        # Determine if batter or pitcher based on positions
        positions = []
        for i in range(1, 9):
            pos = player.get(f"pos_{i}")
            if pos:
                positions.append(pos)

        is_pitcher = any(pos in ["SP", "RP", "CP"] for pos in positions)

        if is_pitcher:
            franchise_data[franchise]["pitchers"][rarity] += 1
        else:
            franchise_data[franchise]["batters"][rarity] += 1

        franchise_data[franchise]["combined"][rarity] += 1

    # Sort franchises alphabetically
    sorted_franchises = sorted(franchise_data.keys())

    # Print batters
    print("=" * 100)
    print("BATTERS BY FRANCHISE AND RARITY")
    print("=" * 100)

    # Build header dynamically
    header = f'{"Franchise":<20}'
    for rarity in sorted_rarities:
        header += f" {rarity:<12}"
    header += f' {"Total":<10}'
    print(header)
    print("-" * 100)

    batter_totals = defaultdict(int)
    for franchise in sorted_franchises:
        batters = franchise_data[franchise]["batters"]
        total = sum(batters.values())

        if total > 0:
            row = f"{franchise:<20}"
            for rarity in sorted_rarities:
                count = batters.get(rarity, 0)
                row += f" {count:<12}"
                batter_totals[rarity] += count
            row += f" {total:<10}"
            batter_totals["Total"] += total
            print(row)

    print("-" * 100)
    total_row = f'{"TOTAL":<20}'
    for rarity in sorted_rarities:
        total_row += f" {batter_totals[rarity]:<12}"
    total_row += f' {batter_totals["Total"]:<10}'
    print(total_row)

    # Print pitchers
    print("\n" + "=" * 100)
    print("PITCHERS BY FRANCHISE AND RARITY")
    print("=" * 100)

    header = f'{"Franchise":<20}'
    for rarity in sorted_rarities:
        header += f" {rarity:<12}"
    header += f' {"Total":<10}'
    print(header)
    print("-" * 100)

    pitcher_totals = defaultdict(int)
    for franchise in sorted_franchises:
        pitchers = franchise_data[franchise]["pitchers"]
        total = sum(pitchers.values())

        if total > 0:
            row = f"{franchise:<20}"
            for rarity in sorted_rarities:
                count = pitchers.get(rarity, 0)
                row += f" {count:<12}"
                pitcher_totals[rarity] += count
            row += f" {total:<10}"
            pitcher_totals["Total"] += total
            print(row)

    print("-" * 100)
    total_row = f'{"TOTAL":<20}'
    for rarity in sorted_rarities:
        total_row += f" {pitcher_totals[rarity]:<12}"
    total_row += f' {pitcher_totals["Total"]:<10}'
    print(total_row)

    # Print combined
    print("\n" + "=" * 100)
    print("COMBINED (BATTERS + PITCHERS) BY FRANCHISE AND RARITY")
    print("=" * 100)

    header = f'{"Franchise":<20}'
    for rarity in sorted_rarities:
        header += f" {rarity:<12}"
    header += f' {"Total":<10}'
    print(header)
    print("-" * 100)

    combined_totals = defaultdict(int)
    for franchise in sorted_franchises:
        combined = franchise_data[franchise]["combined"]
        total = sum(combined.values())

        row = f"{franchise:<20}"
        for rarity in sorted_rarities:
            count = combined.get(rarity, 0)
            row += f" {count:<12}"
            combined_totals[rarity] += count
        row += f" {total:<10}"
        combined_totals["Total"] += total
        print(row)

    print("-" * 100)
    total_row = f'{"TOTAL":<20}'
    for rarity in sorted_rarities:
        total_row += f" {combined_totals[rarity]:<12}"
    total_row += f' {combined_totals["Total"]:<10}'
    print(total_row)
    print("=" * 100)


if __name__ == "__main__":
    asyncio.run(analyze_cardset_rarity(27))