diff --git a/superannotate/db/project_images.py b/superannotate/db/project_images.py index 6592b6748..a539540ad 100644 --- a/superannotate/db/project_images.py +++ b/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 diff --git a/superannotate/mixp/utils/parsers.py b/superannotate/mixp/utils/parsers.py index f3001892a..dd1abb875 100644 --- a/superannotate/mixp/utils/parsers.py +++ b/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), } - } \ No newline at end of file + } + + +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),