from django.contrib import admin
from apps.admin import my_admin_site
from .models import Keywords, Configs
from .named_enum import KeywordsType


# 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')


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

my_admin_site.register(Keywords, KeywordsAdmin)
my_admin_site.register(Configs, ConfigsAdmin)