o wi@sdZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z er2dd lmZmZGd d d e ZGd d d e ZdS)a-W&B Public API for managing teams and team members. This module provides classes for managing W&B teams and their members. Note: This module is part of the W&B Public API and provides methods to manage teams and their members. Team management operations require appropriate permissions. ) annotations)Mapping) TYPE_CHECKINGAny)Self)gql)Attrs)ApiRetryingClientcs2eZdZdZdfdd Zd d Zd d ZZS)MemberzA member of a team. Args: client (`wandb.apis.internal.Api`): The client instance to use team (str): The name of the team this member belongs to attrs (dict): The member attributes clientr teamstrattrsMapping[str, Any]cst|||_||_dSN)super__init___clientr)selfr rr __class__K/home/ubuntu/.local/lib/python3.10/site-packages/wandb/apis/public/teams.pyr"s  zMember.__init__cCslddlm}ddlm}m}z|jt||j|j d}Wn |y(YdSw| |j }|duo5|j S)z^Remove a member from a team. Returns: Boolean indicating success r HTTPError)DELETE_INVITE_GQL DeleteInvite)identityFN) requestsrwandb.apis._generatedrrrexecuterrrmodel_validateresultsuccess)rrrrdatar%rrrdelete's   z Member.deletecCsd|jd|jdS)Nz)name account_typerrrr__repr__;szMember.__repr__)r r rrrr)__name__ __module__ __qualname____doc__rr(r, __classcell__rrrrr s r csbeZdZdZ d$d%fd d Zed$d&ddZd'd(ddZd)ddZd'd*d d!Z d+d"d#Z Z S),TeamaA class that represents a W&B team. This class provides methods to manage W&B teams, including creating teams, inviting members, and managing service accounts. It inherits from Attrs to handle team attributes. Args: client (`wandb.apis.public.Api`): The api instance to use name (str): The name of the team attrs (dict): Optional dictionary of team attributes Note: Team management requires appropriate permissions. Nr r r)rrMapping[str, Any] | Nonecs(t|pi||_||_|dSr)rrrr)load)rr r)rrrrrOs z Team.__init__apir radmin_username str | NonereturnrcCsTddlm}ddlm}z|jt|||dWn |y#Ynw||j|S)a'Create a new team. Args: api: (`Api`) The api instance to use team: (str) The name of the team admin_username: (str) optional username of the admin user of the team, defaults to the current user. Returns: A `Team` object rr)CREATE_TEAM_GQL)teamNameteamAdminUserName)r!rr"r9r r#r)clsr5rr6rr9rrrcreateZs   z Team.createFusername_or_emailadminboolcCsdddlm}ddlm}d|jd|d|vrdnd|i}z |jt||Wd S|y1Yd Sw) apInvite a user to a team. Args: username_or_email: (str) The username or email address of the user you want to invite. admin: (bool) Whether to make this user a team admin. Defaults to `False`. Returns: `True` on success, `False` if user was already invited or didn't exist. rr)CREATE_INVITE_GQLr r?@emailusernameFT)r!rr"rAr)rr#r)rr>r?rrA variablesrrrinvitess  z Team.invite description Member | NonecCs^ddlm}ddlm}z|jt||j|d|d|j dWS|y.YdSw)zCreate a service account for the team. Args: description: (str) A description for this service account Returns: The service account `Member` object, or None on failure rr)CREATE_SERVICE_ACCOUNT_GQL)r rGTN) r!rr"rIrr#rr)r4members)rrGrrIrrrcreate_service_accounts     zTeam.create_service_accountforcedict[str, Any]csxddlm}m}|s js9jt|dji}||}|j }r'| ni_fddjdDjd<jS)zYReturn members that belong to a team. r)GET_TEAM_ENTITY_GQL GetTeamEntityr)csg|] }tjj|qSr)r rr)).0memberr+rr szTeam.load..rK) r"rOrP_attrsrr#rr)r$r model_dump)rrMrOrPr'r%r rr+rr4s    z Team.loadcCsd|jdS)Nz)r)r+rrrr,sz Team.__repr__r)r r r)rrr3)r5r rrr6r7r8r)F)r>rr?r@r8r@)rGrr8rH)rMr@r8rN)r8r) r-r.r/r0r classmethodr=rFrLr4r,r1rrrrr2?s    r2N)r0 __future__rcollections.abcrtypingrrtyping_extensionsr wandb_gqlrwandb.apis.attrsrr5r r r r2rrrrs    &