1
0
Fork 0
mirror of https://github.com/seanmorley15/AdventureLog.git synced 2025-07-23 06:49:37 +02:00

Conditional rendering to the country card to show visited status as a badge

This commit is contained in:
Sean Morley 2024-10-16 20:02:28 -04:00
parent aa8d456e41
commit 3a66a433ca
3 changed files with 33 additions and 7 deletions

View file

@ -7,15 +7,25 @@ class CountrySerializer(serializers.ModelSerializer):
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):
return VisitedRegion.objects.filter(region__country=obj).count()
class Meta:
model = Country
fields = '__all__'
read_only_fields = ['id', 'name', 'country_code', 'subregion', 'flag_url']
read_only_fields = ['id', 'name', 'country_code', 'subregion', 'flag_url', 'num_regions', 'num_visits']
class RegionSerializer(serializers.ModelSerializer):
class Meta: