mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-17 05:22:16 +00:00
8ade2b234c
No idea what the effect will be if we have somebody with more than one space in their name. This behavoir might be quite unpredictable.
391 lines
13 KiB
Python
391 lines
13 KiB
Python
from django.core.exceptions import PermissionDenied
|
|
from django.http.response import HttpResponseRedirect
|
|
from django.http import HttpResponse
|
|
from django.core.urlresolvers import reverse_lazy, reverse, NoReverseMatch
|
|
from django.views import generic
|
|
from django.db.models import Q
|
|
from django.shortcuts import get_object_or_404
|
|
from django.core import serializers
|
|
import simplejson
|
|
from django.contrib import messages
|
|
import datetime
|
|
import operator
|
|
|
|
from RIGS import models, forms
|
|
|
|
"""
|
|
Displays the current rig count along with a few other bits and pieces
|
|
"""
|
|
class Index(generic.TemplateView):
|
|
template_name = 'RIGS/index.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(Index, self).get_context_data(**kwargs)
|
|
context['rig_count'] = models.Event.objects.rig_count()
|
|
return context
|
|
|
|
def login(request, **kwargs):
|
|
if request.user.is_authenticated():
|
|
next = request.REQUEST.get('next', '/')
|
|
return HttpResponseRedirect(request.REQUEST.get('next', '/'))
|
|
else:
|
|
from django.contrib.auth.views import login
|
|
|
|
return login(request, authentication_form=forms.LoginForm)
|
|
|
|
|
|
"""
|
|
Called from a modal window (e.g. when an item is submitted to an event/invoice).
|
|
May optionally also include some javascript in a success message to cause a load of
|
|
the new information onto the page.
|
|
"""
|
|
class CloseModal(generic.TemplateView):
|
|
template_name = 'closemodal.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
return {'messages': messages.get_messages(self.request)}
|
|
|
|
|
|
class PersonList(generic.ListView):
|
|
model = models.Person
|
|
paginate_by = 20
|
|
|
|
def get_queryset(self):
|
|
q = self.request.GET.get('q', "")
|
|
if len(q) >= 3:
|
|
object_list = self.model.objects.filter(Q(name__icontains=q) | Q(email__icontains=q))
|
|
else:
|
|
object_list = self.model.objects.all()
|
|
orderBy = self.request.GET.get('orderBy', None)
|
|
if orderBy is not None:
|
|
object_list = object_list.order_by(orderBy)
|
|
return object_list
|
|
|
|
|
|
class PersonDetail(generic.DetailView):
|
|
model = models.Person
|
|
|
|
|
|
class PersonCreate(generic.CreateView):
|
|
model = models.Person
|
|
|
|
def get_success_url(self):
|
|
if self.request.is_ajax():
|
|
url = reverse_lazy('closemodal')
|
|
update_url = str(reverse_lazy('person_update',kwargs={'pk':self.object.pk}))
|
|
messages.info(self.request, "modalobject="+serializers.serialize("json", [self.object]))
|
|
messages.info(self.request, "modalobject[0]['update_url']='"+update_url+"'")
|
|
else:
|
|
url = reverse_lazy('person_detail', kwargs={
|
|
'pk': self.object.pk,
|
|
})
|
|
return url
|
|
|
|
|
|
class PersonUpdate(generic.UpdateView):
|
|
model = models.Person
|
|
|
|
def get_success_url(self):
|
|
if self.request.is_ajax():
|
|
url = reverse_lazy('closemodal')
|
|
update_url = str(reverse_lazy('person_update',kwargs={'pk':self.object.pk}))
|
|
messages.info(self.request, "modalobject="+serializers.serialize("json", [self.object]))
|
|
messages.info(self.request, "modalobject[0]['update_url']='"+update_url+"'")
|
|
else:
|
|
url = reverse_lazy('person_detail', kwargs={
|
|
'pk': self.object.pk,
|
|
})
|
|
return url
|
|
|
|
|
|
class OrganisationList(generic.ListView):
|
|
model = models.Organisation
|
|
paginate_by = 20
|
|
|
|
def get_queryset(self):
|
|
q = self.request.GET.get('q', "")
|
|
if len(q) >= 3:
|
|
object_list = self.model.objects.filter(Q(name__icontains=q) | Q(address__icontains=q))
|
|
else:
|
|
object_list = self.model.objects.all()
|
|
orderBy = self.request.GET.get('orderBy', "")
|
|
if orderBy is not "":
|
|
object_list = object_list.order_by(orderBy)
|
|
return object_list
|
|
|
|
|
|
class OrganisationDetail(generic.DetailView):
|
|
model = models.Organisation
|
|
|
|
|
|
class OrganisationCreate(generic.CreateView):
|
|
model = models.Organisation
|
|
|
|
def get_success_url(self):
|
|
if self.request.is_ajax():
|
|
url = reverse_lazy('closemodal')
|
|
update_url = str(reverse_lazy('organisation_update',kwargs={'pk':self.object.pk}))
|
|
messages.info(self.request, "modalobject="+serializers.serialize("json", [self.object]))
|
|
messages.info(self.request, "modalobject[0]['update_url']='"+update_url+"'")
|
|
else:
|
|
url = reverse_lazy('organisation_detail', kwargs={
|
|
'pk': self.object.pk,
|
|
})
|
|
return url
|
|
|
|
|
|
class OrganisationUpdate(generic.UpdateView):
|
|
model = models.Organisation
|
|
|
|
def get_success_url(self):
|
|
if self.request.is_ajax():
|
|
url = reverse_lazy('closemodal')
|
|
update_url = str(reverse_lazy('organisation_update',kwargs={'pk':self.object.pk}))
|
|
messages.info(self.request, "modalobject="+serializers.serialize("json", [self.object]))
|
|
messages.info(self.request, "modalobject[0]['update_url']='"+update_url+"'")
|
|
else:
|
|
url = reverse_lazy('organisation_detail', kwargs={
|
|
'pk': self.object.pk,
|
|
})
|
|
return url
|
|
|
|
|
|
class VenueList(generic.ListView):
|
|
model = models.Venue
|
|
paginate_by = 20
|
|
|
|
def get_queryset(self):
|
|
q = self.request.GET.get('q', "")
|
|
if len(q) >= 3:
|
|
object_list = self.model.objects.filter(Q(name__icontains=q) | Q(address__icontains=q))
|
|
else:
|
|
object_list = self.model.objects.all()
|
|
orderBy = self.request.GET.get('orderBy', "")
|
|
if orderBy is not "":
|
|
object_list = object_list.order_by(orderBy)
|
|
return object_list
|
|
|
|
|
|
class VenueDetail(generic.DetailView):
|
|
model = models.Venue
|
|
|
|
|
|
class VenueCreate(generic.CreateView):
|
|
model = models.Venue
|
|
|
|
def get_success_url(self):
|
|
if self.request.is_ajax():
|
|
url = reverse_lazy('closemodal')
|
|
update_url = str(reverse_lazy('venue_update',kwargs={'pk':self.object.pk}))
|
|
messages.info(self.request, "modalobject="+serializers.serialize("json", [self.object]))
|
|
messages.info(self.request, "modalobject[0]['update_url']='"+update_url+"'")
|
|
else:
|
|
url = reverse_lazy('venue_detail', kwargs={
|
|
'pk': self.object.pk,
|
|
})
|
|
return url
|
|
|
|
|
|
class VenueUpdate(generic.UpdateView):
|
|
model = models.Venue
|
|
|
|
def get_success_url(self):
|
|
if self.request.is_ajax():
|
|
url = reverse_lazy('closemodal')
|
|
update_url = str(reverse_lazy('venue_update',kwargs={'pk':self.object.pk}))
|
|
messages.info(self.request, "modalobject="+serializers.serialize("json", [self.object]))
|
|
messages.info(self.request, "modalobject[0]['update_url']='"+update_url+"'")
|
|
else:
|
|
url = reverse_lazy('venue_detail', kwargs={
|
|
'pk': self.object.pk,
|
|
})
|
|
return url
|
|
|
|
|
|
class SecureAPIRequest(generic.View):
|
|
models = {
|
|
'venue': models.Venue,
|
|
'person': models.Person,
|
|
'organisation': models.Organisation,
|
|
'profile': models.Profile,
|
|
'event': models.Event,
|
|
}
|
|
|
|
perms = {
|
|
'venue': 'RIGS.view_venue',
|
|
'person': 'RIGS.view_person',
|
|
'organisation': 'RIGS.view_organisation',
|
|
'profile': None,
|
|
'event': 'RIGS.view_event',
|
|
}
|
|
|
|
'''
|
|
Validate the request is allowed based on user permissions.
|
|
Raises 403 if denied.
|
|
Potential to add API key validation at a later date.
|
|
'''
|
|
|
|
def __validate__(self, request, key, perm):
|
|
if request.user.is_active:
|
|
if request.user.is_superuser or perm is None:
|
|
return True
|
|
elif request.user.has_perm(perm):
|
|
return True
|
|
raise PermissionDenied()
|
|
|
|
def get(self, request, model, pk=None, param=None):
|
|
# Request permission validation things
|
|
key = request.GET.get('apikey', None)
|
|
perm = self.perms[model]
|
|
self.__validate__(request, key, perm)
|
|
|
|
# Response format where applicable
|
|
format = request.GET.get('format', 'json')
|
|
fields = request.GET.get('fields', None)
|
|
if fields:
|
|
fields = fields.split(",")
|
|
|
|
# Supply data for one record
|
|
if pk:
|
|
object = get_object_or_404(self.models[model], pk=pk)
|
|
data = serializers.serialize(format, [object], fields=fields)
|
|
return HttpResponse(data, content_type="application/" + format)
|
|
|
|
# Supply data for autocomplete ajax request in json form
|
|
term = request.GET.get('term', None)
|
|
if term:
|
|
if fields is None: # Default to just name
|
|
fields = ['name']
|
|
|
|
# Build a list of Q objects for use later
|
|
queries = []
|
|
for part in term.split(" "):
|
|
qs = []
|
|
for field in fields:
|
|
q = Q(**{field + "__icontains": part})
|
|
qs.append(q)
|
|
queries.append(reduce(operator.or_, qs))
|
|
|
|
|
|
# Build the data response list
|
|
results = []
|
|
query = reduce(operator.and_, queries)
|
|
objects = self.models[model].objects.filter(query)
|
|
print objects.query
|
|
for o in objects:
|
|
data = {
|
|
'pk': o.pk,
|
|
'value': o.pk,
|
|
'label': o.name,
|
|
}
|
|
try: # See if there is a valid update URL
|
|
data['update'] = reverse("%s_update" % model, kwargs={'pk': o.pk})
|
|
except NoReverseMatch:
|
|
pass
|
|
results.append(data)
|
|
|
|
# return a data response
|
|
json = simplejson.dumps(results)
|
|
return HttpResponse(json, content_type="application/json") # Always json
|
|
|
|
start = request.GET.get('start', None)
|
|
end = request.GET.get('end', None)
|
|
|
|
if model == "event" and start and end:
|
|
# Probably a calendar request
|
|
start_datetime = datetime.datetime.strptime( start, "%Y-%m-%dT%H:%M:%SZ" )
|
|
end_datetime = datetime.datetime.strptime( end, "%Y-%m-%dT%H:%M:%SZ" )
|
|
all_objects = self.models[model].objects
|
|
results = []
|
|
filter = Q(start_date__lte=end_datetime) & Q(start_date__gte=start_datetime)
|
|
objects = all_objects.filter(filter).select_related('person', 'organisation', 'venue', 'mic').order_by('-start_date')
|
|
for item in objects:
|
|
data = {
|
|
'pk': item.pk,
|
|
'title': item.name
|
|
}
|
|
|
|
data['is_rig'] = item.is_rig
|
|
data['status'] = str(item.get_status_display())
|
|
|
|
if item.start_date:
|
|
data['start_date'] = item.start_date.strftime('%Y-%m-%d')
|
|
|
|
if item.has_start_time:
|
|
data['start_time'] = item.start_time.strftime('%H:%M:%SZ')
|
|
|
|
if item.end_date:
|
|
data['end_date'] = item.end_date.strftime('%Y-%m-%d')
|
|
|
|
if item.has_end_time:
|
|
data['end_time'] = item.end_time.strftime('%H:%M:%SZ')
|
|
|
|
if item.meet_at:
|
|
data['meet_at'] = item.meet_at.strftime('%Y-%m-%dT%H:%M:%SZ')
|
|
|
|
if item.access_at:
|
|
data['access_at'] = item.access_at.strftime('%Y-%m-%dT%H:%M:%SZ')
|
|
|
|
if item.venue:
|
|
data['venue'] = item.venue.name
|
|
|
|
if item.person:
|
|
data['person'] = item.person.name
|
|
|
|
if item.organisation:
|
|
data['organisation'] = item.organisation.name
|
|
|
|
if item.mic:
|
|
data['mic'] = {
|
|
'name':item.mic.get_full_name(),
|
|
'initials':item.mic.initials
|
|
}
|
|
|
|
if item.description:
|
|
data['description'] = item.description
|
|
|
|
if item.notes:
|
|
data['notes'] = item.notes
|
|
|
|
data['url'] = str(reverse_lazy('event_detail',kwargs={'pk':item.pk}))
|
|
|
|
results.append(data)
|
|
json = simplejson.dumps(results)
|
|
return HttpResponse(json, content_type="application/json") # Always json
|
|
|
|
return HttpResponse(model)
|
|
|
|
class ProfileDetail(generic.DetailView):
|
|
model = models.Profile
|
|
|
|
def get_queryset(self):
|
|
try:
|
|
pk = self.kwargs['pk']
|
|
except KeyError:
|
|
pk = self.request.user.id
|
|
self.kwargs['pk'] = pk
|
|
|
|
return self.model.objects.filter(pk=pk)
|
|
|
|
class ProfileUpdateSelf(generic.UpdateView):
|
|
model = models.Profile
|
|
fields = ['first_name', 'last_name', 'email', 'initials', 'phone']
|
|
|
|
def get_queryset(self):
|
|
pk = self.request.user.id
|
|
self.kwargs['pk'] = pk
|
|
|
|
return self.model.objects.filter(pk=pk)
|
|
|
|
def get_success_url(self):
|
|
url = reverse_lazy('profile_detail')
|
|
return url
|
|
|
|
class ResetApiKey(generic.RedirectView):
|
|
def get_redirect_url(self, *args, **kwargs):
|
|
self.request.user.api_key = self.request.user.make_api_key()
|
|
|
|
self.request.user.save()
|
|
|
|
return reverse_lazy('profile_detail')
|