diff --git a/alkatorapi/admin.py b/alkatorapi/admin.py index 41a4e09..d621ddb 100644 --- a/alkatorapi/admin.py +++ b/alkatorapi/admin.py @@ -1,5 +1,13 @@ from django.contrib import admin +from django.http import HttpResponseRedirect from .models import User -admin.site.register(User) +@admin.register(User) +class UserAdmin(admin.ModelAdmin): + change_form_template = "invoice_custom_admin_page.html" + + def response_change(self, request, obj): + if "_invoice" in request.POST: + return HttpResponseRedirect(f"/api/invoice?invoice_id={obj.invoice_id}") + return super().response_change(request, obj) \ No newline at end of file diff --git a/alkatorapi/models.py b/alkatorapi/models.py index 603a8ca..ad27b25 100644 --- a/alkatorapi/models.py +++ b/alkatorapi/models.py @@ -1,4 +1,5 @@ from django.db import models +from django.contrib import admin ALKATOR_CHOICES = ( (1, "Alkátor"), diff --git a/alkatorapi/templates/invoice_custom_admin_page.html b/alkatorapi/templates/invoice_custom_admin_page.html new file mode 100644 index 0000000..dd36454 --- /dev/null +++ b/alkatorapi/templates/invoice_custom_admin_page.html @@ -0,0 +1,8 @@ +{% extends 'admin/change_form.html' %} + +{% block submit_buttons_bottom %} + {{ block.super }} +