pyr0ball
e34c2b9982
SharedTableProtocol now covers the full shared-table surface:
- sellers, market_comps, reported_sellers (already in SnipeSharedStore)
- scammer_blocklist (new — is_blocklisted, add/remove/list_blocklist)
- refresh_seller_categories (reads per-user SQLite, writes to Postgres)
TrustScorer updated to accept SharedTableProtocol (was Store).
api/main.py:
- _pg_shared_store global + _make_shared_store(path) helper
- Lifespan init: SNIPE_SHARED_DB_URL → SnipeSharedDB + SnipeSharedStore
- All Store(shared_db) calls for shared tables replaced with
_make_shared_store(shared_db) or shared_store.clone()
- Blocklist endpoints use _make_shared_store (Postgres when configured)
- Community signals stay SQLite-only (low-write, not in protocol)
Postgres migration 001: scammer_blocklist table added.
8 blocklist tests added (gated behind SNIPE_SHARED_DB_URL / @pytest.mark.postgres).
.env.example: SNIPE_SHARED_DB_URL documented.
compose.cloud.yml: GPU_SERVER_URL + SNIPE_SHARED_DB_URL comment added.
248 passed, 8 skipped (postgres-gated).
Closes: #45
86 lines
3 KiB
Python
86 lines
3 KiB
Python
"""Protocol (duck-type interface) for shared table backends (SQLite and Postgres)."""
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Optional, Protocol, runtime_checkable
|
|
|
|
from app.db.models import MarketComp, ScammerEntry, Seller
|
|
|
|
|
|
@runtime_checkable
|
|
class SharedTableProtocol(Protocol):
|
|
"""Protocol that both Store (SQLite) and SnipeSharedStore (Postgres) must satisfy.
|
|
|
|
This enables code that reads/writes shared tables (sellers, market_comps,
|
|
reported_sellers, scammer_blocklist) to remain agnostic to the underlying backend.
|
|
"""
|
|
|
|
def save_seller(self, seller: Seller) -> None:
|
|
"""Persist a single seller record."""
|
|
...
|
|
|
|
def save_sellers(self, sellers: list[Seller]) -> None:
|
|
"""Persist multiple seller records (batch upsert)."""
|
|
...
|
|
|
|
def get_seller(self, platform: str, platform_seller_id: str) -> Optional[Seller]:
|
|
"""Fetch a single seller by platform and platform_seller_id."""
|
|
...
|
|
|
|
def save_market_comp(self, comp: MarketComp) -> None:
|
|
"""Persist a market comparison record."""
|
|
...
|
|
|
|
def get_market_comp(self, platform: str, query_hash: str) -> Optional[MarketComp]:
|
|
"""Fetch a market comparison by platform and query_hash."""
|
|
...
|
|
|
|
def mark_reported(
|
|
self,
|
|
platform: str,
|
|
platform_seller_id: str,
|
|
username: Optional[str] = None,
|
|
reported_by: str = "user",
|
|
) -> None:
|
|
"""Record that a seller has been reported to the platform."""
|
|
...
|
|
|
|
def list_reported(self, platform: str = "ebay") -> list[str]:
|
|
"""Return all platform_seller_ids that have been reported."""
|
|
...
|
|
|
|
def delete_seller_data(self, platform: str, platform_seller_id: str) -> None:
|
|
"""Permanently erase a seller and all related data (GDPR/eBay compliance)."""
|
|
...
|
|
|
|
def refresh_seller_categories(
|
|
self,
|
|
platform: str,
|
|
seller_ids: list[str],
|
|
listing_store: Optional[Any] = None,
|
|
) -> int:
|
|
"""Derive category_history_json for sellers that lack it from stored listings.
|
|
|
|
listing_store: Store holding listings (may differ from self in split-DB mode).
|
|
Returns count of sellers updated.
|
|
"""
|
|
...
|
|
|
|
def is_blocklisted(self, platform: str, platform_seller_id: str) -> bool:
|
|
"""Return True if a seller is on the community scammer blocklist."""
|
|
...
|
|
|
|
def add_to_blocklist(self, entry: ScammerEntry) -> ScammerEntry:
|
|
"""Upsert a seller into the blocklist. Returns the saved entry with id and created_at."""
|
|
...
|
|
|
|
def remove_from_blocklist(self, platform: str, platform_seller_id: str) -> None:
|
|
"""Remove a seller from the blocklist."""
|
|
...
|
|
|
|
def list_blocklist(self, platform: str = "ebay") -> list[ScammerEntry]:
|
|
"""Return all blocklisted sellers for a platform, newest first."""
|
|
...
|
|
|
|
def clone(self) -> SharedTableProtocol:
|
|
"""Create a new independent instance pointing to the same backend."""
|
|
...
|