Add album auto list view serializer

Author: Hooram Nam

api/serializers.py

@@ -26,15 +26,30 @@ class Meta:
                   'image_hash',
                   'image_path')
     def get_thumbnail_url(self, obj):
-        return obj.thumbnail.url
+        try:
+            return obj.thumbnail.url
+        except:
+            return None
     def get_thumbnail_height(self, obj):
-        return obj.thumbnail.height
+        try:
+            return obj.thumbnail.height
+        except:
+            return None
     def get_thumbnail_width(self, obj):
-        return obj.thumbnail.width
+        try:
+            return obj.thumbnail.width
+        except:
+            return None
     def get_square_thumbnail_url(self, obj):
-        return obj.square_thumbnail.url
+        try:
+            return obj.square_thumbnail.url
+        except:
+            return None
     def get_image_url(self, obj):
-        return obj.image.url
+        try:
+            return obj.image.url
+        except:
+            return None
     def get_geolocation(self, obj):
         if obj.geolocation_json:
           return json.loads(obj.geolocation_json)
@@ -205,4 +220,37 @@ def get_people(self,obj):
                 serialized_person = PersonSerializer(face.person).data
                 if serialized_person not in res:
                     res.append(serialized_person)
-        return res
+        return res
+
+
+class AlbumAutoListSerializer(serializers.ModelSerializer):
+    # people = serializers.SerializerMethodField()
+    cover_photo_url = serializers.SerializerMethodField()
+
+    class Meta:
+        model = AlbumAuto
+        fields = (
+            "id",   
+            "title",
+            "timestamp",
+            "created_on",
+            "cover_photo_url",
+            "gps_lat",
+            # 'people',
+            "gps_lon")
+
+    def get_cover_photo_url(self,obj):
+        first_photo = obj.photos.first()
+        return first_photo.image.url
+
+    # def get_people(self,obj):
+    #     # ipdb.set_trace()
+    #     photos = obj.photos.all()
+    #     res = []
+    #     for photo in photos:
+    #         faces = photo.faces.all()
+    #         for face in faces:
+    #             serialized_person = PersonSerializer(face.person).data
+    #             if serialized_person not in res:
+    #                 res.append(serialized_person)
+    #     return res

api/views.py

@@ -12,6 +12,10 @@
 from api.serializers import AlbumPersonSerializer
 from api.serializers import AlbumDateSerializer
 
+
+from api.serializers import AlbumAutoListSerializer
+
+
 from api.face_classify import train_faces, cluster_faces
 from api.social_graph import build_social_graph
 from api.autoalbum import generate_event_albums
@@ -155,6 +159,21 @@ def list(self, *args, **kwargs):
 
 
 
+class AlbumAutoListViewSet(viewsets.ModelViewSet):
+    queryset = AlbumAuto.objects.all().order_by('-timestamp')
+    serializer_class = AlbumAutoListSerializer
+    pagination_class = StandardResultsSetPagination
+
+    @cache_response(key_func=CustomObjectKeyConstructor())
+    def retrieve(self, *args, **kwargs):
+        return super(AlbumAutoListViewSet, self).retrieve(*args, **kwargs)
+
+    @cache_response(key_func=CustomListKeyConstructor())
+    def list(self, *args, **kwargs):
+        return super(AlbumAutoListViewSet, self).list(*args, **kwargs)
+
+
+
 class AlbumPersonViewSet(viewsets.ModelViewSet):
     queryset = Person.objects.all().order_by('name')
     serializer_class = AlbumPersonSerializer

ownphotos/urls.py

@@ -22,6 +22,8 @@
 
 router = routers.DefaultRouter()
 
+router.register(r'api/albums/auto/list', views.AlbumAutoListViewSet)
+
 router.register(r'api/albums/auto', views.AlbumAutoViewSet)
 router.register(r'api/albums/person', views.AlbumPersonViewSet)
 router.register(r'api/albums/date', views.AlbumDateViewSet)