Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions superannotate/db/project_images.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down
52 changes: 51 additions & 1 deletion superannotate/mixp/utils/parsers.py
Original file line number Diff line number Diff line change
Expand Up @@ -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),