feat: reexport TPSC schemas from hub-core

api/schemas/tpsc.py

@@ -1,115 +1,29 @@
-import uuid
-from datetime import datetime
-from typing import Literal
-from pydantic import BaseModel, computed_field
+from hub_core.schemas.tpsc import (
+    AuthType,
+    GDPRMaturity,
+    GDPR_WARNING_LEVELS,
+    PricingModel,
+    TPSCCatalogCreate,
+    TPSCCatalogRead,
+    TPSCEntryCreate,
+    TPSCEntryRead,
+    TPSCGDPRReport,
+    TPSCGDPRWarning,
+    TPSCIngestRequest,
+    TPSCSnapshotRead,
+)
 
-# GDPR maturity scale (CNIL/IAPP CMMI-aligned, adapted for third-party assessment)
-GDPRMaturity = Literal["unknown", "non_compliant", "initial", "developing", "defined", "managed", "certified"]
-
-# Services at these levels trigger a GDPR warning
-GDPR_WARNING_LEVELS = {"unknown", "non_compliant", "initial"}
-
-PricingModel = Literal["free", "paid", "freemium", "usage_based", "unknown"]
-AuthType = Literal["api_key", "oauth", "cli", "none", "unknown"]
-
-
-class TPSCCatalogCreate(BaseModel):
-    slug: str
-    name: str
-    provider: str | None = None
-    category: str | None = None
-    website_url: str | None = None
-    pricing_model: PricingModel = "unknown"
-    gdpr_maturity: GDPRMaturity = "unknown"
-    gdpr_notes: str | None = None
-    dpa_available: bool = False
-    tos_url: str | None = None
-    privacy_policy_url: str | None = None
-    data_processing_regions: list[str] | None = None
-    data_retention_notes: str | None = None
-    status: str = "active"
-
-
-class TPSCCatalogRead(BaseModel):
-    model_config = {"from_attributes": True}
-
-    id: uuid.UUID
-    slug: str
-    name: str
-    provider: str | None
-    category: str | None
-    website_url: str | None
-    pricing_model: str
-    gdpr_maturity: str
-    gdpr_notes: str | None
-    dpa_available: bool
-    tos_url: str | None
-    privacy_policy_url: str | None
-    data_processing_regions: list[str] | None
-    data_retention_notes: str | None
-    status: str
-    created_at: datetime
-    updated_at: datetime
-
-    @computed_field
-    @property
-    def gdpr_warning(self) -> bool:
-        return self.gdpr_maturity in GDPR_WARNING_LEVELS
-
-
-class TPSCEntryCreate(BaseModel):
-    service_slug: str
-    purpose: str | None = None
-    auth_type: str | None = None
-    endpoint_override: str | None = None
-    notes: str | None = None
-
-
-class TPSCEntryRead(BaseModel):
-    model_config = {"from_attributes": True}
-
-    id: uuid.UUID
-    snapshot_id: uuid.UUID
-    catalog_id: uuid.UUID | None
-    service_slug: str
-    purpose: str | None
-    auth_type: str | None
-    endpoint_override: str | None
-    notes: str | None
-    # Denormalised from catalog for convenience
-    gdpr_maturity: str | None = None
-    gdpr_warning: bool = False
-    pricing_model: str | None = None
-
-
-class TPSCIngestRequest(BaseModel):
-    repo_slug: str
-    source_file: str = "tpsc.yaml"
-    entries: list[TPSCEntryCreate]
-
-
-class TPSCSnapshotRead(BaseModel):
-    model_config = {"from_attributes": True}
-
-    id: uuid.UUID
-    repo_id: uuid.UUID | None
-    snapshot_at: datetime
-    source_file: str | None
-    entry_count: int
-    entries: list[TPSCEntryRead] = []
-
-
-class TPSCGDPRWarning(BaseModel):
-    repo_slug: str | None
-    service_slug: str
-    gdpr_maturity: str
-    purpose: str | None
-    pricing_model: str | None
-
-
-class TPSCGDPRReport(BaseModel):
-    generated_at: datetime
-    total_services: int
-    warning_count: int
-    warnings: list[TPSCGDPRWarning]
-    by_maturity: dict[str, int]
+__all__ = [
+    "AuthType",
+    "GDPRMaturity",
+    "GDPR_WARNING_LEVELS",
+    "PricingModel",
+    "TPSCCatalogCreate",
+    "TPSCCatalogRead",
+    "TPSCEntryCreate",
+    "TPSCEntryRead",
+    "TPSCGDPRReport",
+    "TPSCGDPRWarning",
+    "TPSCIngestRequest",
+    "TPSCSnapshotRead",
+]

tests/test_hub_core_imports.py

@@ -1,7 +1,13 @@
 from api.schemas.agent_message import MessageCreate
 from api.schemas.doi import DoIReport
+from api.schemas.tpsc import GDPR_WARNING_LEVELS, TPSCCatalogRead, TPSCGDPRReport
 from hub_core.schemas.agent_message import MessageCreate as CoreMessageCreate
 from hub_core.schemas.doi import DoIReport as CoreDoIReport
+from hub_core.schemas.tpsc import (
+    GDPR_WARNING_LEVELS as CORE_GDPR_WARNING_LEVELS,
+    TPSCCatalogRead as CoreTPSCCatalogRead,
+    TPSCGDPRReport as CoreTPSCGDPRReport,
+)
 
 
 def test_state_hub_reexports_core_message_schema() -> None:
@@ -10,3 +16,9 @@ def test_state_hub_reexports_core_message_schema() -> None:
 
 def test_state_hub_reexports_core_doi_schema() -> None:
     assert DoIReport is CoreDoIReport
+
+
+def test_state_hub_reexports_core_tpsc_schemas() -> None:
+    assert TPSCCatalogRead is CoreTPSCCatalogRead
+    assert TPSCGDPRReport is CoreTPSCGDPRReport
+    assert GDPR_WARNING_LEVELS is CORE_GDPR_WARNING_LEVELS