# -*- coding: utf-8 -*- from __future__ import unicode_literals from rest_framework import serializers from django.contrib.auth.models import User from core import models class CoreModelSerializer(serializers.HyperlinkedModelSerializer): """ Provide the child link (used by most core models) and run model clean() methods during POST operations. """ child = serializers.PrimaryKeyRelatedField( queryset=models.Child.objects.all()) def validate(self, attrs): instance = self.Meta.model(**attrs) instance.clean() return attrs class UserSerializer(serializers.ModelSerializer): class Meta: model = User fields = ('id', 'username') class ChildSerializer(serializers.HyperlinkedModelSerializer): class Meta: model = models.Child fields = ('id', 'first_name', 'last_name', 'birth_date', 'slug') lookup_field = 'slug' class DiaperChangeSerializer(CoreModelSerializer): class Meta: model = models.DiaperChange fields = ('id', 'child', 'time', 'wet', 'solid', 'color') class FeedingSerializer(CoreModelSerializer): class Meta: model = models.Feeding fields = ('id', 'child', 'start', 'end', 'duration', 'type', 'method', 'amount') class NoteSerializer(CoreModelSerializer): class Meta: model = models.Note fields = ('id', 'child', 'note', 'time') class SleepSerializer(CoreModelSerializer): class Meta: model = models.Sleep fields = ('id', 'child', 'start', 'end', 'duration') class TimerSerializer(CoreModelSerializer): user = serializers.PrimaryKeyRelatedField(queryset=User.objects.all()) class Meta: model = models.Timer fields = ('id', 'name', 'start', 'end', 'duration', 'active', 'user') class TummyTimeSerializer(CoreModelSerializer): class Meta: model = models.TummyTime fields = ('id', 'child', 'start', 'end', 'duration', 'milestone') class WeightSerializer(CoreModelSerializer): class Meta: model = models.Weight fields = ('id', 'child', 'weight', 'date')