Cloud-CV/EvalAI

from django.contrib import admin

from base.admin import ImportExportTimeStampedAdmin

from .models import ChallengeHost, ChallengeHostTeam


@admin.register(ChallengeHostTeam)
class ChallengeHostTeamModelAdmin(ImportExportTimeStampedAdmin):
    raw_id_fields = ["created_by"]
    list_display = ("id", "team_name", "created_by", "team_url")
    search_fields = ("id", "team_name", "created_by__username")


@admin.register(ChallengeHost)
class ChallengeHostAdmin(ImportExportTimeStampedAdmin):
    raw_id_fields = ["user", "team_name"]
    list_display = ("id", "user", "team_name", "status", "permissions")
    list_filter = ("status", "permissions")
    search_fields = ("id", "user__username", "team_name__team_name")