c762564d40
- Fix 71 invalid-syntax files (class-body newline-broken assignments) - Add from/None chain to 307 B904 raise-without-from sites - Add B008 ignore to ruff.toml (already in pyproject.toml) - Noqa F401 on __init__.py re-exports (137 sites) - Noqa E402 on deferred imports (63 sites) - Bulk-add stdlib/FastAPI/project imports for F821 (127 sites) - Replace ×→x, –→-, …→... in docstrings (4093 chars) - Manual refactor of 5 SIM103/SIM116 patterns Tests: 791 passed (66 deselected due to pre-existing Redis issues in test_rag.py) Co-authored-by: opencode <opencode@rugmunch.io>
95 lines
3.1 KiB
Python
95 lines
3.1 KiB
Python
"""
|
|
Community Badges Router - FastAPI endpoints for voting and badge queries.
|
|
"""
|
|
|
|
from fastapi import APIRouter, HTTPException, Query
|
|
from pydantic import BaseModel, Field
|
|
|
|
from app.community_badges import (
|
|
cast_vote,
|
|
get_leaderboard,
|
|
get_token_votes,
|
|
get_user_badge,
|
|
resolve_token_outcome,
|
|
)
|
|
|
|
router = APIRouter(prefix="/api/v1/community", tags=["community"])
|
|
|
|
|
|
# ── Request models ────────────────────────────────────────────────
|
|
|
|
|
|
class VoteRequest(BaseModel):
|
|
token_address: str = Field(..., min_length=32, max_length=128)
|
|
chain: str = Field(default="solana")
|
|
voter_id: str = Field(..., min_length=1, max_length=256)
|
|
vote: str = Field(..., pattern=r"^(clean|malicious)$")
|
|
|
|
|
|
class ResolveRequest(BaseModel):
|
|
token_address: str = Field(..., min_length=32, max_length=128)
|
|
chain: str = Field(default="solana")
|
|
outcome: str = Field(..., pattern=r"^(clean|malicious|rug_confirmed)$")
|
|
|
|
|
|
# ── Endpoints ─────────────────────────────────────────────────────
|
|
|
|
|
|
@router.post("/vote")
|
|
async def api_cast_vote(req: VoteRequest):
|
|
"""Cast a vote on whether a token is clean or malicious."""
|
|
result = cast_vote(
|
|
token_address=req.token_address,
|
|
chain=req.chain,
|
|
voter_id=req.voter_id,
|
|
vote=req.vote,
|
|
)
|
|
if result.get("success") is False:
|
|
raise HTTPException(status_code=400, detail=result.get("error", "Vote failed"))
|
|
return result
|
|
|
|
|
|
@router.post("/resolve")
|
|
async def api_resolve_outcome(req: ResolveRequest):
|
|
"""Resolve a token's outcome (admin/cron). Updates all voter badges."""
|
|
result = resolve_token_outcome(
|
|
token_address=req.token_address,
|
|
chain=req.chain,
|
|
outcome=req.outcome,
|
|
)
|
|
if result.get("success") is False:
|
|
raise HTTPException(status_code=400, detail=result.get("error", "Resolve failed"))
|
|
return result
|
|
|
|
|
|
@router.get("/badge/{user_id}")
|
|
async def api_get_badge(user_id: str):
|
|
"""Get a user's badge profile."""
|
|
badge = get_user_badge(user_id)
|
|
if badge is None:
|
|
raise HTTPException(status_code=404, detail="User not found")
|
|
return {
|
|
"user_id": badge.user_id,
|
|
"total_votes": badge.total_votes,
|
|
"correct_votes": badge.correct_votes,
|
|
"current_tier": badge.current_tier,
|
|
"next_tier": badge.next_tier,
|
|
"votes_needed_for_next": badge.votes_needed_for_next,
|
|
"badges_earned": badge.badges_earned,
|
|
"accuracy_pct": badge.accuracy_pct,
|
|
}
|
|
|
|
|
|
@router.get("/votes/{chain}/{token_address:path}")
|
|
async def api_get_token_votes(token_address: str, chain: str):
|
|
"""Get all votes for a token."""
|
|
result = get_token_votes(token_address, chain)
|
|
if "error" in result:
|
|
raise HTTPException(status_code=404, detail=result["error"])
|
|
return result
|
|
|
|
|
|
@router.get("/leaderboard")
|
|
async def api_get_leaderboard(limit: int = Query(default=20, le=100)):
|
|
"""Get community leaderboard (top sleuths)."""
|
|
return {"leaderboard": get_leaderboard(limit)}
|