from django.contrib import admin
from spa.models.Genre import Genre
from spa.models.UserProfile import UserProfile
from spa.models.ChatMessage import ChatMessage
from spa.models.Recurrence import Recurrence
from spa.models.Release import Release
from spa.models.Event import Event
from spa.models.Label import Label
from spa.models.Mix import Mix
from spa.models._Activity import _Activity
from spa.models.MixLike import MixLike
from spa.models.MixPlay import MixPlay
from spa.models.MixFavourite import MixFavourite
from spa.models.Release import ReleaseAudio
from spa.models.Venue import Venue


class DefaultAdmin(admin.ModelAdmin):
    def save_model(self, request, obj, form, change):
        obj.user = request.user.get_profile()
        obj.save()


admin.site.register(Mix)
admin.site.register(_Activity)
admin.site.register(MixLike)
admin.site.register(MixPlay)
admin.site.register(MixFavourite)
admin.site.register(Genre)
admin.site.register(Label)
admin.site.register(Release, DefaultAdmin)
admin.site.register(ReleaseAudio)
admin.site.register(Venue)
admin.site.register(Event)
admin.site.register(UserProfile)
admin.site.register(Recurrence)
admin.site.register(ChatMessage)