spejstore-new/storage/admin.py

from django import forms
from django.contrib import admin

from django_select2.forms import ModelSelect2Widget, Select2MultipleWidget


from .models import Item, ItemImage, Category, Label
from .widgets import ItemSelectWidget, PropsSelectWidget


class ModelAdminMixin(object):
    def has_add_permission(self, request, obj=None):
        return request.user.is_authenticated()

    has_change_permission = has_add_permission
    has_delete_permission = has_add_permission
    has_module_permission = has_add_permission


class ItemForm(forms.ModelForm):
    name = forms.CharField(widget=forms.TextInput())

    class Meta:
        model = Item
        exclude = []
        widgets = {
            "parent": ItemSelectWidget(model=Item),
            "categories": Select2MultipleWidget,
            "props": PropsSelectWidget,
        }


class ItemImageInline(ModelAdminMixin, admin.TabularInline):
    model = ItemImage
    extra = 1


class LabelInline(ModelAdminMixin, admin.TabularInline):
    model = Label


class ItemAdmin(ModelAdminMixin, admin.ModelAdmin):
    list_display = ("_name",)
    list_filter = ("categories",)
    form = ItemForm
    inlines = [ItemImageInline, LabelInline]
    save_on_top = True

    def _name(self, obj):
        return "-" * obj.get_level() + "> " + obj.name

    def save_model(self, request, obj, form, change):
        super(ItemAdmin, self).save_model(request, obj, form, change)

        # Store last input parent to use as default on next creation
        if obj.parent:
            request.session["last-parent"] = str(obj.parent.uuid)
        else:
            request.session["last-parent"] = str(obj.uuid)

    def get_changeform_initial_data(self, request):
        data = {
            "parent": request.GET.get("parent") or request.session.get("last-parent")
        }

        data.update(super(ItemAdmin, self).get_changeform_initial_data(request))
        return data

    class Media:
        js = (
            # Required by select2
            "https://ajax.googleapis.com/ajax/libs/jquery/1.9.1/jquery.min.js",
        )
        css = {"all": ("css/admin.css",)}

    def response_action(self, request, queryset):
        with Item.disabled_tree_trigger():
            return super(ItemAdmin, self).response_action(request, queryset)


class NormalModelAdmin(ModelAdminMixin, admin.ModelAdmin):
    pass


admin.site.site_title = "Hackerspace Storage Admin"
admin.site.site_header = "Hackerspace Storage Admin"

admin.site.register(Item, ItemAdmin)
admin.site.register(Category, NormalModelAdmin)

from django.contrib.auth.models import User
from django.contrib.auth.models import Group

User.add_to_class("get_short_name", User.get_username)
User.add_to_class("get_full_name", User.get_username)

admin.site.unregister(User)
admin.site.unregister(Group)

from social_django.admin import UserSocialAuth, Nonce, Association

admin.site.unregister(UserSocialAuth)
admin.site.unregister(Nonce)
admin.site.unregister(Association)