Felipe Cardoso
fe2bcbd6e7
Introduce methods to count user, public, and upcoming events to enhance CRUD functionality for events. Additionally, add a `PaginatedResponse` schema to simplify and standardize paginated API responses. These updates support improved data querying and response handling.
173 lines
4.6 KiB
Python
173 lines
4.6 KiB
Python
from datetime import timezone
|
|
from typing import List, Optional, Union, Dict, Any
|
|
from uuid import UUID
|
|
|
|
from sqlalchemy import desc, func
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.crud.base import CRUDBase
|
|
from app.models.event import Event
|
|
from app.schemas.events import EventCreate, EventUpdate
|
|
|
|
|
|
class CRUDEvent(CRUDBase[Event, EventCreate, EventUpdate]):
|
|
def get_by_slug(self, db: Session, *, slug: str) -> Optional[Event]:
|
|
"""Get event by slug."""
|
|
return db.query(Event).filter(Event.slug == slug).first()
|
|
|
|
def get_user_events(
|
|
self,
|
|
db: Session,
|
|
*,
|
|
user_id: UUID,
|
|
skip: int = 0,
|
|
limit: int = 100,
|
|
include_inactive: bool = False
|
|
) -> List[Event]:
|
|
"""Get all events created by a specific user."""
|
|
query = db.query(Event).filter(Event.created_by == user_id)
|
|
|
|
if not include_inactive:
|
|
query = query.filter(Event.is_active == True)
|
|
|
|
return query.order_by(desc(Event.event_date)).offset(skip).limit(limit).all()
|
|
|
|
def count_user_events(
|
|
self,
|
|
db: Session,
|
|
*,
|
|
user_id: UUID,
|
|
include_inactive: bool = False
|
|
) -> int:
|
|
"""Count total events created by a specific user."""
|
|
query = db.query(func.count(Event.id)).filter(Event.created_by == user_id)
|
|
|
|
if not include_inactive:
|
|
query = query.filter(Event.is_active == True)
|
|
|
|
return query.scalar()
|
|
|
|
def create_with_owner(
|
|
self,
|
|
db: Session,
|
|
*,
|
|
obj_in: EventCreate,
|
|
owner_id: UUID
|
|
) -> Event:
|
|
"""Create a new event with owner ID."""
|
|
obj_in_data = obj_in.model_dump()
|
|
db_obj = Event(**obj_in_data, created_by=owner_id)
|
|
db.add(db_obj)
|
|
db.commit()
|
|
db.refresh(db_obj)
|
|
return db_obj
|
|
|
|
def update(
|
|
self,
|
|
db: Session,
|
|
*,
|
|
db_obj: Event,
|
|
obj_in: Union[EventUpdate, Dict[str, Any]]
|
|
) -> Event:
|
|
"""Update an event."""
|
|
if isinstance(obj_in, dict):
|
|
update_data = obj_in
|
|
else:
|
|
update_data = obj_in.model_dump(exclude_unset=True)
|
|
|
|
return super().update(db, db_obj=db_obj, obj_in=update_data)
|
|
|
|
def get_upcoming_events(
|
|
self,
|
|
db: Session,
|
|
*,
|
|
skip: int = 0,
|
|
limit: int = 100
|
|
) -> List[Event]:
|
|
"""Get upcoming active events ordered by date."""
|
|
from datetime import datetime
|
|
|
|
now = datetime.now(tz=timezone.utc)
|
|
return (
|
|
db.query(Event)
|
|
.filter(Event.is_active == True)
|
|
.filter(Event.event_date >= now)
|
|
.order_by(Event.event_date)
|
|
.offset(skip)
|
|
.limit(limit)
|
|
.all()
|
|
)
|
|
|
|
def count_upcoming_events(
|
|
self,
|
|
db: Session
|
|
) -> int:
|
|
"""Count total upcoming events."""
|
|
from datetime import datetime
|
|
|
|
now = datetime.now(tz=timezone.utc)
|
|
return (
|
|
db.query(func.count(Event.id))
|
|
.filter(Event.is_active == True)
|
|
.filter(Event.event_date >= now)
|
|
.scalar()
|
|
)
|
|
|
|
def get_public_events(
|
|
self,
|
|
db: Session,
|
|
*,
|
|
skip: int = 0,
|
|
limit: int = 100
|
|
) -> List[Event]:
|
|
"""Get public active events."""
|
|
return (
|
|
db.query(Event)
|
|
.filter(Event.is_active == True)
|
|
.filter(Event.is_public == True)
|
|
.order_by(Event.event_date)
|
|
.offset(skip)
|
|
.limit(limit)
|
|
.all()
|
|
)
|
|
|
|
def count_public_events(
|
|
self,
|
|
db: Session
|
|
) -> int:
|
|
"""Count total public events."""
|
|
return (
|
|
db.query(func.count(Event.id))
|
|
.filter(Event.is_active == True)
|
|
.filter(Event.is_public == True)
|
|
.scalar()
|
|
)
|
|
|
|
def get_public_event(
|
|
self,
|
|
db: Session,
|
|
*,
|
|
slug: str,
|
|
access_code: Optional[str] = None
|
|
) -> Optional[Event]:
|
|
"""Get a public event by slug, optionally checking access code."""
|
|
query = db.query(Event).filter(
|
|
Event.slug == slug,
|
|
Event.is_active == True
|
|
)
|
|
|
|
event = query.first()
|
|
if not event:
|
|
return None
|
|
|
|
if event.is_public:
|
|
return event
|
|
|
|
if access_code and event.access_code == access_code:
|
|
return event
|
|
|
|
return None
|
|
|
|
|
|
# Create a singleton instance for use across the application
|
|
event = CRUDEvent(Event) |