use small in image in api

liqd · Mar 6, 2017 · c59038d · c59038d
1 parent b80c6df
commit c59038d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/euth/users/serializers.py b/euth/users/serializers.py
@@ -1,16 +1,22 @@
+from easy_thumbnails.files import get_thumbnailer
 from rest_framework import serializers
 
 from .models import User
 
 
 class UserSerializer(serializers.ModelSerializer):
-    avatar = serializers.ImageField()
+    avatar = serializers.SerializerMethodField()
 
     class Meta:
         model = User
         fields = ('id', 'username', 'avatar', 'default_avatar')
         read_only_fields = ('id', 'username', 'avatar', 'default_avatar')
 
+    def get_avatar(self, obj):
+        if obj.avatar:
+            image = get_thumbnailer(obj.avatar)['avatar_small']
+            return image.url
+
 
 # mails should not be exposed in API, so there is a separate one for this
 class UserWithMailSerializer(UserSerializer):