AxjD@dZddlZddlmZddlmZmZejddd Ze d d Z e d ej d ddej ddddZ dS)z Runpod | CLI | SSH | Commands N) PrettyTable)generate_ssh_key_pairget_user_pub_keyssshz8Manage and configure SSH keys for secure access to pods.)helpcdS)zManage and configure SSH keys.Nr OC:\Projects\Historical\venv\Lib\site-packages\runpod/cli/groups/ssh/commands.pyssh_clir sr z list-keysct}tgd}|D],}||d|d|df-tj|dS)z2 Lists the SSH keys for the current user. )z Public KeyType Fingerprintnametype fingerprintN)rradd_rowclickecho)key_listtablekeys r list_keysrss !""H === > >EFF s6{CK]1CDEEEE Jur zadd-keyz--keyzThe public key to add.)defaultrz --key-filez*The file containing the public key to add.c|sY|sWtjddtjddt}|dd}t |tjd d S) zh Adds an SSH key to the current user account. If no key is provided, one will be generated. z1Would you like to add an SSH key to your account?T)abortz Please enter a name for this keyz RunPod-Key)rr -z'The key has been added to your account.N)rconfirmpromptstrreplacerr)rkey_filekey_names r add_keyr&s (x( IQUVVVV< . 3   ##C--h''' J899999r ) __doc__r prettytabler functionsrrgroupr commandroptionr&r r r r-s ######???????? USTTT))UT)   gt*BCCC $%Q : :DC : : :r