88 lines
2.4 KiB
Python
88 lines
2.4 KiB
Python
from functools import update_wrapper
|
|
|
|
from django.contrib import admin
|
|
from django.db import models
|
|
from django.http import HttpResponseRedirect
|
|
from django.urls import reverse
|
|
|
|
|
|
class Action(models.Model):
|
|
name = models.CharField(max_length=50, primary_key=True)
|
|
description = models.CharField(max_length=70)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class ActionAdmin(admin.ModelAdmin):
|
|
"""
|
|
A ModelAdmin for the Action model that changes the URL of the add_view
|
|
to '<app name>/<model name>/!add/'
|
|
The Action model has a CharField PK.
|
|
"""
|
|
|
|
list_display = ("name", "description")
|
|
|
|
def remove_url(self, name):
|
|
"""
|
|
Remove all entries named 'name' from the ModelAdmin instance URL
|
|
patterns list
|
|
"""
|
|
return [url for url in super().get_urls() if url.name != name]
|
|
|
|
def get_urls(self):
|
|
# Add the URL of our custom 'add_view' view to the front of the URLs
|
|
# list. Remove the existing one(s) first
|
|
from django.urls import re_path
|
|
|
|
def wrap(view):
|
|
def wrapper(*args, **kwargs):
|
|
return self.admin_site.admin_view(view)(*args, **kwargs)
|
|
|
|
return update_wrapper(wrapper, view)
|
|
|
|
info = self.opts.app_label, self.opts.model_name
|
|
|
|
view_name = "%s_%s_add" % info
|
|
|
|
return [
|
|
re_path("^!add/$", wrap(self.add_view), name=view_name),
|
|
] + self.remove_url(view_name)
|
|
|
|
|
|
class Person(models.Model):
|
|
name = models.CharField(max_length=20)
|
|
|
|
|
|
class PersonAdmin(admin.ModelAdmin):
|
|
def response_post_save_add(self, request, obj):
|
|
return HttpResponseRedirect(
|
|
reverse("admin:admin_custom_urls_person_history", args=[obj.pk])
|
|
)
|
|
|
|
def response_post_save_change(self, request, obj):
|
|
return HttpResponseRedirect(
|
|
reverse("admin:admin_custom_urls_person_delete", args=[obj.pk])
|
|
)
|
|
|
|
|
|
class Car(models.Model):
|
|
name = models.CharField(max_length=20)
|
|
|
|
|
|
class CarAdmin(admin.ModelAdmin):
|
|
def response_add(self, request, obj, post_url_continue=None):
|
|
return super().response_add(
|
|
request,
|
|
obj,
|
|
post_url_continue=reverse(
|
|
"admin:admin_custom_urls_car_history", args=[obj.pk]
|
|
),
|
|
)
|
|
|
|
|
|
site = admin.AdminSite(name="admin_custom_urls")
|
|
site.register(Action, ActionAdmin)
|
|
site.register(Person, PersonAdmin)
|
|
site.register(Car, CarAdmin)
|