bornhack-website/src/tickets/admin.py

from django.contrib import admin

from shop.models import OrderProductRelation

from .models import DiscountTicket, ShopTicket, SponsorTicket, TicketType


class BaseTicketAdmin(admin.ModelAdmin):
    actions = ["generate_pdf"]
    exclude = ["qrcode_base64"]
    readonly_fields = ["token", "badge_token"]

    def generate_pdf(self, request, queryset):
        for ticket in queryset.all():
            ticket.generate_pdf()

    generate_pdf.description = "Generate PDF for the ticket"


@admin.register(TicketType)
class TicketTypeAdmin(admin.ModelAdmin):
    list_display = ["name", "camp"]

    list_filter = ["name", "camp"]


@admin.register(SponsorTicket)
class SponsorTicketAdmin(BaseTicketAdmin):
    list_display = ["pk", "ticket_type", "sponsor", "used"]

    list_filter = ["ticket_type__camp", "used", "ticket_type", "sponsor"]

    search_fields = ["pk", "sponsor__name"]


@admin.register(DiscountTicket)
class DiscountTicketAdmin(BaseTicketAdmin):
    pass


@admin.register(ShopTicket)
class ShopTicketAdmin(BaseTicketAdmin):
    def user_email(self, obj):
        return obj.order.user.email

    def is_paid(self, obj):
        return obj.order.paid

    list_display = [
        "pk",
        "user_email",
        "is_paid",
        "ticket_type",
        "order",
        "product",
        "used",
        "product_quantity",
    ]

    list_filter = ["ticket_type__camp", "used", "ticket_type", "order", "product"]

    search_fields = ["uuid", "order__id", "order__user__email", "name", "email"]

    def product_quantity(self, ticket):
        orp = OrderProductRelation.objects.get(
            product=ticket.product, order=ticket.order
        )

        return (
            str(orp.quantity) if ticket.ticket_type.single_ticket_per_product else "1"
        )

    product_quantity.short_description = "Quantity"


class ShopTicketInline(admin.TabularInline):
    model = ShopTicket