371 lines
13 KiB
Python
371 lines
13 KiB
Python
from django.shortcuts import get_object_or_404
|
||
from django.db.models import Q, Case, When, IntegerField
|
||
from django.utils.text import slugify
|
||
from django.utils import timezone
|
||
|
||
from ninja import Router, Query
|
||
from ninja.errors import HttpError
|
||
from typing import List, Optional
|
||
from uuid import UUID
|
||
|
||
from apps.events.api.schemas import (
|
||
EventAdminDetailSchema,
|
||
EventBriefSchema,
|
||
EventCreateSchema,
|
||
EventListSchema,
|
||
EventSchema,
|
||
EventUpdateSchema,
|
||
MyEventRegistrationOut,
|
||
PaginatedRegistrationSchema,
|
||
RegisterationDetailSchema,
|
||
RegistrationCreateSchema,
|
||
RegistrationSchema,
|
||
RegistrationStatusOut,
|
||
RegistrationStatusUpdateSchema,
|
||
)
|
||
from core.authentication import jwt_auth
|
||
from apps.events.models import Event, Registration
|
||
from apps.payments.models import DiscountCode
|
||
from core.api.schemas import ErrorSchema, MessageSchema
|
||
|
||
events_router = Router()
|
||
|
||
# Event endpoints
|
||
@events_router.get("/", response=List[EventListSchema])
|
||
def list_events(
|
||
request,
|
||
# status: Optional[str] = None,
|
||
status: Optional[List[str]] = Query(None),
|
||
event_type: Optional[str] = None,
|
||
search: Optional[str] = None,
|
||
limit: int = 20,
|
||
offset: int = 0
|
||
):
|
||
"""List events with filtering and pagination"""
|
||
queryset = Event.objects.filter(is_deleted=False).prefetch_related('gallery_images')
|
||
|
||
if status:
|
||
if "," in status:
|
||
parts = [s.strip() for s in status.split(",") if s.strip()]
|
||
queryset = queryset.filter(status__in=parts)
|
||
else:
|
||
queryset = queryset.filter(status__in=status)
|
||
if event_type:
|
||
queryset = queryset.filter(event_type=event_type)
|
||
if search:
|
||
queryset = queryset.filter(
|
||
Q(title__icontains=search) | Q(description__icontains=search)
|
||
)
|
||
|
||
queryset = queryset.annotate(
|
||
published_first=Case(
|
||
When(status='published', then=0),
|
||
default=1,
|
||
output_field=IntegerField()
|
||
)
|
||
).order_by('published_first', '-start_time', '-id')
|
||
|
||
events = queryset[offset:offset + limit]
|
||
return events
|
||
|
||
@events_router.get("/{int:event_id}", response=EventSchema)
|
||
def get_event(request, event_id: int):
|
||
"""Get event details by ID"""
|
||
event = get_object_or_404(
|
||
Event.objects.prefetch_related('gallery_images'),
|
||
id=event_id,
|
||
is_deleted=False
|
||
)
|
||
return event
|
||
|
||
@events_router.get("/slug/{str:slug}", response=EventSchema)
|
||
def get_event_by_slug(request, slug: str):
|
||
"""Get event details by slug"""
|
||
event = get_object_or_404(
|
||
Event.objects.prefetch_related('gallery_images'),
|
||
slug=slug,
|
||
is_deleted=False
|
||
)
|
||
return event
|
||
|
||
@events_router.post("/", response=EventSchema)
|
||
def create_event(request, payload: EventCreateSchema):
|
||
"""Create a new event"""
|
||
gallery_image_ids = payload.dict().pop('gallery_image_ids', [])
|
||
event = Event.objects.create(**payload.dict(exclude={'gallery_image_ids'}))
|
||
|
||
if gallery_image_ids:
|
||
event.gallery_images.set(gallery_image_ids)
|
||
|
||
return event
|
||
|
||
@events_router.put("/{int:event_id}", response=EventSchema)
|
||
def update_event(request, event_id: int, payload: EventUpdateSchema):
|
||
"""Update an existing event"""
|
||
event = get_object_or_404(Event, id=event_id, is_deleted=False)
|
||
|
||
update_data = payload.dict(exclude_unset=True)
|
||
gallery_image_ids = update_data.pop('gallery_image_ids', None)
|
||
|
||
for attr, value in update_data.items():
|
||
setattr(event, attr, value)
|
||
|
||
if 'title' in update_data:
|
||
event.slug = slugify(event.title)
|
||
|
||
event.save()
|
||
|
||
if gallery_image_ids is not None:
|
||
event.gallery_images.set(gallery_image_ids)
|
||
|
||
return event
|
||
|
||
@events_router.delete("/{int:event_id}", response=MessageSchema)
|
||
def delete_event(request, event_id: int):
|
||
"""Soft delete an event"""
|
||
event = get_object_or_404(Event, id=event_id, is_deleted=False)
|
||
event.delete()
|
||
return {"message": "Event deleted successfully"}
|
||
|
||
# Registration endpoints
|
||
@events_router.get("/{int:event_id}/registrations", response=List[RegistrationSchema])
|
||
def list_event_registrations(request, event_id: int, limit: int = 20, offset: int = 0):
|
||
"""List registrations for a specific event"""
|
||
event = get_object_or_404(Event, id=event_id, is_deleted=False)
|
||
queryset = event.registrations.filter(is_deleted=False).select_related('user')
|
||
|
||
registrations = queryset[offset:offset + limit]
|
||
return registrations
|
||
|
||
|
||
@events_router.get("/{int:event_id}/admin-registrations", response={200: PaginatedRegistrationSchema, 403: ErrorSchema}, auth=jwt_auth)
|
||
def list_event_registrations_admin(
|
||
request,
|
||
event_id: int,
|
||
status: Optional[List[str]] = Query(None),
|
||
university: Optional[str] = Query(None),
|
||
major: Optional[str] = Query(None),
|
||
search: Optional[str] = Query(None),
|
||
limit: int = Query(20, ge=1, le=200),
|
||
offset: int = Query(0, ge=0),
|
||
):
|
||
"""List registrations with filters for admin dashboard"""
|
||
user = request.auth
|
||
if not (user.is_staff or user.is_superuser):
|
||
return 403, {"error": "اجازه دسترسی ندارید."}
|
||
|
||
event = get_object_or_404(Event, id=event_id, is_deleted=False)
|
||
qs = (
|
||
event.registrations.filter(is_deleted=False)
|
||
.select_related("user")
|
||
.prefetch_related("payments__discount_code")
|
||
.order_by("-registered_at")
|
||
)
|
||
|
||
status_values = status or request.GET.getlist('status')
|
||
if status_values:
|
||
qs = qs.filter(status__in=status_values)
|
||
|
||
if university:
|
||
qs = qs.filter(
|
||
Q(user__university__code__icontains=university)
|
||
| Q(user__university__name__icontains=university)
|
||
)
|
||
|
||
if major:
|
||
qs = qs.filter(
|
||
Q(user__major__code__icontains=major)
|
||
| Q(user__major__name__icontains=major)
|
||
)
|
||
|
||
if search:
|
||
qs = qs.filter(
|
||
Q(user__username__icontains=search)
|
||
| Q(user__email__icontains=search)
|
||
| Q(user__first_name__icontains=search)
|
||
| Q(user__last_name__icontains=search)
|
||
)
|
||
|
||
total = qs.count()
|
||
results = qs[offset : offset + limit]
|
||
|
||
return PaginatedRegistrationSchema(count=total, next=None, previous=None, results=list(results))
|
||
|
||
@events_router.post(
|
||
"/{int:event_id}/register",
|
||
response=RegistrationSchema,
|
||
auth=jwt_auth,
|
||
)
|
||
def register_for_event(
|
||
request,
|
||
event_id: int,
|
||
payload: RegistrationCreateSchema | None = None,
|
||
):
|
||
"""Register current user for an event"""
|
||
event = get_object_or_404(Event, id=event_id, is_deleted=False)
|
||
user = request.auth
|
||
|
||
if Registration.objects.filter(event=event, user=user, status=Registration.StatusChoices.CONFIRMED).exists():
|
||
raise HttpError(400, "شما قبلا در این ایونت ثبتنام کردهاید.")
|
||
|
||
if event.registration_end_date and event.registration_end_date < timezone.now():
|
||
raise HttpError(400, "مهلت ثبتنام به پایان رسیدهاست")
|
||
|
||
if event.registration_start_date and event.registration_start_date > timezone.now():
|
||
raise HttpError(400, "زمان ثبتنام هنوز آغاز نشده است")
|
||
|
||
if not event.has_available_slots:
|
||
raise HttpError(400, "ظرفیت شرکتکنندگان تکمیل است")
|
||
|
||
# Create or get existing registration
|
||
discount_code = None
|
||
if payload and payload.discount_code:
|
||
discount_code = payload.discount_code
|
||
elif request.GET.get("discount_code"):
|
||
discount_code = request.GET.get("discount_code")
|
||
|
||
registration, created = Registration.objects.get_or_create(
|
||
event=event,
|
||
user=user,
|
||
status=Registration.StatusChoices.PENDING,
|
||
defaults={"final_price": event.price},
|
||
)
|
||
|
||
if registration.status == Registration.StatusChoices.CONFIRMED:
|
||
return HttpError(400, "شما قبلا در این ایونت ثبتنام کردهاید")
|
||
|
||
if registration.status == Registration.StatusChoices.CANCELLED:
|
||
registration = Registration.objects.create(
|
||
event=event,
|
||
user=user,
|
||
status=Registration.StatusChoices.PENDING,
|
||
final_price=event.price,
|
||
)
|
||
elif not created and registration.final_price is None:
|
||
registration.final_price = event.price
|
||
registration.save(update_fields=["final_price"])
|
||
|
||
applied_code = None
|
||
discount_amount = 0
|
||
final_price = event.price
|
||
fields_to_update = []
|
||
|
||
if discount_code:
|
||
applied_code = DiscountCode.objects.filter(
|
||
code=discount_code,
|
||
applicable_events=event,
|
||
is_active=True,
|
||
).first()
|
||
if not applied_code:
|
||
raise HttpError(400, "UcO_ O<>OrU?UOU? U.O1O<31>O\"O<EFBFBD> U+UOO3O<33>")
|
||
final_price, discount_amount = applied_code.calculate_discount(event, user)
|
||
registration.discount_code = applied_code
|
||
registration.discount_amount = discount_amount
|
||
fields_to_update.extend(["discount_code", "discount_amount"])
|
||
|
||
if registration.final_price != final_price:
|
||
registration.final_price = final_price
|
||
fields_to_update.append("final_price")
|
||
|
||
if not event.price or final_price == 0:
|
||
registration.status = Registration.StatusChoices.CONFIRMED
|
||
fields_to_update.append("status")
|
||
|
||
if fields_to_update:
|
||
registration.save(update_fields=list(set(fields_to_update)))
|
||
|
||
return registration
|
||
|
||
@events_router.put("/registrations/{int:registration_id}", response=RegistrationSchema, auth=jwt_auth)
|
||
def update_registration_status(request, registration_id: int, payload: RegistrationStatusUpdateSchema):
|
||
"""Update registration status"""
|
||
user = request.auth
|
||
|
||
registration = get_object_or_404(Registration, id=registration_id, user=user, is_deleted=False)
|
||
registration.status = payload.dict(exclude_unset=True).get('status')
|
||
registration.full_clean()
|
||
registration.save()
|
||
|
||
return registration
|
||
|
||
@events_router.delete("/registrations/{int:registration_id}", response=MessageSchema, auth=jwt_auth)
|
||
def cancel_registration(request, registration_id: int):
|
||
"""Cancel a registration"""
|
||
user = request.auth
|
||
|
||
registration = get_object_or_404(Registration, id=registration_id, user=user, is_deleted=False)
|
||
registration.delete()
|
||
return {"message": "ثبتنام شما لغو شد :("}
|
||
|
||
@events_router.get("/registerations/verify/{UUID:ticket_id}", response=RegisterationDetailSchema, auth=jwt_auth)
|
||
def verify_my_registration(request, ticket_id: UUID):
|
||
try:
|
||
reg = Registration.objects.select_related("event").get(ticket_id=ticket_id, user=request.auth)
|
||
return {
|
||
"event_image": request.build_absolute_uri(reg.event.featured_image.url) if reg.event.featured_image else None,
|
||
"event_title": reg.event.title,
|
||
"event_type": reg.event.get_event_type_display(),
|
||
"ticket_id": reg.ticket_id,
|
||
"status": reg.status,
|
||
"registered_at": reg.registered_at,
|
||
"success_markdown": reg.event.registration_success_markdown,
|
||
}
|
||
except Registration.DoesNotExist:
|
||
raise HttpError(404, "registration not found")
|
||
|
||
|
||
|
||
@events_router.get("/my-registrations", response=List[MyEventRegistrationOut], auth=jwt_auth)
|
||
def my_registrations(request):
|
||
qs = (
|
||
Registration.objects
|
||
.filter(user=request.auth)
|
||
.select_related("event")
|
||
.order_by("-created_at")
|
||
)
|
||
out: List[MyEventRegistrationOut] = []
|
||
for r in qs:
|
||
out.append(
|
||
MyEventRegistrationOut(
|
||
id=r.id,
|
||
created_at=r.created_at,
|
||
status=r.status,
|
||
event=EventBriefSchema(
|
||
id=r.event.id,
|
||
title=r.event.title,
|
||
slug=r.event.slug,
|
||
start_date=r.event.start_time,
|
||
end_date=r.event.end_time,
|
||
location=r.event.location,
|
||
price=r.event.price,
|
||
absolute_image_url=request.build_absolute_uri(r.event.featured_image.url) if r.event.featured_image else None,
|
||
),
|
||
)
|
||
)
|
||
return out
|
||
|
||
@events_router.get("/{event_id}/is-registered", response=RegistrationStatusOut, auth=jwt_auth)
|
||
def is_registered(request, event_id: int):
|
||
exists = Registration.objects.filter(
|
||
user=request.auth,
|
||
event_id=event_id,
|
||
status=Registration.StatusChoices.CONFIRMED
|
||
).exists()
|
||
return {"is_registered": exists}
|
||
@events_router.get("/{int:event_id}/admin-detail", response=EventAdminDetailSchema, auth=jwt_auth)
|
||
def event_admin_detail(request, event_id: int):
|
||
user = request.auth
|
||
if not (user.is_staff or user.is_superuser):
|
||
return 403, {"error": "اجازه دسترسی ندارید."}
|
||
|
||
event = get_object_or_404(
|
||
Event.objects.prefetch_related(
|
||
'gallery_images',
|
||
'registrations__user',
|
||
'registrations__payments__discount_code'
|
||
),
|
||
id=event_id,
|
||
is_deleted=False,
|
||
)
|
||
return event
|