mirror of
https://github.com/seanmorley15/AdventureLog.git
synced 2025-07-19 04:49:37 +02:00
76 lines
No EOL
3.2 KiB
Python
76 lines
No EOL
3.2 KiB
Python
import os
|
|
from .models import Country, Region, VisitedRegion, City, VisitedCity
|
|
from rest_framework import serializers
|
|
from main.utils import CustomModelSerializer
|
|
|
|
|
|
class CountrySerializer(serializers.ModelSerializer):
|
|
def get_public_url(self, obj):
|
|
return os.environ.get('PUBLIC_URL', 'http://127.0.0.1:8000').rstrip('/').replace("'", "")
|
|
|
|
flag_url = serializers.SerializerMethodField()
|
|
num_regions = serializers.SerializerMethodField()
|
|
num_visits = serializers.SerializerMethodField()
|
|
|
|
def get_flag_url(self, obj):
|
|
public_url = self.get_public_url(obj)
|
|
return public_url + '/media/' + 'flags/' + obj.country_code.lower() + '.png'
|
|
|
|
def get_num_regions(self, obj):
|
|
# get the number of regions in the country
|
|
return Region.objects.filter(country=obj).count()
|
|
|
|
def get_num_visits(self, obj):
|
|
request = self.context.get('request')
|
|
user = getattr(request, 'user', None)
|
|
|
|
if user and user.is_authenticated:
|
|
return VisitedRegion.objects.filter(region__country=obj, user_id=user).count()
|
|
|
|
return 0
|
|
|
|
class Meta:
|
|
model = Country
|
|
fields = '__all__'
|
|
read_only_fields = ['id', 'name', 'country_code', 'subregion', 'flag_url', 'num_regions', 'num_visits', 'longitude', 'latitude', 'capital']
|
|
|
|
|
|
class RegionSerializer(serializers.ModelSerializer):
|
|
num_cities = serializers.SerializerMethodField()
|
|
country_name = serializers.CharField(source='country.name', read_only=True)
|
|
class Meta:
|
|
model = Region
|
|
fields = '__all__'
|
|
read_only_fields = ['id', 'name', 'country', 'longitude', 'latitude', 'num_cities', 'country_name']
|
|
|
|
def get_num_cities(self, obj):
|
|
return City.objects.filter(region=obj).count()
|
|
|
|
class CitySerializer(serializers.ModelSerializer):
|
|
region_name = serializers.CharField(source='region.name', read_only=True)
|
|
country_name = serializers.CharField(source='region.country.name', read_only=True
|
|
)
|
|
class Meta:
|
|
model = City
|
|
fields = '__all__'
|
|
read_only_fields = ['id', 'name', 'region', 'longitude', 'latitude', 'region_name', 'country_name']
|
|
|
|
class VisitedRegionSerializer(CustomModelSerializer):
|
|
longitude = serializers.DecimalField(source='region.longitude', max_digits=9, decimal_places=6, read_only=True)
|
|
latitude = serializers.DecimalField(source='region.latitude', max_digits=9, decimal_places=6, read_only=True)
|
|
name = serializers.CharField(source='region.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = VisitedRegion
|
|
fields = ['id', 'user_id', 'region', 'longitude', 'latitude', 'name']
|
|
read_only_fields = ['user_id', 'id', 'longitude', 'latitude', 'name']
|
|
|
|
class VisitedCitySerializer(CustomModelSerializer):
|
|
longitude = serializers.DecimalField(source='city.longitude', max_digits=9, decimal_places=6, read_only=True)
|
|
latitude = serializers.DecimalField(source='city.latitude', max_digits=9, decimal_places=6, read_only=True)
|
|
name = serializers.CharField(source='city.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = VisitedCity
|
|
fields = ['id', 'user_id', 'city', 'longitude', 'latitude', 'name']
|
|
read_only_fields = ['user_id', 'id', 'longitude', 'latitude', 'name'] |