o Mf) @sdZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZmZdd lmZmZmZmZmZdd lmZmZddlmZm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&m'Z'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2m3Z3ddl4m5Z5e6e7Z8ee j9Z:e dedgddddGdddej;Zd$ej?ej@d%d&gd'dGd(d)d)ejAZBe d*edgd+d,ej>d-ej?ej@d%d&ej>d.ej?ej@d/d&gd0d1d2d3d4dGd5d6d6eejCZDe dedgd7d8dde d9edgd:d;ddGdgd?d@dAeidBdGdCdDdDejGZHe dEed>gdFdGdAeidBdGdHdIdIe.ZIdS)JzThis file and its contents are licensed under the Apache License 2.0. Please see the included NOTICE for copyright information and LICENSE for a copy of the license. N)flag_set)GetParentObjectMixin) load_func)settings)reverse)method_decorator)openapi)swagger_auto_schema) OrganizationOrganizationMember)OrganizationIdSerializerOrganizationInviteSerializer OrganizationMemberUserSerializerOrganizationSerializerOrganizationsParamsSerializer)genericsstatus)NotFoundPermissionDenied)get_object_or_404)PageNumberPagination) FormParser JSONParserMultiPartParser)IsAuthenticated)Response)APIView)User)ViewClassPermissionall_permissions)bool_from_requestget OrganizationszList your organizationsz_ Return a list of the organizations you've created or that you have access to. )tagsoperation_summaryoperation_description)name decoratorcsneZdZejZeee fZ e e j e je je je jdZeZddZfddZeddfdd ZZS) OrganizationListAPI)GETPUTPOSTPATCHDELETEcCs|j|jjjjdddS)NT)deleted_at__isnull)organizationmember__in)filterrequestuser om_throughdistinct)selfquerysetr7`/var/www/html/analyze/labelStudio/lib/python3.10/site-packages/label_studio/organizations/api.pyfilter_queryset=s z#OrganizationListAPI.filter_querysetctt|j|g|Ri|SN)superr(r!r5r1argskwargs __class__r7r8r!BzOrganizationListAPI.getN auto_schemacr:r;)r<r(postr=r@r7r8rEEzOrganizationListAPI.post)__name__ __module__ __qualname__r objectsallr6rrrparser_classesrrorganizations_vieworganizations_changeorganizations_createpermission_requiredr serializer_classr9r!r rE __classcell__r7r7r@r8r('s  r(cs$eZdZdZdZfddZZS)OrganizationMemberPagination page_sizecs,|j|jvr|j|jdkrdSt|S)Nz-1i@B)page_size_query_param query_paramsr< get_page_size)r5r1r@r7r8rXNs  z*OrganizationMemberPagination.get_page_size)rGrHrIrUrVrXrRr7r7r@r8rSJsrSzGet organization members listz:Retrieve a list of the organization members and their IDs.idz5A unique integer value identifying this organization.)r&typein_ description)r#r$r%manual_parametersc@sFeZdZeeefZeej ej ej ej dZ e Z eZddZddZdS)OrganizationMemberListAPI)r)r*r,r-cCst|jjdd|jdS)Ncontributed_to_projectsF)r_r1)r r1r))r5r7r7r8get_serializer_contextssz0OrganizationMemberListAPI.get_serializer_contextcCsxtj|jjj|j|jd}td|jjr6t|jj d}|j dd|j d}|r0|j dS|jdS|jdS)Npk.fix_backend_dev_3134_exclude_deactivated_usersdataT)raise_exceptionactiveuser__username)rrr1r2 organizationsr? lookup_fieldrrr)is_validvalidated_datar!active_membersorder_bymembers)r5org serializerrgr7r7r8 get_querysetys     z&OrganizationMemberListAPI.get_querysetN)rGrHrIrrrrLrrrMrNrPrrQrSpagination_classr`rrr7r7r7r8r^Xs  r^deletez"Soft delete an organization memberz+Soft delete a member from the organization.rbuser_pkzPA unique integer value identifying the user to be deleted from the organization.zMember deleted successfully.zUser cannot soft delete self.Member not found)ii)r#r$r%r] responsesc@sHeZdZeejdZej Z e e e fZeefZeZdgZdddZdS)OrganizationMemberDetailAPI)r-rtNcCsz|}||jjkrtdtt|d}tt||d}|jdur$td|j |jj kr4t ddit j dS|t ddS) Nz@You can delete members only for your current active organizationra)r2 organizationrvdetailzUser cannot soft delete selfrrw)get_parent_objectr2active_organizationrrrr deleted_atruser_idrYrrHTTP_405_METHOD_NOT_ALLOWED soft_delete)r5r1rbrurpr2memberr7r7r8rts    z"OrganizationMemberDetailAPI.delete)NN)rGrHrIrrrNrPr rJrKparent_querysetrrrrLrHasObjectPermissionpermission_classesrrQhttp_method_namesrtr7r7r7r8rys  ryz Get organization settingsz8Retrieve the settings for a specific organization by ID.patchzUpdate organization settingsz6Update the settings for a specific organization by ID.csdeZdZeeefZej Z e j Z eZdZdZfddZfddZeddfd d ZZS) OrganizationAPIzorganizations-dashboardrbcr:r;)r<rr!r=r@r7r8r!rBzOrganizationAPI.getcr:r;)r<rrr=r@r7r8rrBzOrganizationAPI.patchNrCcr:r;)r<rputr=r@r7r8rrFzOrganizationAPI.put)rGrHrIrrrrLr rJrKr6rrNrPrrQredirect_routeredirect_kwargr!rr rrRr7r7r@r8rs    rInviteszGet organization invite linkzWGet a link to use to invite a new member to an organization in Label Studio Enterprise.)r#r$r%rxc@s*eZdZefZejZe j Z ddZ dS)OrganizationInviteAPIcOsb|jj}dtd|j}ttdrtjr|tjdd}t ||jdd}| t |j dd S) N {}?token={} user-signupFORCE_SCRIPT_NAME invite_urltokenrdrr|) r2r~formatrrhasattrrrreplacer rkrrer5r1r>r?rprrqr7r7r8r!szOrganizationInviteAPI.getN) rGrHrIrrLr rJrKr6rrNrPr!r7r7r7r8rs   rrEzReset organization tokenzQReset the token used in the invitation link to invite someone to an organization.c@s eZdZejZefZddZdS)OrganizationResetTokenAPIcOsd|jj}|td|jd|jdtd|j}t ||jdd}| t |j ddS) NzNew token for organization z is rrrrdr|) r2r~ reset_tokenloggerdebugrbrrrr rkrrerr7r7r8rE szOrganizationResetTokenAPI.postN) rGrHrIrorganizations_inviterPrrLrEr7r7r7r8rs  r)J__doc__loggingcore.feature_flagsr core.mixinsrcore.utils.commonr django.confr django.urlsrdjango.utils.decoratorsrdrf_yasgrdrf_yasg.utilsr organizations.modelsr r organizations.serializersr r rrrrest_frameworkrrrest_framework.exceptionsrrrest_framework.genericsrrest_framework.paginationrrest_framework.parsersrrrrest_framework.permissionsrrest_framework.responserrest_framework.viewsr users.modelsrlabel_studio.core.permissionsrrlabel_studio.core.utils.paramsr getLoggerrGr MEMBER_PERMrListCreateAPIViewr(rS Parameter TYPE_INTEGERIN_PATH ListAPIViewr^RetrieveDestroyAPIViewryRetrieveUpdateAPIViewrRetrieveAPIViewrrr7r7r7r8s                  "