from django.contrib import admin

# Register your models here.
from squad_battle.models import SquadBattleServerSideHandlerWorker, AccountSquadBattleData, FifaAccountSquadGame, \
    FifaAccountMomentsGame


class FifaAccountSquadGameAdmin(admin.ModelAdmin):
    search_fields = ['fifa_account__user_name__icontains']
    list_display = ['fifa_account', 'create_time', 'end_time', 'game_number', 'difficulty', 'enemy_team_rate',
                    'win_status']
    autocomplete_fields = ['fifa_account']


class SquadBattleServerSideHandlerWorkerAdmin(admin.ModelAdmin):
    list_display = ['create_time', 'fifa_account', 'is_done', 'has_error', 'status']
    autocomplete_fields = ['fifa_account', 'sbc_worker']
    search_fields = ['fifa_account__user_name']


class AccountSquadBattleDataAdmin(admin.ModelAdmin):
    list_display = ['fifa_account', 'event_id', 'score', 'user_tier_level', 'games_played']
    autocomplete_fields = ['fifa_account']
    search_fields = ['fifa_account__user_name']


class FifaAccountMomentsGameAdmin(admin.ModelAdmin):
    search_fields = ['fifa_account__user_name__icontains']
    list_display = ['fifa_account', 'create_time', 'end_time', 'moment_number', 'win_status', 'need_fix_squad']
    autocomplete_fields = ['fifa_account']
    list_filter = ['need_fix_squad', 'win_status']

admin.site.register(FifaAccountSquadGame, FifaAccountSquadGameAdmin)
admin.site.register(SquadBattleServerSideHandlerWorker, SquadBattleServerSideHandlerWorkerAdmin)
admin.site.register(AccountSquadBattleData, AccountSquadBattleDataAdmin)
admin.site.register(FifaAccountMomentsGame, FifaAccountMomentsGameAdmin)