]jyJddlmZddlmZddlmZddlmZGddeZy) )Optional) BaseModel)Client)UserceZdZUeed<eed<dZeeed<eed<eed<eed<e ed<Gd d Z d e fd Z d e fd Z dZy) Workspaceidtitle descriptioncolor is_personal created_byclientceZdZdZy)Workspace.ConfigTN)__name__ __module__ __qualname__arbitrary_types_allowedM/root/env/lib/python3.12/site-packages/label_studio_sdk/_legacy/workspaces.pyConfigrs"&rruserc|jjdd|jd|j|jd}|jS)zAdd user to workspace Parameters ---------- user: label_studio_sdk.users.User User POST/api/workspaces/ /memberships workspacerjson)r make_requestr r$selfrresponses radd_userzWorkspace.add_usersN;;++ twwi| 4#ww8,  }}rc|jjdd|jd|j|jd}|jdk7rt t |j y)zRemove user from workspace Parameters ---------- user: label_studio_sdk.users.User User DELETErr r!r#N)rr%r status_code ValueErrorstrcontentr&s r remove_userzWorkspace.remove_user$sk;;++ twwi| 4#ww8,    3 &S!1!123 3 'rcddlm}g}|jjdd|jd}|j }|D]3}|d}|j |j|j|5|S)zGet projects in current workspace Returns ------- projects: list of label_studio_sdk.project.Project Project r)ProjectGETrz /projectsr )r project_id)projectr3rr%r r$append get_from_id)r'r3 final_resultsr(projects project_datar5s r get_projectszWorkspace.get_projects4s % ;;++ %dggYi8 ==?$ L%d+J  ##;;)$  rN)rrrint__annotations__r/r rboolrrrr)r1r<rrrr r sV G J!#K## JO N'' T 44 rr N) typingrpydanticrrrusersrr rrrrCsB Br