Axjq dZddlZddlZddlZddlZddlmZmZej dZ dZ dZ dZ dZdS) z Runpod | CLI | SSH | Functions N)get_userupdate_user_settingsz ~/.runpod/sshch|}t|dkrtdtj|d}t j|}dtj| d dzS)z+ Get the fingerprint of an SSH key zInvalid SSH public keyzSHA256:zutf-8=) splitlen ValueErrorbase64 b64decodehashlibsha256digest b64encodedecodestrip) public_keypartskey_data fingerprints PC:\Projects\Historical\venv\Lib\site-packages\runpod/cli/groups/ssh/functions.pyget_ssh_key_fingerprintrs     E 5zzA~~1222a))H.**1133K v' 44;;GDDJJ3OO OOct}|ddn|d}|d}d|D}g}|D]}|d}t|dkr+i}|d|d <|d |d <t||d <t|dkr|dnd |d<|||S)z) Get the current user's SSH keys pubKeyN cg|] }|dk| S)r).0keys r z%get_user_pub_keys..&s - - -C3"99C999r rrtyperr"rzN/Aname)rr r rappend)userkeyskey_listr" key_partskey_dicts rget_user_pub_keysr-s ::Dh'22T(^D ::d  D - -4 - - -DH " "IIcNN  y>>A   $Q<#A,"9#">">+.y>>A+=+=9Q<<5!!!! Orctjtjtdt jd}|tjt|tj tjt|dttd|ddd 5}| d | d |}| |d d d n #1swxYwYt|||fS) z Generate an RSA SSH key pair and save it to disk. Args: - filename (str): The base filename to save the key pair. The public key will have '.pub' appended to it. T)exist_oki)bitsi/z.pubwzUTF-8)encodingr$N)osmakedirspathjoin SSH_FILESparamikoRSAKeygeneratewrite_private_key_filechmodopenget_name get_base64write add_ssh_key)filename private_key public_filers rgenerate_ssh_key_pairrF:seK Y''$7777/***55K&&rw||Ix'H'HIIIHRW\\)X . .666 ++X+++S7 C C C&{#,,..VV1G1G1I1IVVHVV *%%%&&&&&&&&&&&&&&&   ""s&AD66D:=D:ct}|ddn|d}||vrtddS||rdndz|z}t|dS)zwAdd an SSH public key to the current user's Runpod account. Checks if the key already exists before adding it. rNrzKey already existsz )rprintr)rr( current_keys updated_keyss rrBrBUsu ::Dh/22T(^L\!! "###\#A66rBZOLL*+++++r)__doc__r rr4r9runpod.api.ctl_commandsrrr6 expanduserr8rr-rFrBr rrrNs  BBBBBBBB G   / /  P P P:###6,,,,,r