spejstore-new/storage/admin.py

from django import forms
from django.contrib import admin


from .models import Item, ItemImage, Category, StaffProxyModel

from .widgets import 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 = {
            "props": PropsSelectWidget,
        }


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


class ItemAdmin(ModelAdminMixin, admin.ModelAdmin):
    list_display = ("_name",)
    list_filter = ("categories",)
    form = ItemForm
    inlines = [ItemImageInline]
    save_on_top = True
    autocomplete_fields = ["parent", "owner", "taken_by", "categories"]
    search_fields = ["name", "owner__name", "taken_by__name"]

    def formfield_for_foreignkey(self, db_field, request, **kwargs):
        formfield = super().formfield_for_foreignkey(db_field, request, **kwargs)
        if db_field.name == "owner":
            formfield.queryset = formfield.queryset.order_by("username")
        return formfield

    def _name(self, obj):
        return ("-" * (obj.get_level() or 0)) + "> " + 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:
        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):
    search_fields = ["name"]
    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)


from django.contrib.auth.admin import UserAdmin


class StaffProxyModelAdmin(UserAdmin):
    def has_module_permission(self, request):
        return request.user.is_superuser

    def has_add_permission(self, request, obj=None):
        return request.user.is_superuser

    def __has_view_permission(self, request, obj=None):
        return True

    has_view_permission = __has_view_permission

    has_change_permission = has_add_permission
    has_delete_permission = has_add_permission
    has_module_permission = has_add_permission

    def get_form(self, request, obj=None, **kwargs):
        form = super().get_form(request, obj, **kwargs)
        is_superuser = request.user.is_superuser

        if not is_superuser:
            for f in form.base_fields:
                form.base_fields[f].disabled = True

        return form


# admin.site.register(StaffProxyModel, StaffProxyModelAdmin)

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)

admin.site.register(StaffProxyModel, StaffProxyModelAdmin)