from django.contrib.auth.decorators import login_required
from django.core.exceptions import PermissionDenied
from django.shortcuts import render, get_object_or_404
from django.contrib.auth.models import User

from .models import Corporation


@login_required
def adhesion(request):
    return render(request, 'adhesions/adhesion.html', {
        'adhesion': request.user.profile.adhesion,
    })


@login_required
def corporation(request, pk):
    corporation = get_object_or_404(Corporation, pk=pk)
    try:
        corporation.members.get(pk=request.user.pk)
    except User.DoesNotExist:
        raise PermissionDenied
    return render(request, 'adhesions/corporation.html', {
        'corporation': corporation,
    })