planetstack/core/plus/views.py

   1 #views.py
   2 from django.views.generic import TemplateView
   3
   4 from core.models import Slice,SliceRole,SlicePrivilege,Site,Reservation
   5
   6 class DashboardWelcomeView(TemplateView):
   7     template_name = 'admin/dashboard/welcome.html'
   8
   9     def get(self, request, *args, **kwargs):
  10         context = self.get_context_data(**kwargs)
  11         sliceList = Slice.objects.all()
  12         try:
  13             site = Site.objects.filter(id=request.user.site.id)
  14         except:
  15             site = Site.objects.filter(name="Princeton")
  16         context['site'] = site[0]
  17
  18         slicePrivs = SlicePrivilege.objects.filter(user=request.user)
  19         userSliceInfo = []
  20         for entry in slicePrivs:
  21
  22             try:
  23                 reservationList = Reservation.objects.filter(slice=entry.slice)
  24                 reservations = (True,reservationList)
  25
  26             except:
  27                 reservations = None
  28
  29             userSliceInfo.append({'slice': Slice.objects.get(id=entry.slice.id),
  30                                'role': SliceRole.objects.get(id=entry.role.id).role,
  31                                'reservations': reservations})
  32
  33         context['userSliceInfo'] = userSliceInfo
  34         return self.render_to_response(context=context)