views.py 789 B

1234567891011121314151617181920212223242526
  1. from django.contrib.auth.decorators import login_required
  2. from django.core.exceptions import PermissionDenied
  3. from django.shortcuts import render, get_object_or_404
  4. from django.contrib.auth.models import User
  5. from .models import Corporation
  6. @login_required
  7. def user(request):
  8. return render(request, 'adhesions/user.html', {
  9. 'adherent': request.user,
  10. 'adhesion': request.user.profile.adhesion,
  11. })
  12. @login_required
  13. def corporation(request, pk):
  14. corporation = get_object_or_404(Corporation, pk=pk)
  15. try:
  16. corporation.members.get(pk=request.user.pk)
  17. except User.DoesNotExist:
  18. raise PermissionDenied
  19. return render(request, 'adhesions/corporation.html', {
  20. 'adherent': corporation,
  21. 'adhesion': corporation.adhesion,
  22. })