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.register(models.Menu)
class Menuadmin(admin.ModelAdmin):
    list_display = ('menuname','menucode','type','url')
    list_display_links = ('menuname',)
    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)