import json import gzip from datetime import datetime from django.shortcuts import render from django.http import JsonResponse, HttpResponse from django.core.serializers.json import DjangoJSONEncoder from django.views import View from django.core.cache import cache from django.contrib.auth.decorators import permission_required from django.contrib.auth.mixins import PermissionRequiredMixin from django.contrib.gis.geos import Point from .models import * from .workers.points2trips import TripConverter def index(request): return render(request, 'gps_logger/index.html') @permission_required("gps-logger.markers.change") def upload(request): return render(request, 'gps_logger/upload.html') class MarkerCreateView(PermissionRequiredMixin, View): permission_required = ("gps-logger.markers.change") def post(self, request): insert_count = censor_count = replace_count = 0 data = json.loads(request.body) for file in data: if len(data[file]) == 0: continue objects = [Marker( timestamp=datetime.strptime(x['timestamp'].replace("Z","-0000"), r"%Y-%m-%dT%H:%M:%S.%f%z"), location = Point(x['lng'], x['lat']), alt = x['alt'], hdop = x.get('hdop', None), speed = x.get('speed', None) ) for x in data[file]] start_date = objects[0].timestamp end_date = objects[-1].timestamp oldMarkers = Marker.objects.filter(timestamp__range=(start_date, end_date)).all() identical = len(oldMarkers) == len(objects) if identical: for i, marker in enumerate(oldMarkers): if marker.timestamp != objects[i].timestamp: identical = False break if identical: for i, marker in enumerate(oldMarkers): objects[i].id = marker.id Marker.objects.bulk_update(objects, ['location', 'alt', 'hdop', 'speed']) replace_count += len(oldMarkers) else: insert_count += len(objects) - len(oldMarkers) oldMarkers.delete() Marker.objects.bulk_create(objects) censor_count = censor() return JsonResponse({ 'censored': censor_count, 'replaced': replace_count, 'inserted': insert_count - censor_count }, status=201) class CreateTripsView(PermissionRequiredMixin, View): permission_required = ("gps-logger.trips.change") def get(self, request): data = list(Marker.objects.all()) tc = TripConverter(data) tc.save() cache.delete('trips') return JsonResponse({ 'inserted': len(tc.trips) }, status=201) def marker_view(request): data = cache.get('markers') if not data: values = list(Marker.objects.values()) for marker in values: location = marker['location'] if location: marker['lng'] = location.x marker['lat'] = location.y del marker['location'] data = json.dumps(values, cls=DjangoJSONEncoder) cache.set('markers', data, 3600*24*30) return HttpResponse(data, content_type='application/json') def trip_view(request): data = cache.get('trips') if not data: values = list(Trip.objects.values()) for trip in values: trip['path'] = json.loads(gzip.decompress(trip['path'])) trip['startTime'] = trip['startTime'].timestamp() trip['endTime'] = trip['endTime'].timestamp() trip['movementTime'] = trip['movementTime'].total_seconds() data = json.dumps(values) cache.set('trips', data, 3600*24*30) return HttpResponse(data, content_type='application/json') class DeleteCensoredView(PermissionRequiredMixin, View): permission_required = ("gps-logger.markers.change",) def get(self, request): counter = censor() return JsonResponse({'censored': counter}) def censor() -> int: counter = 0 locations = CensoredLocation.objects.all() for location in locations: counter += location.delete_markers() if(cache.get('markers')): cache.delete('markers') return counter