a918421945
- Remove unused imports (F401) across 17 files - Fix f-strings without placeholders (F541) in 3 files - Add noqa: E712 to SQLAlchemy == True comparisons (valid ORM pattern) - Preserve alembic side-effect import with noqa: F401 Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com> Agent-Logs-Url: https://github.com/christianlouis/pop_puller_to_gmail/sessions/99c3a25f-3479-473d-ac95-9faaa0ddd55b
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
"""Notification configuration endpoints"""
|
|
|
|
from typing import List
|
|
from fastapi import APIRouter, Depends, status
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy import select
|
|
|
|
from app.core.database import get_db
|
|
from app.core.deps import get_current_active_user
|
|
from app.models.database_models import User, NotificationConfig
|
|
from app.models.schemas import (
|
|
NotificationConfigCreate,
|
|
NotificationConfigResponse,
|
|
)
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post(
|
|
"", response_model=NotificationConfigResponse, status_code=status.HTTP_201_CREATED
|
|
)
|
|
async def create_notification_config(
|
|
config_in: NotificationConfigCreate,
|
|
current_user: User = Depends(get_current_active_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""Create notification configuration"""
|
|
config = NotificationConfig(user_id=current_user.id, **config_in.dict())
|
|
db.add(config)
|
|
await db.commit()
|
|
await db.refresh(config)
|
|
return config
|
|
|
|
|
|
@router.get("", response_model=List[NotificationConfigResponse])
|
|
async def list_notification_configs(
|
|
current_user: User = Depends(get_current_active_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""List all notification configurations"""
|
|
result = await db.execute(
|
|
select(NotificationConfig).where(NotificationConfig.user_id == current_user.id)
|
|
)
|
|
return result.scalars().all()
|