92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
"""eth-labels.db source — 115K EVM labels from local SQLite.
|
|
|
|
Schema (verified):
|
|
accounts(id, chain_id, address, label, name_tag, created_at, updated_at)
|
|
|
|
chain_id mapping:
|
|
1 = ethereum, 42161 = arbitrum, 137 = polygon, etc.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import logging
|
|
from datetime import UTC, datetime
|
|
|
|
from app.domain.labels.models import Label, LabelSource
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
DB_PATH = "/home/dev/rmi/eth-labels.db"
|
|
|
|
# chain_id -> chain name mapping (from EVM chains list)
|
|
_CHAIN_ID_MAP = {
|
|
1: "ethereum",
|
|
42161: "arbitrum",
|
|
10: "optimism",
|
|
137: "polygon",
|
|
56: "bsc",
|
|
43114: "avalanche",
|
|
250: "fantom",
|
|
100: "gnosis",
|
|
}
|
|
|
|
|
|
async def query_eth_labels_db(address: str, chain: str = "ethereum") -> list[Label]:
|
|
"""Query the local eth-labels.db SQLite for an address.
|
|
|
|
Args:
|
|
address: EVM address (0x...)
|
|
chain: Chain name (only EVM chains in this DB)
|
|
|
|
Returns:
|
|
List of labels found (one per chain_id that has a match).
|
|
"""
|
|
def _query() -> list[Label]:
|
|
import sqlite3
|
|
|
|
try:
|
|
conn = sqlite3.connect(DB_PATH, timeout=2.0)
|
|
conn.row_factory = sqlite3.Row
|
|
try:
|
|
cursor = conn.execute(
|
|
"SELECT chain_id, label, name_tag, updated_at FROM accounts WHERE address = ? LIMIT 50",
|
|
(address.lower(),),
|
|
)
|
|
labels = []
|
|
for row in cursor:
|
|
chain_id = row["chain_id"]
|
|
chain_name = _CHAIN_ID_MAP.get(chain_id, f"chain-{chain_id}")
|
|
label_value = row["name_tag"] or row["label"]
|
|
if not label_value:
|
|
continue
|
|
labels.append(
|
|
Label(
|
|
address=address.lower(),
|
|
chain=chain_name,
|
|
label_type="entity",
|
|
label_value=label_value,
|
|
source=LabelSource.ETH_LABELS_DB,
|
|
confidence=0.9, # eth-labels.db is curated
|
|
verified_at=_parse_date(row["updated_at"]),
|
|
attributes={"chain_id": chain_id, "raw_label": row["label"]},
|
|
)
|
|
)
|
|
return labels
|
|
finally:
|
|
conn.close()
|
|
except Exception as e:
|
|
log.warning("eth_labels_db_query_fail address=%s err=%s", address[:10], e)
|
|
raise
|
|
|
|
return await asyncio.to_thread(_query)
|
|
|
|
|
|
def _parse_date(s: str | None) -> datetime:
|
|
"""Parse SQLite date string to datetime."""
|
|
if not s:
|
|
return datetime.now(UTC)
|
|
try:
|
|
# Format: '2024-06-30 21:58:54'
|
|
return datetime.fromisoformat(s).replace(tzinfo=UTC)
|
|
except Exception:
|
|
return datetime.now(UTC)
|