a39ae4ccba
Replace coarse milestone/referral suppression with 8 named action categories (vote, presidential, committee_report, calendar, procedural, referral, new_document, new_amendment), each independently togglable per follow mode (Follow / Pocket Veto / Pocket Boost). - notification_utils: categorize_action() replaces is_milestone_action / is_referral_action; _build_payload stores action_category in payload - congress_poller: use categorize_action() in _update_bill_if_changed - notification_dispatcher: _should_dispatch() checks per-mode filter dict from notification_prefs; follow mode looked up before filter check - schemas + api: alert_filters (nested dict) wired through settings GET/PUT endpoints; no DB migration required - frontend: tabbed Alert Filters section (Follow / Pocket Veto / Pocket Boost), each with independent 8-toggle filter set, milestone parent checkbox (indeterminate-aware), Load defaults button, and per-tab Save button Authored-By: Jack Levy
340 lines
11 KiB
Python
340 lines
11 KiB
Python
from datetime import date, datetime
|
||
from typing import Any, Generic, Optional, TypeVar
|
||
|
||
from pydantic import BaseModel
|
||
|
||
|
||
# ── Notifications ──────────────────────────────────────────────────────────────
|
||
|
||
# ── Bill Notes ────────────────────────────────────────────────────────────────
|
||
|
||
class BillNoteSchema(BaseModel):
|
||
id: int
|
||
bill_id: str
|
||
content: str
|
||
pinned: bool
|
||
created_at: datetime
|
||
updated_at: datetime
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
class BillNoteUpsert(BaseModel):
|
||
content: str
|
||
pinned: bool = False
|
||
|
||
|
||
# ── Notifications ──────────────────────────────────────────────────────────────
|
||
|
||
class NotificationSettingsResponse(BaseModel):
|
||
ntfy_topic_url: str = ""
|
||
ntfy_auth_method: str = "none" # none | token | basic
|
||
ntfy_token: str = ""
|
||
ntfy_username: str = ""
|
||
ntfy_password: str = ""
|
||
ntfy_enabled: bool = False
|
||
rss_enabled: bool = False
|
||
rss_token: Optional[str] = None
|
||
# Digest
|
||
digest_enabled: bool = False
|
||
digest_frequency: str = "daily" # daily | weekly
|
||
# Quiet hours — stored as local-time hour integers (0-23); timezone is IANA name
|
||
quiet_hours_start: Optional[int] = None
|
||
quiet_hours_end: Optional[int] = None
|
||
timezone: Optional[str] = None # IANA name, e.g. "America/New_York"
|
||
alert_filters: Optional[dict] = None
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
class NotificationSettingsUpdate(BaseModel):
|
||
ntfy_topic_url: Optional[str] = None
|
||
ntfy_auth_method: Optional[str] = None
|
||
ntfy_token: Optional[str] = None
|
||
ntfy_username: Optional[str] = None
|
||
ntfy_password: Optional[str] = None
|
||
ntfy_enabled: Optional[bool] = None
|
||
rss_enabled: Optional[bool] = None
|
||
digest_enabled: Optional[bool] = None
|
||
digest_frequency: Optional[str] = None
|
||
quiet_hours_start: Optional[int] = None
|
||
quiet_hours_end: Optional[int] = None
|
||
timezone: Optional[str] = None # IANA name sent by the browser on save
|
||
alert_filters: Optional[dict] = None
|
||
|
||
|
||
class NotificationEventSchema(BaseModel):
|
||
id: int
|
||
bill_id: str
|
||
event_type: str
|
||
payload: Optional[Any] = None
|
||
dispatched_at: Optional[datetime] = None
|
||
created_at: datetime
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
class NtfyTestRequest(BaseModel):
|
||
ntfy_topic_url: str
|
||
ntfy_auth_method: str = "none"
|
||
ntfy_token: str = ""
|
||
ntfy_username: str = ""
|
||
ntfy_password: str = ""
|
||
|
||
|
||
class FollowModeTestRequest(BaseModel):
|
||
mode: str # pocket_veto | pocket_boost
|
||
event_type: str # new_document | new_amendment | bill_updated
|
||
|
||
|
||
class NotificationTestResult(BaseModel):
|
||
status: str # "ok" | "error"
|
||
detail: str
|
||
event_count: Optional[int] = None # RSS only
|
||
|
||
T = TypeVar("T")
|
||
|
||
|
||
class PaginatedResponse(BaseModel, Generic[T]):
|
||
items: list[T]
|
||
total: int
|
||
page: int
|
||
per_page: int
|
||
pages: int
|
||
|
||
|
||
# ── Member ────────────────────────────────────────────────────────────────────
|
||
|
||
class MemberSchema(BaseModel):
|
||
bioguide_id: str
|
||
name: str
|
||
first_name: Optional[str] = None
|
||
last_name: Optional[str] = None
|
||
party: Optional[str] = None
|
||
state: Optional[str] = None
|
||
chamber: Optional[str] = None
|
||
district: Optional[str] = None
|
||
photo_url: Optional[str] = None
|
||
official_url: Optional[str] = None
|
||
congress_url: Optional[str] = None
|
||
birth_year: Optional[str] = None
|
||
address: Optional[str] = None
|
||
phone: Optional[str] = None
|
||
terms_json: Optional[list[Any]] = None
|
||
leadership_json: Optional[list[Any]] = None
|
||
sponsored_count: Optional[int] = None
|
||
cosponsored_count: Optional[int] = None
|
||
latest_trend: Optional["MemberTrendScoreSchema"] = None
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
# ── Bill Brief ────────────────────────────────────────────────────────────────
|
||
|
||
class BriefSchema(BaseModel):
|
||
id: int
|
||
brief_type: str = "full"
|
||
summary: Optional[str] = None
|
||
key_points: Optional[list[Any]] = None
|
||
risks: Optional[list[Any]] = None
|
||
deadlines: Optional[list[dict[str, Any]]] = None
|
||
topic_tags: Optional[list[str]] = None
|
||
llm_provider: Optional[str] = None
|
||
llm_model: Optional[str] = None
|
||
govinfo_url: Optional[str] = None
|
||
share_token: Optional[str] = None
|
||
created_at: Optional[datetime] = None
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
# ── Bill Action ───────────────────────────────────────────────────────────────
|
||
|
||
class BillActionSchema(BaseModel):
|
||
id: int
|
||
action_date: Optional[date] = None
|
||
action_text: Optional[str] = None
|
||
action_type: Optional[str] = None
|
||
chamber: Optional[str] = None
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
# ── News Article ──────────────────────────────────────────────────────────────
|
||
|
||
class NewsArticleSchema(BaseModel):
|
||
id: int
|
||
source: Optional[str] = None
|
||
headline: Optional[str] = None
|
||
url: Optional[str] = None
|
||
published_at: Optional[datetime] = None
|
||
relevance_score: Optional[float] = None
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
# ── Trend Score ───────────────────────────────────────────────────────────────
|
||
|
||
class TrendScoreSchema(BaseModel):
|
||
score_date: date
|
||
newsapi_count: int
|
||
gnews_count: int
|
||
gtrends_score: float
|
||
composite_score: float
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
class MemberTrendScoreSchema(BaseModel):
|
||
score_date: date
|
||
newsapi_count: int
|
||
gnews_count: int
|
||
gtrends_score: float
|
||
composite_score: float
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
class MemberNewsArticleSchema(BaseModel):
|
||
id: int
|
||
source: Optional[str] = None
|
||
headline: Optional[str] = None
|
||
url: Optional[str] = None
|
||
published_at: Optional[datetime] = None
|
||
relevance_score: Optional[float] = None
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
# ── Bill ──────────────────────────────────────────────────────────────────────
|
||
|
||
class BillSchema(BaseModel):
|
||
bill_id: str
|
||
congress_number: int
|
||
bill_type: str
|
||
bill_number: int
|
||
title: Optional[str] = None
|
||
short_title: Optional[str] = None
|
||
introduced_date: Optional[date] = None
|
||
latest_action_date: Optional[date] = None
|
||
latest_action_text: Optional[str] = None
|
||
status: Optional[str] = None
|
||
chamber: Optional[str] = None
|
||
congress_url: Optional[str] = None
|
||
sponsor: Optional[MemberSchema] = None
|
||
latest_brief: Optional[BriefSchema] = None
|
||
latest_trend: Optional[TrendScoreSchema] = None
|
||
updated_at: Optional[datetime] = None
|
||
has_document: bool = False
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
class BillDetailSchema(BillSchema):
|
||
actions: list[BillActionSchema] = []
|
||
news_articles: list[NewsArticleSchema] = []
|
||
trend_scores: list[TrendScoreSchema] = []
|
||
briefs: list[BriefSchema] = []
|
||
has_document: bool = False
|
||
|
||
|
||
# ── Follow ────────────────────────────────────────────────────────────────────
|
||
|
||
class FollowCreate(BaseModel):
|
||
follow_type: str # bill | member | topic
|
||
follow_value: str
|
||
|
||
|
||
class FollowSchema(BaseModel):
|
||
id: int
|
||
user_id: int
|
||
follow_type: str
|
||
follow_value: str
|
||
follow_mode: str = "neutral"
|
||
created_at: datetime
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
class FollowModeUpdate(BaseModel):
|
||
follow_mode: str
|
||
|
||
|
||
# ── Settings ──────────────────────────────────────────────────────────────────
|
||
|
||
# ── Auth ──────────────────────────────────────────────────────────────────────
|
||
|
||
class UserCreate(BaseModel):
|
||
email: str
|
||
password: str
|
||
|
||
|
||
class UserResponse(BaseModel):
|
||
id: int
|
||
email: str
|
||
is_admin: bool
|
||
notification_prefs: dict
|
||
created_at: Optional[datetime] = None
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
class TokenResponse(BaseModel):
|
||
access_token: str
|
||
token_type: str = "bearer"
|
||
user: "UserResponse"
|
||
|
||
|
||
# ── Settings ──────────────────────────────────────────────────────────────────
|
||
|
||
class SettingUpdate(BaseModel):
|
||
key: str
|
||
value: str
|
||
|
||
|
||
class SettingsResponse(BaseModel):
|
||
llm_provider: str
|
||
llm_model: str
|
||
congress_poll_interval_minutes: int
|
||
newsapi_enabled: bool
|
||
pytrends_enabled: bool
|
||
|
||
|
||
# ── Collections ────────────────────────────────────────────────────────────────
|
||
|
||
class CollectionCreate(BaseModel):
|
||
name: str
|
||
is_public: bool = False
|
||
|
||
def validate_name(self) -> str:
|
||
name = self.name.strip()
|
||
if not 1 <= len(name) <= 100:
|
||
raise ValueError("name must be 1–100 characters")
|
||
return name
|
||
|
||
|
||
class CollectionUpdate(BaseModel):
|
||
name: Optional[str] = None
|
||
is_public: Optional[bool] = None
|
||
|
||
|
||
class CollectionSchema(BaseModel):
|
||
id: int
|
||
name: str
|
||
slug: str
|
||
is_public: bool
|
||
share_token: str
|
||
bill_count: int
|
||
created_at: datetime
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
class CollectionDetailSchema(CollectionSchema):
|
||
bills: list[BillSchema]
|
||
|
||
|
||
class BriefShareResponse(BaseModel):
|
||
brief: BriefSchema
|
||
bill: BillSchema
|