Skip to content

Instantly share code, notes, and snippets.

@adorsk
Forked from wolever/reverseadmin.py
Created April 13, 2016 02:41
Show Gist options
  • Save adorsk/0abb63a4acc8d025b716a03acdbecf85 to your computer and use it in GitHub Desktop.
Save adorsk/0abb63a4acc8d025b716a03acdbecf85 to your computer and use it in GitHub Desktop.
Forked from https://gist.github.com/mzbyszewska/8b6afc312b024832aa85 , updated to work with Django 1.8
'''
adminreverse from here http://djangosnippets.org/snippets/2032/
changed for working with ForeignKeys
'''
'''
reverseadmin
============
Module that makes django admin handle OneToOneFields in a better way.
A common use case for one-to-one relationships is to "embed" a model
inside another one. For example, a Person may have multiple foreign
keys pointing to an Address entity, one home address, one business
address and so on. Django admin displays those relations using select
boxes, letting the user choose which address entity to connect to a
person. A more natural way to handle the relationship is using
inlines. However, since the foreign key is placed on the owning
entity, django admins standard inline classes can't be used. Which is
why I created this module that implements "reverse inlines" for this
use case.
Example:
from django.db import models
class Address(models.Model):
street = models.CharField(max_length = 255)
zipcode = models.CharField(max_length = 10)
city = models.CharField(max_length = 255)
class Person(models.Model):
name = models.CharField(max_length = 255)
business_addr = models.ForeignKey(Address,
related_name = 'business_addr')
home_addr = models.OneToOneField(Address, related_name = 'home_addr')
other_addr = models.OneToOneField(Address, related_name = 'other_addr')
You use reverseadmin in the following way:
from django.contrib import admin
from django.db import models
from models import Person
from reverseadmin import ReverseModelAdmin
class AddressForm(models.Form):
pass
class PersonAdmin(ReverseModelAdmin):
inline_type = 'tabular'
inline_reverse = ('business_addr', ('home_addr', AddressForm), ('other_addr' (
'form': OtherForm
'exclude': ()
)))
admin.site.register(Person, PersonAdmin)
inline_type can be either "tabular" or "stacked" for tabular and
stacked inlines respectively.
The module is designed to work with Django 1.1.1. Since it hooks into
the internals of the admin package, it may not work with later Django
versions.
'''
# -*- coding: utf-8 -*-
from django.contrib.admin import ModelAdmin
from django.contrib.admin.options import InlineModelAdmin
from django.contrib.admin.util import flatten_fieldsets
from django.db.models import OneToOneField, ForeignKey
from django.forms import ModelForm
from django.forms.models import BaseModelFormSet, modelformset_factory
from django.utils.functional import curry
from django.core.exceptions import ObjectDoesNotExist, ValidationError
class ReverseInlineFormSet(BaseModelFormSet):
model = None
parent_fk_name = ''
def __init__(self,
data = None,
files = None,
instance = None,
prefix = None,
queryset = None,
save_as_new = False):
self.mem_instance = instance
# sometimes object does not exist
try:
object = getattr(instance, self.parent_fk_name)
except ObjectDoesNotExist:
object = None
if object:
qs = self.model.objects.filter(pk = object.id)
else:
qs = self.model.objects.none()
self.extra = 1
super(ReverseInlineFormSet, self).__init__(data, files,
prefix = prefix,
queryset = qs)
for form in self.forms:
parent_related = getattr(instance.__class__, self.parent_fk_name)
if not parent_related.field.null:
form.empty_permitted = False
def save_new_objects(self,commit):
"""
The save_new_objects method is overwritten since the id of the new objects has to be
passed to the parent object. Otherwise the new objects will not be connected
to theirs "should be parent object".
"""
super(ReverseInlineFormSet, self).save_new_objects(commit)
for dream in self.new_objects:
setattr(self.mem_instance, self.parent_fk_name, dream)
return self.new_objects
def reverse_inlineformset_factory(parent_model,
model,
parent_fk_name,
form = ModelForm,
fields = None,
exclude = None,
formfield_callback = lambda f: f.formfield()):
kwargs = {
'form': form,
'formfield_callback': formfield_callback,
'formset': ReverseInlineFormSet,
'extra': 0,
'can_delete': False,
'can_order': False,
'fields': fields,
'exclude': exclude or [],
'max_num': 1,
}
FormSet = modelformset_factory(model, **kwargs)
FormSet.parent_fk_name = parent_fk_name
FormSet.is_reverse_inline = True
return FormSet
class ReverseInlineModelAdmin(InlineModelAdmin):
'''
Use the name and the help_text of the owning models field to
render the verbose_name and verbose_name_plural texts.
'''
def __init__(self,
parent_model,
parent_fk_name,
model, admin_site,
inline_type):
self.template = 'admin/edit_inline/%s.html' % inline_type
self.parent_fk_name = parent_fk_name
self.model = model
field_descriptor = getattr(parent_model, self.parent_fk_name)
field = field_descriptor.field
self.verbose_name_plural = field.verbose_name.title()
self.verbose_name = field.help_text
if not self.verbose_name:
self.verbose_name = self.verbose_name_plural
super(ReverseInlineModelAdmin, self).__init__(parent_model, admin_site)
def get_formset(self, request, obj = None, **kwargs):
if self.declared_fieldsets:
fields = flatten_fieldsets(self.declared_fieldsets)
else:
fields = None
if self.exclude is None:
exclude = []
else:
exclude = list(self.exclude)
# if exclude is an empty list we use None, since that's the actual
# default
exclude = (exclude + kwargs.get("exclude", [])) or None
defaults = {
"form": self.form,
"fields": fields,
"exclude": exclude,
"formfield_callback": curry(self.formfield_for_dbfield, request=request),
}
defaults.update(kwargs)
return reverse_inlineformset_factory(self.parent_model,
self.model,
self.parent_fk_name,
**defaults)
class ReverseModelAdmin(ModelAdmin):
'''
Patched ModelAdmin class.
'''
def __init__(self, model, admin_site):
super(ReverseModelAdmin, self).__init__(model, admin_site)
if self.exclude is None:
self.exclude = []
inline_instances = []
for field_name in self.inline_reverse:
kwargs = {}
if isinstance(field_name, tuple):
if isinstance(field_name[1], dict):
kwargs = field_name[1]
elif isinstance(field_name[1], ModelForm):
kwargs['form'] = field_name[1]
field_name = field_name[0]
field = model._meta.get_field(field_name)
if isinstance(field, (OneToOneField, ForeignKey)):
name = field.name
parent = field.related_model
inline = ReverseInlineModelAdmin(model,
name,
parent,
admin_site,
self.inline_type)
if kwargs:
inline.__dict__.update(kwargs)
inline_instances.append(inline)
self.exclude.append(name)
self.reverse_inlines = inline_instances
self.inline_instances = self.reverse_inlines + [
inline_class(self.model, self.admin_site)
for inline_class in self.inlines
]
def get_inline_instances(self, request, *args, **kwargs):
"""
ModelAdmin's method is overriten here for setting inline instances the way we need in inverse inlines.
It is necessary for compatibility with changes in Django 1.4
"""
return self.inline_instances
def save_model(self, request, obj, form, change):
self._save_model_args = (request, obj, form, change)
def save_related(self, request, form, formsets, change):
for formset in formsets:
if getattr(formset, "is_reverse_inline", False):
self.save_formset(request, form, formset, change=change)
super(ReverseModelAdmin, self).save_model(*self._save_model_args)
form.save_m2m()
for formset in formsets:
if not getattr(formset, "is_reverse_inline", False):
self.save_formset(request, form, formset, change=change)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment