SplatNet-Backend/routes/Ranking.py

from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session as DBSession
from sqlalchemy import desc
from database import SessionLocal, PlayerRank, EquipmentLast
from fastapi_cache import FastAPICache
from fastapi_cache.backends.inmemory import InMemoryBackend
from fastapi_cache.decorator import cache

router = APIRouter()

def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()

@router.get("/leaderboard")
@cache(expire=60)
async def get_leaderboard(db: DBSession = Depends(get_db)):
    try:
        top_100_query = (
            db.query(PlayerRank.PId)
            .order_by(desc(PlayerRank.FesPower))
            .limit(100)
            .all()
        )
        top_100_tuples = set(top_100_query)
        
        modes = {0, 1, 2}
        leaderboard = {}

        for mode in modes:
            top_players = (
                db.query(PlayerRank, EquipmentLast)
                .outerjoin(EquipmentLast, PlayerRank.PId == EquipmentLast.PId)
                .filter(PlayerRank.GameMode == mode)
                .order_by(desc(PlayerRank.RankingScore))
                .limit(9)
                .all()
            )

            mode_results = []
            for player, gear in top_players:
                player_data = {
                    "PId": player.PId,
                    "MiiName": player.MiiName,
                    "Rank": player.Rank,
                    "WinSum": player.WinSum,
                    "LoseSum": player.LoseSum,
                    "RankingScore": round(player.RankingScore, 2),
                    "weapon": gear.weapon if gear else None,
                    "headgear": gear.Gear_Head if gear else None,
                    "clothes": gear.Gear_Clothes if gear else None,
                    "shoes": gear.Gear_Shoes if gear else None,
                }

                if mode == 2:
                    is_top_100 = (player.PId,) in top_100_tuples
                    player_data["FesPower"] = player.FesPower
                    player_data["is_top_100_fes"] = is_top_100
                
                mode_results.append(player_data)
            
            leaderboard[f"mode_{mode}"] = mode_results

        return leaderboard

    except Exception as e:
        print(f"Error: {e}")
        return {"error": str(e)}