/XdZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Zdd l mZe j j"d e j j$d e j j&d iZdZGddej,ZGddeZGddeZy)zoslogin client functions.)absolute_import)division)unicode_literals)Optional)apis) apis_util)base) exceptions) propertiesv1alphav1betav1c.tjd|S)Noslogin)rGetClientInstance)versions ,lib/googlecloudsdk/api_lib/oslogin/client.py _GetClientr"s   7 33ceZdZdZy)OsloginExceptionzDOsloginException is for non-code-bug errors in oslogin client utils.N__name__ __module__ __qualname____doc__rrrr&sLrrceZdZdZy)OsloginKeyNotFoundErrorzFOsloginKeyNotFoundError is raised when requested SSH key is not found.Nrrrrrr*sNrrceZdZdZdZdZdZ ddZddZ dde e fd Z d Z d Z dd Zd Z ddZ dde e fdZy) OsloginClientz%Class for working with oslogin users.cHt||_tjjj j |_ t|j|_|jj|_ y#tj$rd|_d|_ YywxYwN) VERSION_MAPrr VALUEScoreprojectGetrclientMESSAGES_MODULEmessagesrUnknownVersionError)self release_tracks r__init__zOsloginClient.__init__1sy}-DL$$))11557DLt||,dkkk11dm  ( (dkdms5A==!B! B!c"|jSr#)__bool__r-s r __nonzero__zOsloginClient.__nonzero__;s ==?rc,t|jSr#)boolr)r2s rr1zOsloginClient.__bool__>s  rNc<|jj}|jdk(r|dj|||}n6|r|jj }nd}|dj||||}|j jj|}|S)a[Return the OS Login profile for a user. The login profile includes some information about the user, a list of Posix accounts with things like home directory location, and public SSH keys for logging into instances. Args: user: str, The email address of the OS Login user. project: str, The project ID associated with the desired profile. system_id: str, If supplied, only return profiles associated with the given system ID. include_security_keys: bool, If true, include security key information. Returns: The login profile for the user. r users/{0})name projectIdsystemIdN)r8r9r:view) r+"OsloginUsersGetLoginProfileRequestrformatViewValueValuesEnum SECURITY_KEYr)usersGetLoginProfile) r-userr' system_idinclude_security_keysprofile_requestmessager;ress rrAzOsloginClient.GetLoginProfileAs&mmFFO ||t!!$'g 22??!!$' g ++   + +G 4C Jrch|ra|j}|jjj|}|jj|j |}n*|jj|j }|j j j|y)zDelete the posix accounts for an account in the current project. Args: project_ref: The oslogin.users.projects resource. operating_system: str, 'linux' or 'windows' (case insensitive). Returns: None )r8operatingSystemTyper8N)upperr+!OsloginUsersProjectsDeleteRequest"OperatingSystemTypeValueValuesEnum RelativeNamer)users_projectsDelete)r- project_refoperating_systemos_value os_messagerFs rDeletePosixAccountsz!OsloginClient.DeletePosixAccountsjs!'')hMMCC77A ??'')(@*g ??'')@+g KK%%g.rregionc|jj}|jj||}|r|gng}|jdk(r&|dj ||j ||} nA|r|j j} nd} |dj ||j ||| } |jjj| } | S)aUpload an SSH public key to the user's login profile. Args: user: str, The email address of the OS Login user. public_key: str, An SSH public key. expiration_time: int, microseconds since epoch. include_security_keys: bool, If true, return security key information. region: str, The region to which to make sure the key is imported. Returns: The login profile for the user. keyexpirationTimeUsecrr7)parentr9 sshPublicKeyregionsN)r[r9r\r]r;) r+%OsloginUsersImportSshPublicKeyRequest SshPublicKeyrr=r'r>r?r)r@ImportSshPublicKey) r-rB public_keyexpiration_timerDrVimport_requestpublic_key_messager]rFr;rGs rr`z OsloginClient.ImportSshPublicKeys(]]HHN33 *4,!vhbG ||t##D)LL) g 11>>##D)LL) g ++   . .w 7C Jrc|jjdj||}|jjj |y)zDelete an SSH public key from the user's login profile. Args: user: str, The email address of the OS Login user. fingerprint: str, The fingerprint of the SSH key to delete. Returns: None users/{0}/sshPublicKeys/{1}rJN)r+&OsloginUsersSshPublicKeysDeleteRequestr=r)users_sshPublicKeysrP)r-rB fingerprintrFs rDeleteSshPublicKeyz OsloginClient.DeleteSshPublicKeysGmmBB * 1 1$ DCFGKK##**73rc|jjdj||}|jjj |}|S)zGet an SSH public key from the user's login profile. Args: user: str, The email address of the OS Login user. fingerprint: str, The fingerprint of the SSH key to delete. Returns: The requested SSH public key. rfrJ)r+#OsloginUsersSshPublicKeysGetRequestr=r)rhr()r-rBrirFrGs rGetSshPublicKeyzOsloginClient.GetSshPublicKeysNmm?? * 1 1$ D@FG ++ ) ) - -g 6C Jrc|jj||}|jjdj||||}|jj j |}|S)aUpdate an existing SSH public key in a user's login profile. Args: user: str, The email address of the OS Login user. fingerprint: str, The fingerprint of the SSH key to update. public_key: str, An SSH public key. update_mask: str, A mask to control which fields get updated. expiration_time: int, microseconds since epoch. Returns: The login profile for the user. rXrf)r8r\ updateMask)r+r_%OsloginUsersSshPublicKeysPatchRequestr=r)rhPatch) r-rBrira update_maskrbrdrFrGs rUpdateSshPublicKeyz OsloginClient.UpdateSshPublicKeysv33 ?4mmAA * 1 1$ D'BG ++ ) ) / / 8C Jrc|jj|}|jjdj||||}|jj j |S)aQSign an SSH public key for a given user. Args: user: str, The email address of the OS Login user. public_key: str, An SSH public key. project_id: str, The project ID associated with the VM. region: str, The region where the signed SSH public key may be used. Returns: A signed SSH public key. )r\z$users/{0}/projects/{1}/locations/{2})r[signSshPublicKeyRequest)r+SignSshPublicKeyRequest0OsloginUsersProjectsZonesSignSshPublicKeyRequestr=r)users_projects_zonesSignSshPublicKey)r-rBra project_idrVrdrFs rryzOsloginClient.SignSshPublicKeyst>>?mmLL5<< *f !3 MG ;; + + < rs &',1(=*  &&   %%x  ##T+ 4M,,MO.OqFqr