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>
82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
"""
|
|
Cross-Token Tracking API Router - Track wallets across multiple tokens.
|
|
Connects to /api/v1/cross-token/*
|
|
"""
|
|
|
|
from fastapi import APIRouter, HTTPException
|
|
from pydantic import BaseModel
|
|
|
|
router = APIRouter(prefix="/api/v1/cross-token", tags=["cross-token"])
|
|
|
|
# Lazy import - cross_token.py has broken internal deps
|
|
PROJECT_TOKENS = {
|
|
"CRM": "Eme5T2s2HB7B8W4YgLG1eReQpnadEVUnQBRjaKTdBAGS",
|
|
"SOSANA": "SoSaNaTokenAddressPlaceholder123456789",
|
|
"PBTC": "pBTC Token Address Placeholder",
|
|
"SHIFT_AI": "ShiftAITokenAddressPlaceholder123456",
|
|
}
|
|
|
|
|
|
class CrossTokenRequest(BaseModel):
|
|
wallet: str
|
|
token_addresses: list[str] | None = None
|
|
|
|
|
|
@router.get("/projects")
|
|
async def list_projects():
|
|
"""List all tracked token projects."""
|
|
return {"projects": PROJECT_TOKENS, "count": len(PROJECT_TOKENS)}
|
|
|
|
|
|
@router.post("/affiliations")
|
|
async def track_affiliations(req: CrossTokenRequest):
|
|
"""Track a wallet's affiliations across all known token projects."""
|
|
try:
|
|
from app.cross_token import CrossTokenAffiliationTracker
|
|
|
|
tracker = CrossTokenAffiliationTracker()
|
|
affiliations = await tracker.track_wallet_affiliations(wallet=req.wallet)
|
|
return {
|
|
"wallet": req.wallet,
|
|
"affiliations": [
|
|
{
|
|
"project": a.project,
|
|
"token_address": a.token_address,
|
|
"balance": a.balance,
|
|
"first_acquired": str(a.first_acquired) if a.first_acquired else None,
|
|
"evidence_strength": a.evidence_strength,
|
|
}
|
|
for a in affiliations
|
|
],
|
|
"project_count": len(affiliations),
|
|
}
|
|
except ImportError as e:
|
|
raise HTTPException(status_code=503, detail=f"Module unavailable: {e}") from e
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e)[:500]) from e
|
|
|
|
|
|
@router.get("/connections/{wallet}")
|
|
async def get_cross_connections(wallet: str):
|
|
"""Find cross-token connections for a wallet."""
|
|
try:
|
|
from app.cross_token import CrossTokenAffiliationTracker
|
|
|
|
tracker = CrossTokenAffiliationTracker()
|
|
affiliations = await tracker.track_wallet_affiliations(wallet=wallet)
|
|
multi = [a for a in affiliations if a.evidence_strength != "unverified"]
|
|
return {
|
|
"wallet": wallet,
|
|
"multi_project": len(multi),
|
|
"projects": [a.project for a in multi],
|
|
"total_affiliations": len(affiliations),
|
|
}
|
|
except ImportError as e:
|
|
raise HTTPException(status_code=503, detail=str(e)) from e
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e)[:500]) from e
|
|
|
|
|
|
@router.get("/health")
|
|
async def cross_token_health():
|
|
return {"status": "ok", "service": "cross-token-tracker"}
|