o 2wit@stdZddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ee dZ d efd d ZdS) zo The info commands gets the information on a Sheep or XP and can be used to retrieve the job status, logs etc. )partialN) DecoratedMain)Shepherd) simple_logfatalzInfo:maincCst|}|jdur#t|jdkrtd||j}|dur"tdn||j}td|td|jj |jr>td|j|j r^| |j}dt|d}|rZ|t |d 7}t||jry|jdurktd n|rttd n|j|jr|jdurtd |jstd |jdtt|jdtjd|jr|jstd |jdtddddd|jgdSdS)Nrz1If a job id is provided, you shouldn't pass argv.z!Could not find any matching sheepz Found sheepz Folder isz Main log iszMetrics[z]: z!Could not cancel non existing jobzJob is not runningz(No log, sheep hasn't been scheduled yet.zLog z does not existritailz-n200z-f)rjob_idlenargvrget_sheep_from_job_idget_sheep_from_argvlogxpfoldermetricsget_xp_historyjsondumpscanceljobis_doneexistsshutil copyfileobjopensysstdoutr osexecvp)argsrshepherdsheeproutr(F/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/dora/info.py info_actionsH             r*)__doc__ functoolsrrr"rr rrsheprrrrr*r(r(r(r)s