from django.shortcuts import render, get_object_or_404, redirect
from django.contrib.auth.decorators import login_required
from .models import Song


@login_required
def upload(request):
    return render(request, 'songs/upload.html')


@login_required
def manage(request):
    return render(request, 'songs/manage.html')


@login_required
def edit(request, song_id):
    song = get_object_or_404(Song, pk=song_id, user=request.user)
    if not request.POST:
        return render(request, 'songs/edit.html', {'song': song})

    # Save data.
    artist = request.POST.get('artist')
    title = request.POST.get('title')
    song.artist = artist
    song.title = title
    song.save()
    return redirect('songs:manage')