bornhack-website/shop/admin.py

from django.contrib import admin

from . import models

admin.site.register(models.EpayCallback)
admin.site.register(models.EpayPayment)
admin.site.register(models.CoinifyAPIInvoice)
admin.site.register(models.CoinifyAPICallback)
admin.site.register(models.Invoice)
admin.site.register(models.CreditNote)
admin.site.register(models.CustomOrder)


@admin.register(models.ProductCategory)
class ProductCategoryAdmin(admin.ModelAdmin):
    list_display = [
        'name',
    ]


@admin.register(models.Product)
class ProductAdmin(admin.ModelAdmin):
    list_display = [
        'name',
        'category',
        'price',
        'available_in',
    ]


class ProductInline(admin.TabularInline):
    model = models.OrderProductRelation


class TicketInline(admin.TabularInline):
    model = models.Ticket
    exclude = ['qrcode_base64']


@admin.register(models.Order)
class OrderAdmin(admin.ModelAdmin):
    change_form_template = 'admin/change_order_form.html'
    readonly_fields = ('paid',)

    def get_email(self, obj):
        return obj.user.email

    list_display = [
        'id',
        'user',
        'get_email',
        'total',
        'payment_method',
        'open',
        'paid',
        'cancelled',
        'refunded',
    ]

    list_filter = [
        'camp',
        'payment_method',
        'open',
        'paid',
        'cancelled',
        'refunded',
        'user',
    ]

    exclude = ['products']

    inlines = [ProductInline, TicketInline]

    actions = ['mark_order_as_paid', 'mark_order_as_refunded']

    def mark_order_as_paid(self, request, queryset):
        for order in queryset.filter(paid=False):
            order.mark_as_paid()
    mark_order_as_paid.description = 'Mark order(s) as paid'

    def mark_order_as_refunded(self, request, queryset):
        for order in queryset.filter(refunded=False):
            order.mark_as_refunded(request)
    mark_order_as_refunded.description = 'Mark order(s) as refunded'


def get_user_email(obj):
    return obj.order.user.email


@admin.register(models.Ticket)
class TicketModelAdmin(admin.ModelAdmin):
    list_display = [
        'order',
        'product',
        'name',
        'email',
        get_user_email,
        'get_token',
        'checked_in'
    ]

    list_filter = ['product', 'checked_in']


    actions = ['mark_as_arrived']

    def mark_as_arrived(self, request, queryset):
        queryset.update(checked_in=True)