Spaces:
Build error
Build error
File size: 3,993 Bytes
877587b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 |
from rest_framework import serializers
from .models import User, Place, Review
class UserSerializer(serializers.ModelSerializer):
"""Serializer for user registration and display."""
class Meta:
model = User
fields = ['id', 'name', 'phone']
read_only_fields = ['id']
def create(self, validated_data):
"""Create a new user with phone authentication."""
user = User.objects.create_user(
phone=validated_data['phone'],
name=validated_data['name'],
username=validated_data['phone'] # Use phone as username
)
return user
class PlaceSerializer(serializers.ModelSerializer):
"""Serializer for place list/search results."""
average_rating = serializers.SerializerMethodField()
class Meta:
model = Place
fields = ['id', 'name', 'address', 'average_rating']
read_only_fields = ['id']
def get_average_rating(self, obj):
"""Get the average rating for this place."""
return obj.average_rating
class ReviewSerializer(serializers.ModelSerializer):
"""Serializer for reviews with user details."""
user_name = serializers.CharField(source='user.name', read_only=True)
class Meta:
model = Review
fields = ['id', 'user_name', 'rating', 'text', 'created_at']
read_only_fields = ['id', 'user_name', 'created_at']
class PlaceDetailSerializer(serializers.ModelSerializer):
"""Detailed serializer for a single place with all reviews."""
average_rating = serializers.SerializerMethodField()
reviews = serializers.SerializerMethodField()
class Meta:
model = Place
fields = ['id', 'name', 'address', 'average_rating', 'reviews']
read_only_fields = ['id']
def get_average_rating(self, obj):
"""Get the average rating for this place."""
return obj.average_rating
def get_reviews(self, obj):
"""
Get all reviews sorted with current user's review first,
then all others sorted by newest first.
"""
request = self.context.get('request')
if not request or not request.user.is_authenticated:
reviews = obj.reviews.all()
else:
# Get current user's review
user_review = obj.reviews.filter(user=request.user).first()
# Get all other reviews
other_reviews = obj.reviews.exclude(user=request.user).order_by('-created_at')
# Combine: user's review first, then others
if user_review:
reviews = [user_review] + list(other_reviews)
else:
reviews = list(other_reviews)
return ReviewSerializer(reviews, many=True).data
class AddReviewSerializer(serializers.Serializer):
"""Serializer for adding a review (creates place if needed)."""
place_name = serializers.CharField(max_length=255)
place_address = serializers.CharField()
rating = serializers.IntegerField(min_value=1, max_value=5)
review_text = serializers.CharField()
def validate(self, data):
"""Validate the review data."""
return data
def create(self, validated_data):
"""
Create a review and associated place if it doesn't exist.
Uses get_or_create to ensure atomic operation.
"""
user = self.context['request'].user
# Get or create the place
place, created = Place.objects.get_or_create(
name=validated_data['place_name'],
address=validated_data['place_address']
)
# Create or update the review
review, created = Review.objects.update_or_create(
user=user,
place=place,
defaults={
'rating': validated_data['rating'],
'text': validated_data['review_text']
}
)
return review
|