from django.contrib import admin
from .models import *
from .views import *

class EnvdataAdmin(admin.ModelAdmin):
    admin.site.site_header = "ENV后台管理"
    list_display = ('spot', 'errcode', 'rectime')
    list_filter = ('spot', 'errcode')
    fields = ('spot', 'errcode', 'electricity', 'fire', 'water', 'temperature', 'humidity', 'rectime')
    readonly_fields = ('spot', 'errcode', 'electricity', 'fire', 'water', 'temperature', 'humidity', 'rectime')
    list_per_page = 30

class AlarmAdmin(admin.ModelAdmin):
    list_display = ('spot', 'almname', 'almtime')
    list_filter = ('spot', 'almname')
    fields = ('spot', 'errcode', 'almname', 'almvalue', 'almtime')
    readonly_fields = ('spot', 'errcode', 'almname', 'almvalue', 'almtime')
    list_per_page = 30

class SpotAdmin(admin.ModelAdmin):
    list_display = ('name', 'ip')

class AlarmsetAdmin(admin.ModelAdmin):
    list_display = ('spot',)

class SmsAdmin(admin.ModelAdmin):
    list_display = ('name',)

class SendlogAdmin(admin.ModelAdmin):
    list_display = ('sendtype', 'sendresult', 'sendtime')
    readonly_fields = ('sendtype', 'sendresult', 'sendtime', 'sendto')

class ChartAdmin(admin.ModelAdmin):
    def changelist_view(self, request, extra_content=None):
        return chart_index(request)

class SysAdmin(admin.ModelAdmin):
    def changelist_view(self, request, extra_content=None):
        return sys_cmd(request)

admin.site.register(Spot, SpotAdmin)
admin.site.register(Envdata, EnvdataAdmin)
admin.site.register(Alarm, AlarmAdmin)
admin.site.register(Alarmset, AlarmsetAdmin)
admin.site.register(Sms, SmsAdmin)
admin.site.register(Sendlog, SendlogAdmin)
admin.site.register(Envchart, ChartAdmin)
admin.site.register(Sysmanager, SysAdmin)