Fixed annotation metadata copy

Author: jsbroks

backend/database/annotations.py

@@ -11,8 +11,8 @@
 
 class AnnotationModel(DynamicDocument):
 
-    COCO_PROPERTIES = ["id", "image_id", "category_id", "segmentation", \
-                       "iscrowd", "color", "area", "bbox", "metadata", \
+    COCO_PROPERTIES = ["id", "image_id", "category_id", "segmentation",
+                       "iscrowd", "color", "area", "bbox", "metadata",
                        "keypoints"]
 
     id = SequenceField(primary_key=True)
@@ -30,7 +30,7 @@ class AnnotationModel(DynamicDocument):
     height = IntField()
 
     color = StringField()
-    
+
     keypoints = ListField(default=[])
 
     metadata = DictField(default={})
@@ -42,9 +42,8 @@ class AnnotationModel(DynamicDocument):
     milliseconds = IntField(default=0)
     events = EmbeddedDocumentListField(Event)
 
-
     def __init__(self, image_id=None, **data):
-        
+
         from .images import ImageModel
 
         if image_id is not None:
@@ -60,7 +59,7 @@ def __init__(self, image_id=None, **data):
 
     def save(self, copy=False, *args, **kwargs):
 
-        if not self.dataset_id and not copy:
+        if self.dataset_id and not copy:
             dataset = DatasetModel.objects(id=self.dataset_id).first()
 
             if dataset is not None:
@@ -113,7 +112,7 @@ def __call__(self):
         }
 
         return im.Annotation(**data)
-    
+
     def add_event(self, e):
         self.update(push__events=e)
 

backend/webserver/api/annotations.py

@@ -11,7 +11,8 @@
 api = Namespace('annotation', description='Annotation related operations')
 
 create_annotation = reqparse.RequestParser()
-create_annotation.add_argument('image_id', type=int, required=True, location='json')
+create_annotation.add_argument(
+    'image_id', type=int, required=True, location='json')
 create_annotation.add_argument('category_id', type=int, location='json')
 create_annotation.add_argument('metadata', type=dict, location='json')
 create_annotation.add_argument('segmentation', type=list, location='json')
@@ -41,8 +42,9 @@ def post(self):
         image = current_user.images.filter(id=image_id, deleted=False).first()
         if image is None:
             return {"message": "Invalid image id"}, 400
-        
-        logger.info(f'{current_user.username} has created an annotation for image {image_id}')
+
+        logger.info(
+            f'{current_user.username} has created an annotation for image {image_id}')
 
         try:
             annotation = AnnotationModel(
@@ -69,7 +71,7 @@ def get(self, annotation_id):
 
         if annotation is None:
             return {"message": "Invalid annotation id"}, 400
-        
+
         return query_util.fix_ids(annotation)
 
     @login_required
@@ -80,10 +82,12 @@ def delete(self, annotation_id):
         if annotation is None:
             return {"message": "Invalid annotation id"}, 400
 
-        image = current_user.images.filter(id=annotation.image_id, deleted=False).first()
+        image = current_user.images.filter(
+            id=annotation.image_id, deleted=False).first()
         image.flag_thumbnail()
 
-        annotation.update(set__deleted=True, set__deleted_date=datetime.datetime.now())
+        annotation.update(set__deleted=True,
+                          set__deleted_date=datetime.datetime.now())
         return {'success': True}
 
 
@@ -92,5 +96,3 @@ def delete(self, annotation_id):
 #     def get(self, annotation_id):
 #         """ Returns the binary mask of an annotation """
 #         return query_util.fix_ids(AnnotationModel.objects(id=annotation_id).first())
-
-