Removed events from annotator post

jsbroks · Apr 4, 2019 · e2fe385 · e2fe385
1 parent 341aeba
commit e2fe385
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/backend/webserver/api/annotator.py b/backend/webserver/api/annotator.py
@@ -137,7 +137,8 @@ class AnnotatorId(Resource):
     @login_required
     def get(self, image_id):
         """ Called when loading from the annotator client """
-        image = ImageModel.objects(id=image_id).first()
+        image = ImageModel.objects(id=image_id)\
+            .exclude('events').first()
 
         if image is None:
             return {'success': False, 'message': 'Could not load image'}, 400
@@ -146,8 +147,8 @@ def get(self, image_id):
         if dataset is None:
             return {'success': False, 'message': 'Could not find associated dataset'}, 400
 
-        categories = CategoryModel.objects(deleted=False).in_bulk(dataset.categories).items()
-
+        categories = CategoryModel.objects(deleted=False)\
+            .in_bulk(dataset.categories).items()
 
         # Get next and previous image
         images = ImageModel.objects(dataset_id=dataset.id, deleted=False)
@@ -177,7 +178,8 @@ def get(self, image_id):
             category = query_util.fix_ids(category[1])
 
             category_id = category.get('id')
-            annotations = AnnotationModel.objects(image_id=image_id, category_id=category_id, deleted=False).all()
+            annotations = AnnotationModel.objects(image_id=image_id, category_id=category_id, deleted=False)\
+                .exclude('events').all()
 
             category['show'] = True
             category['visualize'] = False