from django.contrib import admin

from .models import Zone, Distance, Equipe, Challenge

class zoneAdmin(admin.ModelAdmin):
    list_display = ('zone', 'etat')
    def zone(self, obj):
    	return obj.nom+" (code: "+str(obj.code)+" mdp: "+str(obj.mdp)+")"
    
class distanceAdmin(admin.ModelAdmin):
    list_display = ('zone1','zone2', 'distance')
    def zone1(self, obj):
    	return obj.zone1.nom
    def zone2(self, obj):
    	return obj.zone2.nom
    
class equipeAdmin(admin.ModelAdmin):
    list_display = ('pk','nom')
    
class challengeAdmin(admin.ModelAdmin):
    list_display = ('equipe','zone','rank')
    def equipe(self, obj):
    	return obj.equipe.nom
    def zone(self, obj):
    	return obj.zone.nom

admin.site.register(Zone, zoneAdmin)
admin.site.register(Distance, distanceAdmin)
admin.site.register(Equipe, equipeAdmin)
admin.site.register(Challenge, challengeAdmin)