@i`ddlmZddlZddlZddlZddlZddlZddl Z ddl m Z ddl mZddlmZddlmZej&eZGdd eZd Zd e d efd ZdZe dfde fdZde de fdZd ede fdZede fdZdede fdZe j@de!e Z"de"de"fdZ#y))contextmanagerN)Path) DecoratedMain)fatal)XPc eZdZy) CommandErrorN)__name__ __module__ __qualname__b/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/dora/git_save.pyr r srr c rtj|ftjtjd|}|jrOdj d|D}t d|d|jd|jjz|jjjS)N)stdoutstderr c3FK|]}tj|yw)N)shlexquote).0cs r zrun_command..s?w!u{{1~ws!zCommand z failed (z): ) sprunPIPESTDOUT returncodejoinr rdecodestrip)commandkwargsproc command_strs r run_commandr's 66' F"''")) Fv FD hh?w?? {m9T__,=U CdkkFXFXFZ Z\ \ ;;    % % ''rrootmainctgd}g}|jj}||jdz}tj j |}d}|;|jJt|jjj}|jdD]}|stj|}g} t|dk(r| j|dn't|dk(r|ddk(sJ| |d|dgz } n d } | D]-} |d } n'|| z j} | j|/| r|j||r dj#|} t%d | yy#t $rd } YvwxYw) N)gitstatusz --porcelainz.grids rz->TFzRRepository is not clean! The following files should be commited or git ignored: )r'dora grid_packagepackage importlibutil find_specoriginrresolveparentsplitrlenappend relative_to ValueErrorr r)r(r)outfiltered grid_namespec grid_pathlinepartspaths line_cleanpathrpathfiless rcheck_repo_cleanrK%s 6 7CH &&ILL8+ >> # #I .D#'I {{&&&%--/66  $  D!  u:? LLq " Z1_8t# ## eAha) )E" D " D[))+E #!!),  OOD !/ 0 (# $$)7, -  #"  #s.E:: FFcJttgdjS)N)r+z rev-parsez--show-toplevel)rr'r8rrr get_git_rootrMQs  CD E M M OOr.repoc tgd|S)N)r+logz-1z --format=%H)cwd)r')rOs rget_git_commitrSUs : EErsourcetargetc |j|jdzz }tddddt|zt|gt |}|j|z }|j ||S)Nz.tmpr+clonez --depth=1zfile://)r9namer'strrSrename)rTrU tmp_target actual_commit actual_targets r shallow_cloner^Ysg&++"67Ji#f+.Es:WX #:.MMMM1Mm$ rreturnc,t}t}t|||jj|jj z }|j dd||z }|js t||}|jsJ|S)z,Return a fresh clone in side the given path.T)parentsexist_ok) rMrSrKr1dir_codesmkdirexistsr^)r)rTcommitcodesrUs r get_new_clonerifsx ^F  FVT" IIMMDII,, ,E KKtK, V^F ==?vv. ==?? MrrWc#Ktdj}t}|j|}t |t j d<t j||z  dt j|t j d=y#t j|t j d=wxYww)zcContext manager that temporarily relocates to a clean clone of the current git repository. rN_DORA_ORIGINAL_DIRN)rr8rMr=rYosenvironchdir)rWrRr( relative_paths r enter_clonerpts s)   C >DOOD)M'*3xBJJ#$HHU] "#-   JJ+ ,  JJ+ ,sA)C,B0'C(B??Cxpc0|jjsJ|j}|jrQ|j r|j n0|j r|j|jdz n |j|y)Nold_code) r1git_save code_folderrf is_symlinkunlinkis_dirrZr9 symlink_to)rqrWcodes r assign_cloner{sg 77    >>D {{} ??  KKM [[] KK j0 1<OOErAnyPathrHct|}t|}dtjvr: ddl}t|j j t|}||Sttjd}|jr||S|||z S#t$r3|js ttj|z }YwxYw)anWhen using `git_save`, this takes a potentially relative path with respect to the original execution folder and return an absolute path. This is required if you use relative path with respect to this original folder. When using both `git_save` and Hydra, two change of directory happens: - Dora moves to git clone - Hydra moves to XP folder Hydra provides a `to_absolute_path()` function. In order to simplify your code, if `git_save` was not used, and Hydra is in use, this will fallback to calling Hydra version, so that you only need to ever call this function to cover all cases. rkrN) typerrlrm hydra.utilsutilsto_absolute_pathrY ImportError is_absolutegetcwd)rHklass_pathhydra original_cwds rrrs JE JE2::- C  55c%jABEU|BJJ';<=    < -. . 2$$&RYY[)E1 2sB""9CC)$ contextlibrimportlib.utilr4loggingrlr subprocessrtypingtppathlibrr)rrQrrqr getLoggerr logger Exceptionr r'rKrMrSr^rirpr{TypeVarrYr|rrrrrs&    8 $ 9 ()-4)-})-XP!%S FF $    $ -t--" R   "**YT * /7 /w /r