safesite/safesite/admin.py

19 lines
809 B
Python

from django.contrib import admin
from . import models
# Register your models here.
admin.site.register(models.User)
admin.site.register(models.Trouble)
admin.site.register(models.Dicclass)
admin.site.register(models.Dickey)
admin.site.register(models.Partment)
admin.site.register(models.MySchedule)
admin.site.register(models.GridTaskSet)
@admin.register(models.Menu)
class Menuadmin(admin.ModelAdmin):
list_display = ('menuname','menucode','type','url', 'sortnum')
list_display_links = ('menuname',)
search_fields = ('menuname', 'menucode',)
def formfield_for_foreignkey(self, db_field, request, **kwargs):
if db_field.name == "parentid":
kwargs["queryset"] = models.Menu.objects.filter(type=1)
return super().formfield_for_foreignkey(db_field, request, **kwargs)