from django.contrib import admin
from .models import Keywords, Configs, UserRole
from .named_enum import KeywordsType
from apps.myadmin.admin import admin_site


# Register your models here.
class KeywordsAdmin(admin.ModelAdmin):
    list_display = ('keyword', 'type_verbose_name', 'on_off')
    search_fields = ('keyword',)
    list_filter = ('type', 'on_off',)

    def type_verbose_name(self, obj):
        return KeywordsType.get_verbose_name(obj.type)
    type_verbose_name.short_description = '类型'


# Register your models here.
class ConfigsAdmin(admin.ModelAdmin):
    list_display = ('id', 'value', 'comment')


# Register your models here.
class UserRoleAdmin(admin.ModelAdmin):
    list_display = ('id', 'auth_user_id', 'role')


admin_site.register(Keywords, KeywordsAdmin)
admin_site.register(Configs, ConfigsAdmin)
admin_site.register(UserRole, UserRoleAdmin)
# admin.site.site_header = '宝马OCR'
# admin.site.site_title = '宝马OCR'