Merge pull request #43 from superannotateai/SAS-3444

Add mixp events

Author: dshabin

superannotate/db/project_images.py

@@ -600,7 +600,7 @@ def assign_images(project, image_names, user):
             response.status_code, "Couldn't assign images " + response.text
         )
 
-
+@Trackable
 def assign_folder(project, folder_name, users):
     """Assigns folder to users. With SDK, the user can be
     assigned to a role in the project with the share_project function.
@@ -647,7 +647,7 @@ def assign_folder(project, folder_name, users):
         )
     logger.info(f'Assigned {folder_name} to users: {list(verified_users)}')
 
-
+@Trackable
 def unassign_folder(project, folder_name):
     """Removes assignment of given folder for all assignees. 
     With SDK, the user can be assigned to a role in the project 
@@ -677,7 +677,7 @@ def unassign_folder(project, folder_name):
             response.status_code, "Couldn't unassign folder " + response.text
         )
 
-
+@Trackable
 def unassign_images(project, image_names):
     """Removes assignment of given images for all assignees.With SDK,
     the user can be assigned to a role in the project with the share_project

superannotate/mixp/utils/parsers.py

@@ -1655,4 +1655,54 @@ def delete_images(*args, **kwargs):
                 "project_name": get_project_name(project),
                 "Image Count": len(image_names),
             }
-    }
+    }
+
+
+def unassign_folder(*args, **kwargs):
+    return {"event_name": "unassign_folder", "properties": {}}
+
+
+def assign_folder(*args, **kwargs):
+    users = kwargs.get("users", None)
+    if not users:
+        users = args[2]
+    return {
+        "event_name": "assign_folder",
+        "properties": {
+            "User Count": len(users),
+        }
+    }
+
+def unassign_images(*args, **kwargs):
+    image_names = kwargs.get("image_names", None)
+    if not image_names:
+        image_names = args[1]
+
+    project = kwargs.get("project", None)
+    if not project:
+        project = args[0]
+
+    from superannotate.db.project_api import get_project_and_folder_metadata
+    project, folder = get_project_and_folder_metadata(project)
+    is_root = True
+    if folder:
+        is_root = False
+
+    return {
+        "event_name": "unassign_images",
+        "properties": {
+            "Assign Folder": is_root,
+            "Image Count": len(image_names)
+        }
+    }
+
+
+
+
+# def unassign_images(project, image_names):
+
+
+
+#
+# Assign Folder: IsRoot(project) ,
+# Image Count: len(image_names),