import datetime import operator from functools import reduce import simplejson from django.contrib.auth.decorators import login_required from django.contrib import messages from django.core import serializers from django.core.exceptions import PermissionDenied from django.db.models import Q from django.http import HttpResponse from django.shortcuts import get_object_or_404 from django.urls import reverse_lazy, reverse, NoReverseMatch from django.views import generic from django.views.decorators.clickjacking import xframe_options_exempt from RIGS import models from assets import models as asset_models from training import models as training_models def is_ajax(request): return request.headers.get('x-requested-with') == 'XMLHttpRequest' class Index(generic.TemplateView): # Displays the current rig count along with a few other bits and pieces template_name = '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 class SecureAPIRequest(generic.View): models = { 'venue': models.Venue, 'person': models.Person, 'organisation': models.Organisation, 'profile': models.Profile, 'event': models.Event, 'supplier': asset_models.Supplier, 'training_item': training_models.TrainingItem, } perms = { 'venue': 'RIGS.view_venue', 'person': 'RIGS.view_person', 'organisation': 'RIGS.view_organisation', 'profile': 'RIGS.view_profile', 'event': None, 'supplier': None, 'training_item': None, # TODO } ''' 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(",") filters = request.GET.get('filters', []) if filters: filters = filters.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('q', 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) for filter in filters: q = Q(**{field: True}) 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) for o in objects: data = { 'pk': o.pk, 'value': o.pk, 'text': 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:%S") end_datetime = datetime.datetime.strptime(end, "%Y-%m-%dT%H:%M:%S") objects = self.models[model].objects.events_in_bounds(start_datetime, end_datetime) results = [] for item in objects: data = { 'pk': item.pk, 'title': item.name, 'is_rig': item.is_rig, 'status': str(item.get_status_display()), 'earliest': item.earliest_time.isoformat(), 'latest': item.latest_time.isoformat(), 'url': str(item.get_absolute_url()) } results.append(data) json = simplejson.dumps(results) return HttpResponse(json, content_type="application/json") # Always json return HttpResponse(model) class ModalURLMixin: def get_close_url(self, update, detail): if is_ajax(self.request): url = reverse_lazy('closemodal') update_url = str(reverse_lazy(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(detail, kwargs={ 'pk': self.object.pk, }) return url class GenericListView(generic.ListView): template_name = 'generic_list.html' paginate_by = 20 def get_context_data(self, **kwargs): context = super(GenericListView, self).get_context_data(**kwargs) context['page_title'] = self.model.__name__ + "s" if is_ajax(self.request): context['override'] = "base_ajax.html" return context def get_queryset(self): q = self.request.GET.get('q', "") filter = Q(name__icontains=q) | Q(email__icontains=q) | Q(address__icontains=q) | Q(notes__icontains=q) | Q( phone__startswith=q) | Q(phone__endswith=q) # try and parse an int try: val = int(q) filter = filter | Q(pk=val) except: # noqa # not an integer pass object_list = self.model.objects.filter(filter) orderBy = self.request.GET.get('orderBy', "name") if orderBy != "": object_list = object_list.order_by(orderBy) return object_list class GenericDetailView(generic.DetailView): template_name = "generic_detail.html" def get_context_data(self, **kwargs): context = super(GenericDetailView, self).get_context_data(**kwargs) context['page_title'] = "{} | {}".format(self.model.__name__, self.object.name) if is_ajax(self.request): context['override'] = "base_ajax.html" return context class GenericUpdateView(generic.UpdateView): template_name = "generic_form.html" def get_context_data(self, **kwargs): context = super(GenericUpdateView, self).get_context_data(**kwargs) context['page_title'] = "Edit {}".format(self.model.__name__) if is_ajax(self.request): context['override'] = "base_ajax.html" return context class GenericCreateView(generic.CreateView): template_name = "generic_form.html" def get_context_data(self, **kwargs): context = super(GenericCreateView, self).get_context_data(**kwargs) context['page_title'] = "Create {}".format(self.model.__name__) if is_ajax(self.request): context['override'] = "base_ajax.html" return context class SearchHelp(generic.TemplateView): template_name = 'search_help.html' class CloseModal(generic.TemplateView): """ 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. """ template_name = 'closemodal.html' def get_context_data(self, **kwargs): return {'messages': messages.get_messages(self.request)} class OEmbedView(generic.View): def get(self, request, pk=None): embed_url = reverse(self.url_name, args=[pk]) full_url = "{0}://{1}{2}".format(request.scheme, request.META['HTTP_HOST'], embed_url) data = { 'html': ''.format(full_url), 'version': '1.0', 'type': 'rich', 'height': '250' } json = simplejson.JSONEncoderForHTML().encode(data) return HttpResponse(json, content_type="application/json")