mirror of
https://github.com/seanmorley15/AdventureLog.git
synced 2025-07-24 15:29:36 +02:00
90 lines
No EOL
3.8 KiB
Python
90 lines
No EOL
3.8 KiB
Python
from django.shortcuts import render
|
|
from .models import Country, Region, VisitedRegion
|
|
from .serializers import CountrySerializer, RegionSerializer, VisitedRegionSerializer
|
|
from rest_framework import viewsets, status
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from django.shortcuts import get_object_or_404
|
|
from rest_framework.response import Response
|
|
from rest_framework.decorators import api_view, permission_classes
|
|
import os
|
|
import json
|
|
from django.conf import settings
|
|
from django.contrib.staticfiles import finders
|
|
|
|
@api_view(['GET'])
|
|
@permission_classes([IsAuthenticated])
|
|
def regions_by_country(request, country_code):
|
|
# require authentication
|
|
country = get_object_or_404(Country, country_code=country_code)
|
|
regions = Region.objects.filter(country=country).order_by('name')
|
|
serializer = RegionSerializer(regions, many=True)
|
|
return Response(serializer.data)
|
|
|
|
@api_view(['GET'])
|
|
@permission_classes([IsAuthenticated])
|
|
def visits_by_country(request, country_code):
|
|
country = get_object_or_404(Country, country_code=country_code)
|
|
visits = VisitedRegion.objects.filter(region__country=country, user_id=request.user.id)
|
|
|
|
serializer = VisitedRegionSerializer(visits, many=True)
|
|
return Response(serializer.data)
|
|
|
|
class CountryViewSet(viewsets.ReadOnlyModelViewSet):
|
|
queryset = Country.objects.all()
|
|
serializer_class = CountrySerializer
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
class RegionViewSet(viewsets.ReadOnlyModelViewSet):
|
|
queryset = Region.objects.all()
|
|
serializer_class = RegionSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
class VisitedRegionViewSet(viewsets.ModelViewSet):
|
|
serializer_class = VisitedRegionSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get_queryset(self):
|
|
return VisitedRegion.objects.filter(user_id=self.request.user.id)
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
# Set the user_id to the request user's ID
|
|
request.data['user_id'] = request.user.id
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
self.perform_create(serializer)
|
|
headers = self.get_success_headers(serializer.data)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
|
|
|
|
class GeoJSONView(viewsets.ViewSet):
|
|
"""
|
|
Combine all GeoJSON data from .json files in static/data into a single GeoJSON object.
|
|
"""
|
|
def list(self, request):
|
|
combined_geojson = {
|
|
"type": "FeatureCollection",
|
|
"features": []
|
|
}
|
|
|
|
# Use Django's static file finder to locate the 'data' directory
|
|
data_dir = finders.find('data')
|
|
|
|
if not data_dir or not os.path.isdir(data_dir):
|
|
return Response({"error": "Data directory does not exist."}, status=404)
|
|
|
|
for filename in os.listdir(data_dir):
|
|
if filename.endswith('.json'):
|
|
file_path = os.path.join(data_dir, filename)
|
|
try:
|
|
with open(file_path, 'r') as f:
|
|
json_data = json.load(f)
|
|
# Check if the JSON data is GeoJSON
|
|
if isinstance(json_data, dict) and "type" in json_data:
|
|
if json_data["type"] == "FeatureCollection":
|
|
combined_geojson["features"].extend(json_data.get("features", []))
|
|
elif json_data["type"] == "Feature":
|
|
combined_geojson["features"].append(json_data)
|
|
# You can add more conditions here for other GeoJSON types if needed
|
|
except (IOError, json.JSONDecodeError) as e:
|
|
return Response({"error": f"Error reading file {filename}: {str(e)}"}, status=500)
|
|
|
|
return Response(combined_geojson) |