from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from django.contrib.auth.models import User

from .models import (
    Rower, Workout,GraphImage,FavoriteChart,SiteAnnouncement,
    Team,TeamInvite,TeamRequest,
    WorkoutComment,C2WorldClassAgePerformance,PlannedSession,
    GeoCourse,GeoPolygon,GeoPoint,VirtualRace,VirtualRaceResult,
    )

# Register your models here so you can use them in the Admin module

# Rower details directly under the User 
class RowerInline(admin.StackedInline):
    model = Rower
    can_delete = False
    verbose_name_plural = 'rower'
    filter_horizontal = ('team','friends')

#class UserAdmin(UserAdmin):
class UserAdmin(admin.ModelAdmin):
    inlines = (RowerInline,)
    list_display = ('username','email','first_name','last_name','rowerplan')
    search_fields = ["username","first_name","last_name","email"]

    def rowerplan(self, obj):
        return obj.rower.rowerplan

class WorkoutAdmin(admin.ModelAdmin):
    list_display = ('date','user','name','workouttype','boattype')

class FavoriteChartAdmin(admin.ModelAdmin):
    list_display = ('user','xparam','yparam1','yparam2','plottype','workouttype','reststrokes')

class C2WorldClassAgePerformanceAdmin(admin.ModelAdmin):
    list_display = ('sex','weightcategory','age','distance','power','name','season')
    
class SiteAnnouncementAdmin(admin.ModelAdmin):
    list_display = ('announcement','created','modified','expires','dotweet')

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

class TeamInviteAdmin(admin.ModelAdmin):
    list_display = ('issuedate','team','user','code')
    
class TeamRequestAdmin(admin.ModelAdmin):
    list_display = ('issuedate','team','user','code')

class WorkoutCommentAdmin(admin.ModelAdmin):
    list_display = ('created','user','workout')

class PlannedSessionAdmin(admin.ModelAdmin):
    list_display = ('name','startdate','enddate','manager','sessionvalue','sessionunit')

class GraphImageAdmin(admin.ModelAdmin):
    list_display = ('creationdatetime','workout','filename')

class GeoPolygonInline(admin.StackedInline):
    model = GeoPolygon


class GeoCourseAdmin(admin.ModelAdmin):
    list_display = ('manager','name')
    inlines = (GeoPolygonInline,)


class VirtualRaceAdmin(admin.ModelAdmin):
    list_display = ('manager','name','startdate','country')

class VirtualRaceResultAdmin(admin.ModelAdmin):
    list_display = ('race','userid','username','boattype','age','weightcategory')
    
admin.site.unregister(User)
admin.site.register(User,UserAdmin)
admin.site.register(Workout,WorkoutAdmin)
admin.site.register(GraphImage,GraphImageAdmin)
admin.site.register(Team,TeamAdmin)
admin.site.register(FavoriteChart,FavoriteChartAdmin)
admin.site.register(SiteAnnouncement,SiteAnnouncementAdmin)
admin.site.register(TeamInvite,TeamInviteAdmin)
admin.site.register(TeamRequest,TeamRequestAdmin)
admin.site.register(WorkoutComment,WorkoutCommentAdmin)
admin.site.register(C2WorldClassAgePerformance,
                    C2WorldClassAgePerformanceAdmin)
admin.site.register(PlannedSession,PlannedSessionAdmin)
admin.site.register(GeoCourse, GeoCourseAdmin)
admin.site.register(VirtualRace, VirtualRaceAdmin)
admin.site.register(VirtualRaceResult, VirtualRaceResultAdmin)