mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-01-20 03:40:26 +00:00
6b69ec750e
- Add JWT authentication for cabinet users - Add Telegram WebApp authentication - Add subscription management endpoints - Add balance and transactions endpoints - Add referral system endpoints - Add tickets support for cabinet - Add webhooks and websocket for real-time updates - Add email verification service 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
146 lines
4.9 KiB
Python
146 lines
4.9 KiB
Python
"""Notification settings routes for cabinet."""
|
|
|
|
import logging
|
|
from datetime import datetime
|
|
from typing import Optional, Dict, Any
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from pydantic import BaseModel, Field
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.database.models import User
|
|
|
|
from ..dependencies import get_cabinet_db, get_current_cabinet_user
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
router = APIRouter(prefix="/notifications", tags=["Cabinet Notifications"])
|
|
|
|
|
|
# ============ Schemas ============
|
|
|
|
class NotificationSettingsResponse(BaseModel):
|
|
"""User notification settings."""
|
|
subscription_expiry_enabled: bool = True
|
|
subscription_expiry_days: int = 3
|
|
traffic_warning_enabled: bool = True
|
|
traffic_warning_percent: int = 80
|
|
balance_low_enabled: bool = True
|
|
balance_low_threshold: int = 100 # kopeks
|
|
news_enabled: bool = True
|
|
promo_offers_enabled: bool = True
|
|
|
|
|
|
class NotificationSettingsUpdate(BaseModel):
|
|
"""Update notification settings."""
|
|
subscription_expiry_enabled: Optional[bool] = None
|
|
subscription_expiry_days: Optional[int] = Field(None, ge=1, le=30)
|
|
traffic_warning_enabled: Optional[bool] = None
|
|
traffic_warning_percent: Optional[int] = Field(None, ge=50, le=99)
|
|
balance_low_enabled: Optional[bool] = None
|
|
balance_low_threshold: Optional[int] = Field(None, ge=0)
|
|
news_enabled: Optional[bool] = None
|
|
promo_offers_enabled: Optional[bool] = None
|
|
|
|
|
|
# ============ Helpers ============
|
|
|
|
def _get_notification_settings(user: User) -> Dict[str, Any]:
|
|
"""Get notification settings from user object."""
|
|
# Try to get from user's settings field or use defaults
|
|
settings_data = getattr(user, 'notification_settings', None) or {}
|
|
|
|
return {
|
|
"subscription_expiry_enabled": settings_data.get("subscription_expiry_enabled", True),
|
|
"subscription_expiry_days": settings_data.get("subscription_expiry_days", 3),
|
|
"traffic_warning_enabled": settings_data.get("traffic_warning_enabled", True),
|
|
"traffic_warning_percent": settings_data.get("traffic_warning_percent", 80),
|
|
"balance_low_enabled": settings_data.get("balance_low_enabled", True),
|
|
"balance_low_threshold": settings_data.get("balance_low_threshold", 100),
|
|
"news_enabled": settings_data.get("news_enabled", True),
|
|
"promo_offers_enabled": settings_data.get("promo_offers_enabled", True),
|
|
}
|
|
|
|
|
|
def _update_notification_settings(user: User, updates: Dict[str, Any]) -> Dict[str, Any]:
|
|
"""Update notification settings on user object."""
|
|
current_settings = _get_notification_settings(user)
|
|
|
|
for key, value in updates.items():
|
|
if value is not None:
|
|
current_settings[key] = value
|
|
|
|
return current_settings
|
|
|
|
|
|
# ============ Routes ============
|
|
|
|
@router.get("", response_model=NotificationSettingsResponse)
|
|
async def get_notification_settings(
|
|
user: User = Depends(get_current_cabinet_user),
|
|
):
|
|
"""Get user's notification settings."""
|
|
settings = _get_notification_settings(user)
|
|
return NotificationSettingsResponse(**settings)
|
|
|
|
|
|
@router.patch("", response_model=NotificationSettingsResponse)
|
|
async def update_notification_settings(
|
|
request: NotificationSettingsUpdate,
|
|
user: User = Depends(get_current_cabinet_user),
|
|
db: AsyncSession = Depends(get_cabinet_db),
|
|
):
|
|
"""Update user's notification settings."""
|
|
updates = request.model_dump(exclude_unset=True)
|
|
|
|
if not updates:
|
|
# No updates provided, return current settings
|
|
settings = _get_notification_settings(user)
|
|
return NotificationSettingsResponse(**settings)
|
|
|
|
# Update settings
|
|
new_settings = _update_notification_settings(user, updates)
|
|
|
|
# Store in user object
|
|
if not hasattr(user, 'notification_settings') or user.notification_settings is None:
|
|
user.notification_settings = {}
|
|
|
|
user.notification_settings = new_settings
|
|
user.updated_at = datetime.utcnow()
|
|
|
|
await db.commit()
|
|
await db.refresh(user)
|
|
|
|
return NotificationSettingsResponse(**new_settings)
|
|
|
|
|
|
@router.post("/test")
|
|
async def send_test_notification(
|
|
user: User = Depends(get_current_cabinet_user),
|
|
):
|
|
"""Send a test notification to the user."""
|
|
# This would typically trigger a notification via Telegram bot
|
|
# For now, just return success
|
|
return {
|
|
"success": True,
|
|
"message": "Test notification request received. You will receive a test message shortly.",
|
|
}
|
|
|
|
|
|
@router.get("/history")
|
|
async def get_notification_history(
|
|
limit: int = 20,
|
|
offset: int = 0,
|
|
user: User = Depends(get_current_cabinet_user),
|
|
db: AsyncSession = Depends(get_cabinet_db),
|
|
):
|
|
"""Get user's notification history."""
|
|
# For now, return empty list - notification history can be implemented later
|
|
# when there's a notification log table
|
|
return {
|
|
"notifications": [],
|
|
"total": 0,
|
|
"limit": limit,
|
|
"offset": offset,
|
|
}
|