from django.shortcuts import render, get_object_or_404 from django.http import HttpResponse from models import Event, EventParticipation from musicians.models import Musician from serializers import ParticipationSerializer import datetime from rest_framework.decorators import api_view from rest_framework.response import Response from rest_framework import status from django.forms.models import ModelForm from django.conf.urls import patterns, url from django.contrib.auth.decorators import login_required from django.utils.decorators import method_decorator # ---------------------------------------- API --------------------------------------------------------- @api_view( ['GET', 'PUT'] ) def event_api( request, username = None, eventId = None ): try: participationQs = EventParticipation.objects.filter( event__date__gte = datetime.date.today() ) if username: participationQs = EventParticipation.objects.filter( musician__user__username = username ) if eventId: participationQs = participationQs.filter( event__pk = eventId ) except EventParticipation.DoesNotExist: return HttpResponse( status=404 ) if request.method == 'GET': serializer = ParticipationSerializer( participationQs ) return Response( serializer.data ) elif request.method == 'PUT': serializer = ParticipationSerializer ( participationQs, data = request.DATA, many=True ) if serializer.is_valid(): for serializedObject in serializer.object: if serializedObject.musician.user != request.user: if not request.user.has_perm('change_others_participation') : return Response( status = status.HTTP_403_FORBIDDEN ) serializer.save() return Response( serializer.data ) else: return Response( status = status.HTTP_400_BAD_REQUEST ) # ------------------------------------ Normal Views ---------------------------------------------------- @login_required def main_view( request ): if request.user.has_perm( 'eventplanner.change_others_participation'): return events_grid( request ) else: return eventplanning( request ) @login_required def eventplanning( request ): """ View for a specific user, to edit his events """ # All events in the future sorted by date all_future_events = list ( Event.objects.filter( date__gte = datetime.date.today() ) ) musician = get_object_or_404( Musician, user=request.user ) for e in all_future_events: e.participation = EventParticipation.get_or_create( event = e, musician = musician ) context = { 'events' : all_future_events } return render ( request, 'eventplanner/eventplanning_view.html', context ) @login_required def events_grid( request ): musicians = Musician.objects.all() musicianNames = [ m.user.username for m in musicians ] all_future_events = list ( Event.objects.filter( date__gte = datetime.date.today() ) ) for e in all_future_events: e.participation = [ EventParticipation.get_or_create( event = e, musician = m ) for m in musicians ] context = { 'events': all_future_events, 'musicianNames' : musicianNames } return render ( request, 'eventplanner/events_grid.html', context ) # ------------------------------------ Detail Views ---------------------------------------------------- from django.views.generic.edit import UpdateView class EventForm( ModelForm ): class Meta: model = Event fields= ['title', 'date','time','type'] class EventUpdate( UpdateView ): form_class = EventForm model = Event template_name_suffix = "_update_form" success_url = '/events/grid' @method_decorator(login_required) def dispatch(self, request, *args, **kwargs): return super(EventUpdate, self).dispatch(request, *args, **kwargs) # ------------------------------------ URLS ---------------------------------------------------- urls = patterns('', url(r'^$', main_view ), url(r'^grid$', events_grid ), url(r'^planning$', eventplanning ), url(r'^(?P\d+)$', EventUpdate.as_view() ), url(r'^api/', event_api, name="event_api" ), url(r'^api/(\w+)/$', event_api, name="event_api_per_user" ), url(r'^api/(\w+)/(\d+)$', event_api, name="event_api_per_user_event"), )