o wi@sdZddlmZddlmZddlmZmZddlm Z ddl m Z ddl Z ddl mZer6d d lmZmZGd d d eZdS) aW&B Public API for managing users and API keys. This module provides classes for managing W&B users and their API keys. Note: This module is part of the W&B Public API and provides methods to manage users and their authentication. Some operations require admin privileges. ) annotations)MutableMapping) TYPE_CHECKINGAny)Self)gqlN)Attrs)ApiRetryingClientcs|eZdZdZd%fdd Zed&d d Zed'd(ddZed)ddZ ed)ddZ d*ddZ d+d,d!d"Z d-d#d$Z ZS).UseraxA user on a W&B instance. This allows managing a user's API keys and accessing information like team memberships. The `create` class method can be used to create a new user. Args: client: The GraphQL client to use for network operations. attrs: A subset of the User type in the GraphQL schema. clientr attrsMutableMapping[str, Any]cst|||_d|_dSN)super__init___client _user_api)selfr r __class__K/home/ubuntu/.local/lib/python3.10/site-packages/wandb/apis/public/users.pyr's  z User.__init__return Api | NonecCs*|jdur|jrtj|jdd|_|jS)z4A `wandb.Api` instance using the user's credentials.Nr)api_key)rapi_keyswandbr rrrruser_api,sz User.user_apiFapir emailstradminboolrc CsJddlm}m}t|}|j|||d}||jj}||j| S)aCreate a new user. This is an internal method. Use the `create_user()` method of `wandb.Api` instead. Args: api: The API instance to use to create the user. email: The email for the user. admin: Whether this user should be a global instance admin. Returns: A `User` object. r)CREATE_USER_FROM_ADMIN_GQLCreateUserFromAdmin)r"r$) wandb.apis._generatedr&r'rr executemodel_validateresultuser model_dump) clsr!r"r$r&r'gql_opdatar,rrrcreate3s z User.create list[str]cC,|jddur gSdd|jddDS)a4Names of the user's API keys. This property returns the names of the the API keys, *not* the secret associated with the key. The name of the key cannot be used as an API key. The list is empty if the user has no API keys or if API keys have not been loaded. apiKeysNcSg|]}|ddqSnodenamer.0krrr [z!User.api_keys..edges_attrsgetrrrrrNs z User.api_keyscCr3)zNames of the user's teams. This is an empty list if the user has no team memberships or if teams data was not loaded. teamsNcSr5r6rr9rrrr<fr=zUser.teams..r>r?rrrrrB]sz User.teamsrcCspddlm}ddlm}|j|}|jdd|dd}z|jt |d|iWd S|y7YdSw) a1Delete a user's API key. Only the owner of the key or an admin can delete it. Args: api_key: The name of the API key to delete. Use one of the names returned by the `api_keys` property. Returns: True on success, false on failure. r HTTPError)DELETE_API_KEY_GQLr4r>r7idFT) requestsrDr(rErindexr@rr)r)rrrDrEidx api_key_idrrrdelete_api_keyhs   zUser.delete_api_keyN description str | Nonec Csddlm}ddlm}m}z't|}|jj|d|i}| |j j }|j dd d|iW|jS|tfyAYdSw) a.Generate a new API key. Args: description: A description for the new API key. This can be used to identify the purpose of the API key. Returns: The generated API key (the full secret, not just the name), or None on failure. rrC)GENERATE_API_KEY_GQLGenerateApiKeyrLr4r>r7N)rGrDr(rNrOrr r r)r*r+rr@appendr-AttributeErrorr8)rrLrDrNrOr/r0 key_fragmentrrrgenerate_api_keys zUser.generate_api_keycCst|jd}rd|dS|jd}rd|dS|jd}r*d|dS|jd}r8d|dSdS)Nr"zusernamerFr8z r?)rr"rUid_r8rrr__repr__s    z User.__repr__)r r rr)rr)F)r!r r"r#r$r%rr)rr2)rr#rr%r)rLrMrrM)rr#)__name__ __module__ __qualname____doc__rpropertyr classmethodr1rrBrKrSrW __classcell__rrrrr s     r )r[ __future__rcollections.abcrtypingrrtyping_extensionsr wandb_gqlrrwandb.apis.attrsrr!r r r rrrrs