Axj& TdZddlZddlZddlZddlZddlmZddlmZm Z ddl m Z ej dd d Z e d d Ze d ejddejdddejdddejdddejddddZe dejddZe d ejd!ejd"ejd#d$%ejd&d$%d'ZdS)(z Runpod | CLI | Pod | Commands N) PrettyTable) create_podget_pods)ssh_cmdpodzManage and interact with pods.)helpcdS)z&A collection of CLI functions for Pod.Nr OC:\Projects\Historical\venv\Lib\site-packages\runpod/cli/groups/pod/commands.pypod_clirsr listctgd}tD]3}||d|d|d|df4tj|dS)z. Lists the pods for the current user. )IDNameStatusImageidname desiredStatus imageNameN)rradd_rowclickecho)tablers r list_podsrss 999 : :EzzXX s4y#f+s?/CSEUVWWWW Jur createrF)requiredz--imagezThe image to use for the pod.)defaultr z --gpu-typez The GPU type to use for the pod.z --gpu-countz&The number of GPUs to use for the pod.z--support-public-ipTz&Whether or not to support a public IP.c||d}|stjdd}tjdd}|rd}d }d |d <tjd t |||fi|}tjd |dddS)z Creates a pod. ) gpu_countsupport_public_ipzEnter pod namezRunPod-CLI-Podr z%Would you like to launch default pod?T)abortzrunpod/base:0.0.0zNVIDIA GeForce RTX 3090z22/tcpportszLaunching default pod...zPod rz has been created.N)rpromptconfirmrr)rimagegpu_typer#r$kwargs quick_launchnew_pods r create_new_podr/!s.F H|,6FGGG=!HPTUUUL/#,"w -...uh99&99G J7gdm77788888r connectpod_idctjd|dtj|}|dS)z Connects to a pod. zConnecting to pod ...N)rrr SSHConnectionlaunch_terminal)r1sshs r connect_to_podr7DsH  J/F///000   ' 'Cr sync source_pod_id dest_pod_idsource_workspacez /workspacer%dest_workspacec  ddlm}|}|stjdtjdtjdtjdtjdtjdtjdd Stjd t |d nO#t $rB}tjd t |tjd Yd }~nd }~wwxYwtjd|d|d|d|t tjd d}d|d}d|} tjd|dtj |5} tjd|d| j d|d\} } } | } tjd| d| j d|d\} } }| }|dkr`tjd|d| d d d  d t!vrt#j|d Sd S#t&$rYd SwxYwtjd!|dd"|}d#t"j|d$|d%t"j|}| |g| j d&|d'\} } } | }|d(krZtjd) d d d  d t!vrt#j|d Sd S#t&$rYd SwxYw| j d*|d+\} } } | }tjd,|d-tjd.t1jd/d05}|j}| ||d d d n #1swxYwY| d1|gd d d n #1swxYwYtjd2|dtj |5}tjd3|d|d4|gtjd5d"|}|||tjd6|d7| dd#|d8| d9| d:|d; }||g|j d|d7| d\} } } | }tjd<|d=|d1|gtjdtjd>tjd?| tjd@|d|d7| tjdtjdAtjdB|tjdC|d7| d d d n #1swxYwYn#t $r}tjdDt |tjdtjdEtjdFtjdGtjdHtjdIYd }~nd }~wwxYw d t!vrt#j|d Sd S#t&$rYd SwxYw# d t!vrt#j|ww#t&$rYwwxYwxYw)JuD Sync data between two pods via SSH. Transfers files from source_pod_id:source_workspace to dest_pod_id:dest_workspace. The workspace will be zipped and transferred to avoid file name conflicts. 📋 PREREQUISITES: 1. SSH Key Setup: • You must have an SSH key configured in your Runpod account • If you don't have one, create it with: runpod ssh add-key • List your keys with: runpod ssh list-keys 2. Pod Configuration: • Both pods must have SSH access enabled • For running pods using official Runpod templates, you may need to add your public key to the PUBLIC_KEY environment variable and restart the pod ⚠️ IMPORTANT NOTES: • If a pod was started before adding your SSH key, you'll need to: 1. Stop the pod 2. Add PUBLIC_KEY environment variable with your public key 3. Restart the pod • The sync creates a unique folder (sync_XXXXXXXX) in the destination to avoid file conflicts 📖 EXAMPLES: Basic sync (uses /workspace as default): runpod pod sync pod1 pod2 Custom paths: runpod pod sync pod1 pod2 /workspace/data /workspace/backup Different directories: runpod pod sync pod1 pod2 /home/user/files /workspace/imported r)get_user_pub_keysu-❌ No SSH keys found in your Runpod account!u🔑 To create an SSH key, run:z runpod ssh add-keyu📖 For more help, see:z runpod ssh add-key --helpNu ✅ Found z SSH key(s) in your accountu,⚠️ Warning: Could not verify SSH keys: zContinuing with sync attempt...u🔄 Syncing from :z to sync_z.tar.gzu📡 Connecting to source pod r3u📊 Counting files in zfind z -type f | wc -lu 📁 Found z files in source workspaceztest -d z% && echo 'exists' || echo 'not_found'existsu❌ Error: Source workspace z does not exist on pod local_temp_pathu📦 Creating archive of z/tmp/zcd z && tar -czf  ztest -f z# && echo 'created' || echo 'failed'createdu1❌ Error: Failed to create archive on source podzdu -h z | cut -f1u"✅ Archive created successfully ()u/⬇️ Downloading archive to local machine...F)deletesuffixzrm -f u#📡 Connecting to destination pod u%📂 Preparing destination workspace z mkdir -p u/⬆️ Uploading archive to destination pod...u📦 Extracting archive to /z && mkdir -p z && cd z && tar -xzf z --strip-components=1u📁 Extracted z files to destinationu!🎉 Sync completed successfully!u📊 Files transferred: u📍 Destination location: u 💡 To access the synced files:z runpod ssh z cd u❌ Error during sync: u🔧 Troubleshooting tips:u,• Ensure both pods have SSH access enableduQ• Check that your SSH key is added to your Runpod account: runpod ssh list-keysuH• For running pods, you may need to add PUBLIC_KEY env var and restartu1• Verify the source and destination paths exist)groups.ssh.functionsr>rrlen Exceptionstruuiduuid4rr4r6 exec_commandreaddecodestriplocalsosunlinkOSErrorpathdirnamebasename run_commandstempfileNamedTemporaryFilerget_fileput_file)r9r:r;r<r> user_keyse transfer_id temp_zip_name dest_folder source_ssh_stdout file_countstderrresultrD archive_path tar_commandarchive_result archive_size temp_filedest_sshdest_archive_pathextract_commanddest_file_counts r sync_podsruOs ^6======%%''  Q JF G G G JrNNN J8 9 9 9 J. / / / JrNNN J1 2 2 2 J5 6 6 6 F JOC NNOOO P P P P 666 J#a&&JJKKK 4555555556 JhMhh4Dhh+hhXfhhiiidjll##BQB'K0K000M'+''K` FMFFFGGG  "= 1 1( ?Z JF1AFFF G G G%>667a?O7a7a7abbLAvq--//5577J JKZKKK L L L!+ ; ;<~GW<~<~<~   Avv[[]]))++1133F!! r:Jrrcprrsss( ?( ?( ?( ?( ?( ?p  FHH,, /*****-,    DD W JH3CHHH I I I2=22LD0@ A ADDP\DD_a_f_o_oqA`B`BDDK  # #[M 2 2 2&>667s,7s7s7sttLAvq#[[]]113399;;N** NOOO5( ?( ?( ?( ?( ?( ?p  FHH,, /*****-,    DD }&>667X 7X7X7XYYLAvq!;;==//117799L JKLKKK L L L JH I I I,E)LLL CPY"+.##L/BBB C C C C C C C C C C C C C C C  # #% > >Q( ?( ?( ?( ?( ?( ?( ?( ?( ?( ?( ?( ?( ?( ?( ?V  IIIIJJJ  "; / / @8 JR~RRR S S S  ! !#?~#?#?"@ A A A JH I I I 7 7 7    o/@ A A A JV^VVkVVV W W WXNXXXXU`XXpAXXXO  ! !?"3 4 4 4$<445k^5k5kk5k5k5kllLAvq$kkmm2244::<*>> ? ? ? Ja[aa>aaT_aa b b b JrNNN J9 : : : J5 55 6 6 6 J>>>>> ? ? ?A @ @ @ @ @ @ @ @ @ @ @ @ @ @ @D HHH 5SVV55666 2 /000 ABBB fggg ]^^^ FGGGGGGGGH  FHH,, /*****-,    DD   FHH,, /****-    D s1BC"%C D8DD+,\=DT  \=)$K KK"C-T  \=$P PPBT 0S T S T !S "T > \= T\=T/\=G$\1% \=1\55\=8\59\=<`*= _-B_(#`*(_--`*1$` `'&`'*a",$aa" aa"aa")__doc__rVr]rOr prettytablerrunpodrrutilsrgrouprcommandrargumentoptionr/r7rur r r r~sq  ######'''''''' U9:::11;:1  ''' i,KLLL lD/QRRR mQ-UVVV 4.V99WVSRML('96     "L999 ,777cc87:9!  cccr