Add assign_images,df_to_annotations

Author: dshabin

src/superannotate/__init__.py

@@ -1,4 +1,3 @@
-from superannotate.lib.app.analytics.class_analytics import aggregate_annotations_as_df
 from superannotate.lib.app.analytics.class_analytics import attribute_distribution
 from superannotate.lib.app.analytics.class_analytics import class_distribution
 from superannotate.lib.app.annotation_helpers import add_annotation_bbox_to_json
@@ -38,6 +37,7 @@
 from superannotate.lib.app.interface.sdk_interface import (
     add_annotation_template_to_image,
 )
+from superannotate.lib.app.interface.sdk_interface import aggregate_annotations_as_df
 from superannotate.lib.app.interface.sdk_interface import assign_folder
 from superannotate.lib.app.interface.sdk_interface import assign_images
 from superannotate.lib.app.interface.sdk_interface import attach_image_urls_to_project

src/superannotate/lib/app/interface/sdk_interface.py

@@ -994,6 +994,7 @@ def delete_images(project, image_names=None):
     )
 
 
+@Trackable
 def assign_images(project, image_names, user):
     """Assigns images to a user. The assignment role, QA or Annotator, will
     be deduced from the user's role in the project. With SDK, the user can be
@@ -1009,6 +1010,21 @@ def assign_images(project, image_names, user):
     project_name, folder_name = extract_project_folder(project)
     if not folder_name:
         folder_name = "root"
+
+    contributors = controller.get_project_metadata(
+        project_name=project_name, include_contributors=True
+    ).data["contributors"]
+    contributor = None
+    for c in contributors:
+        if c["user_id"] == user:
+            contributor = user
+
+    if not contributor:
+        logger.warning(
+            f"Skipping {user}. {user} is not a verified contributor for the {project_name}"
+        )
+        return
+
     controller.assign_images(project_name, folder_name, image_names, user)
 
 
@@ -3338,3 +3354,49 @@ def _upload_s3_image(image_path: str):
         duplicates.extend(duplications)
 
     return uploaded, failed_images, duplicates
+
+
+@Trackable
+def aggregate_annotations_as_df(
+    project_root,
+    include_classes_wo_annotations=False,
+    include_comments=False,
+    include_tags=False,
+    verbose=True,
+    folder_names=None,
+):
+    """Aggregate annotations as pandas dataframe from project root.
+
+    :param project_root: export path of the project
+    :type project_root: Pathlike (str or Path)
+    :param include_classes_wo_annotations: enables inclusion of classes info
+                                           that have no instances in annotations
+    :type include_classes_wo_annotations: bool
+    :param include_comments: enables inclusion of comments info as commentResolved column
+    :type include_comments: bool
+    :param include_tags: enables inclusion of tags info as tag column
+    :type include_tags: bool
+    :param folder_names: Aggregate the specified folders from project_root.
+                                If None aggregate all folders in the project_root.
+    :type folder_names: (list of str)
+
+    :return: DataFrame on annotations with columns:
+                                        "imageName", "instanceId",
+                                        "className", "attributeGroupName", "attributeName", "type", "error", "locked",
+                                        "visible", "trackingId", "probability", "pointLabels",
+                                        "meta" (geometry information as string), "commentResolved", "classColor",
+                                        "groupId", "imageWidth", "imageHeight", "imageStatus", "imagePinned",
+                                        "createdAt", "creatorRole", "creationType", "creatorEmail", "updatedAt",
+                                        "updatorRole", "updatorEmail", "tag", "folderName"
+    :rtype: pandas DataFrame
+    """
+    from superannotate.lib.app.analytics.common import aggregate_annotations_as_df
+
+    aggregate_annotations_as_df(
+        project_root,
+        include_classes_wo_annotations,
+        include_comments,
+        include_tags,
+        verbose,
+        folder_names,
+    )

src/superannotate/lib/app/mixp/utils/parsers.py

@@ -1102,20 +1102,24 @@ def assign_images(*args, **kwargs):
     user = kwargs.get("user", None)
     if not user:
         user = args[2]
-    from superannotate.db.users import get_team_contributor_metadata
 
-    res = get_team_contributor_metadata(user)
+    contributors = controller.get_project_metadata(
+        project_name=project, include_contributors=True
+    ).data["contributors"]
+    contributor = None
+    for c in contributors:
+        if c["user_id"] == user:
+            contributor = c
     user_role = "ADMIN"
-    if res["user_role"] == 3:
+    if contributor["user_role"] == 3:
         user_role = "ANNOTATOR"
-    if res["user_role"] == 4:
+    if contributor["user_role"] == 4:
         user_role = "QA"
-    from superannotate.db.project_api import get_project_and_folder_metadata
-
-    project, folder = get_project_and_folder_metadata(project)
+    _, folder_name = extract_project_folder(project)
     is_root = True
-    if folder:
+    if folder_name:
         is_root = False
+
     return {
         "event_name": "assign_images",
         "properties": {

src/superannotate/lib/infrastructure/services.py

@@ -148,7 +148,7 @@ class SuperannotateBackendService(BaseBackendService):
     URL_COPY_IMAGES_FROM_FOLDER = "images/copy-image-or-folders"
     URL_MOVE_IMAGES_FROM_FOLDER = "image/move"
     URL_GET_COPY_PROGRESS = "images/copy-image-progress"
-    URL_ASSIGN_IMAGES = "images/editAssignment"
+    URL_ASSIGN_IMAGES = "images/editAssignment/"
     URL_ASSIGN_FOLDER = "folder/editAssignment"
     URL_S3_ACCESS_POINT = "/project/{}/get-image-s3-access-point"
     URL_S3_UPLOAD_STATUS = "/project/{}/getS3UploadStatus"