146 lines
4.0 KiB
Python
146 lines
4.0 KiB
Python
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())
|
|
wiki_link = forms.CharField(required=False, 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",
|
|
]
|
|
|
|
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)
|