from django.contrib import admin
from django.http import HttpResponseRedirect
from .models import Profile, Racer, Product, Invoice, InvoiceProduct


@admin.register(Racer)
class RacerAdmin(admin.ModelAdmin):
    pass


@admin.register(Product)
class ProductAdmin(admin.ModelAdmin):
    pass


@admin.register(Invoice)
class InvoiceAdmin(admin.ModelAdmin):
    change_form_template = "invoice_custom_admin_page.html"

    def response_change(self, request, obj):
        if "_invoice" in request.POST:
            return HttpResponseRedirect("/api/invoice?invoice_id=" + obj.invoice_id)
        return super(self).response_change(request, obj)


@admin.register(InvoiceProduct)
class InvoiceProductAdmin(admin.ModelAdmin):
    pass


@admin.register(Profile)
class ProfileAdmin(admin.ModelAdmin):
    pass