o 8)it:@sFUddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZd d lmZd dlmZd dlmZd dlmZd dlmZd dlm Z!d dl"m#Z#d dl$m%Z%d dl$m&Z&d dl'm(Z(d dl'm)Z)d dl'm*Z*erd dlm+Z+e,e-Z.e/dZ0dZ1gdZ2Gddde Z3ddd d\d+d,Z4Gd-d.d.e%Z5d]d1d2Z6d]d3d4Z7d]d5d6Z8d]d7d8Z9e3j:e6e3j;e8e3jd9e?d:<d^d@dAZ@  d_d`dGdHZAdadKdLZB  d_dbdOdPZCdcdTdUZDdddXdYZEdedZd[ZFdS)f) annotationsN)date)datetime)timezone)Enum)samefile)Path) TYPE_CHECKING)Callable)Sequence) Configuration)_types)discover)CompletedProcess)require_command)run)data_from_mime)Workdir)get_latest_file_mtime) ScmVersion)meta)tag_to_version)hg_gitz(?<=\btag: )([^,]+)\bz %(describe)gitZdescribez--dirtyz--tagsz--longz --abbrev=40z--matchz*[0-9]*c@s eZdZdZdZdZdZdZdS) GitPreParsez!Available git pre-parse functionswarn_on_shallowfail_on_shallowfetch_on_shallowfail_on_missing_submodulesN)__name__ __module__ __qualname____doc__WARN_ON_SHALLOWFAIL_ON_SHALLOWFETCH_ON_SHALLOWFAIL_ON_MISSING_SUBMODULESr(r(M/home/app/PyTorch/.pytorch/lib/python3.10/site-packages/setuptools_scm/git.pyr8s rFchecktimeoutargs Sequence[str | os.PathLike[str]]reporr+boolr, int | Nonereturn_CompletedProcesscCstdd|dg||||dS)Nrz --git-dirz.git)cwdr+r,)_run)r-r/r+r,r(r(r)run_gitAs r6c@sxeZdZdZed!ddZd"d d Zd#d d Zd$ddZd$ddZ d"ddZ d%ddZ d#ddZ d&ddZ d'ddZd S)( GitWorkdir$experimental, may change at any timewd_t.PathTr2GitWorkdir | NonecCst|}tddg|jtd}|durdS|dd}|s%t|}nt|}ddlm}|||}t d|t ||sBdS|t|S)N rev-parsez --show-prefixparser )strip_path_suffixz real root %s) rresolver6 parse_successstrosfspathZ_compatr@logdebugr)clsr9Zreal_wdZstr_wdr@r(r(r)from_potential_worktreeSs         z"GitWorkdir.from_potential_worktreer0cCstgd|jjtddS)N)statusz --porcelainz--untracked-files=noF)r>default)r6pathrBr0selfr(r(r)is_dirtyis zGitWorkdir.is_dirty str | NonecCs4tgd|jjtddptgd|jjtddS)N)r<z --abbrev-refHEADzbranch err (abbrev-err)r> error_msg)z symbolic-refz--shortrQzbranch err (symbolic-ref)r6rLrBrCrMr(r(r) get_branchqs"zGitWorkdir.get_branch date | NonecCs0d dd}tgddd |j}|j|d d S) Ntimestamp_textrCr2rVcSsrd|vr td|dStjdkr|dr|ddd}t|}td||t j  }td||S) Nz%czgit too old -> timestamp is %r) Zr?z+00:00zdt: %sz dt utc: %s) rFwarningsys version_infoendswithr fromisoformatrG astimezonerutcr)rWdtZdt_utcr(r(r)parse_timestamps    z1GitWorkdir.get_head_date..parse_timestamp)z-czlog.showSignature=false)rFz-n1rQz --format=%cIz$logging the iso date for head failedrR)rWrCr2rV)r6rLrB)rNrcresr(r(r) get_head_dates zGitWorkdir.get_head_datec Cs~|sdSztddg|j}|jdkrWdS|jd}t||jWSty>}z t d|WYd}~dSd}~ww)zGet the latest modification time of changed files in the working directory. Returns the date of the most recently modified file that has changes, or None if no files are changed or if an error occurs. Ndiffz --name-onlyr z Failed to get dirty tag date: %s) rOr6rL returncodestdoutstripsplitr ExceptionrFrG)rNZchanged_files_resZ changed_fileser(r(r)get_dirty_tag_dates  zGitWorkdir.get_dirty_tag_datecCs|jdS)Nz .git/shallow)rLjoinpathis_filerMr(r(r) is_shallowszGitWorkdir.is_shallowNonecCstddg|jddddS)Nfetchz --unshallowTr*)r6rLrMr(r(r) fetch_shallowszGitWorkdir.fetch_shallowcCstgd|jjtdS)N)r<z--verifyz--quietrQr=rTrMr(r(r)nodes  zGitWorkdir.nodeintcCs tddg|j}|jddS)Nzrev-listrQrhr )r6rLrjcount)rNrer(r(r)count_all_nodesszGitWorkdir.count_all_nodesr3cCsttdd|jS)Nr )r6DEFAULT_DESCRIBErLrMr(r(r)default_describeszGitWorkdir.default_describeN)r9r:r2r;)r2r0)r2rP)r2rV)r2rs)r2rx)r2r3)r r!r"r# classmethodrIrOrUrfrorrrvrwrzr|r(r(r(r)r7Ps         r7r9rscCs$|rtd|jddSdS)r8"z!" is shallow and may cause errorsN)rrwarningswarnrLr9r(r(r)rsrcCs,|rtd|jd|dSdS)r8r~z," was shallow, git fetch was used to rectifyN)rrrrrLrvrr(r(r)rs rcCs|r t|jddS)r8z8 is shallow, please correct with "git fetch --unshallow"N)rr ValueErrorrLrr(r(r)rs  rc Cs|jd}|s dStddg|j}|jdkr!td|jdS|jr.|j dng}g}|D]}|}| drR| }t |d krR|d }| |q4|sb|rbt d |jd |rud |}t d|jd|d dS)a* Fail if submodules are defined but not initialized/cloned. This pre_parse function checks if there are submodules defined in .gitmodules but not properly initialized (cloned). This helps prevent packaging incomplete projects when submodules are required for a complete build. z .gitmodulesN submodulerJrz$Failed to check submodule status: %srh-r z=Submodules are defined in .gitmodules but not initialized in zJ. Please run 'git submodule update --init --recursive' to initialize them.z, z"Submodules are not initialized in z: )rLexistsr6rirFrGstderrrjrkrl startswithlenappendrjoin) r9Zgitmodules_pathZ status_resultZ status_linesZuninitialized_submoduleslinepartsZsubmodule_pathZsubmodule_listr(r(r)rs8        rz/dict[GitPreParse, Callable[[GitWorkdir], None]]_GIT_PRE_PARSE_FUNCTIONSconfigr rootr:r;cCsL|jr t|jStj||jdD]}t|}|dur |Sqt|S)z8 Return the working directory (``GitWorkdir``). )Zsearch_parentsN)parentr7rIrZwalk_potential_rootsZsearch_parent_directories)rrZpotential_rootZ potential_wdr(r(r)get_working_directorys   rdescribe_commandstr | list[str] | None pre_parse#Callable[[GitWorkdir], None] | NoneScmVersion | NonecCsJtdt||}|r#|dur|}n t|jjjt}t||||dSdS)z :param pre_parse: experimental pre_parse action, may change at any time. Takes precedence over config.git_pre_parse if provided. rN)rr) _require_commandrrgetscmrrr_git_parse_inner)rrrrr9Zeffective_pre_parser(r(r)r>)s  r>&GitWorkdir | hg_git.GitWorkdirHgClient_t.CMD_TYPE | Nonecsjjjdur jjj}|dur2t|trt|}|ddkr+t|dd|j}n t ||j}n| }d fdd }|j |d S) Nrrr outputrCr2rcs"t|\}}}}t||||dS)Ntagdistancedirtyrwr)_git_parse_describer)rrrrwrrr(r)parse_describeYsz-version_from_describe..parse_describer=)rrCr2r) rrr isinstancerCshlexrlr6rLr5r|rB)r9rrZ describe_resrr(rr)version_from_describeFs     r?Callable[[GitWorkdir | hg_git.GitWorkdirHgClient], None] | Nonerc Cs|r||t|||}|dur:||jpd}|}|dur%d}d}n |}d|}|}t|||||d}|} |} | durT|rT| } | durT| } | dur`t t j } tj|| | dS)N0.0rTgr)branch node_date)rZ version_clsZfallback_versionrwrzrOrrUrfrornowrrar dataclassesreplace) rr9rrversionrrwrrrrZ dirty_dater(r(r)r`s0  rdescribe_outputrC!tuple[str, int, str | None, bool]cCsd|drd}|dd}nd}|dd}t|dkr#|}d}d}n |\}}}t|}||||fS) Nz-dirtyTiFrrrXr)r^rsplitrrx)rrrlrnumberrwZnumber_r(r(r)rs     rdatadict[str, str]c Cstd||dt}t|vrtdnt|\}}}}t||||dSt |ddD]}t ||}|durBt||dSq/|d}|durNdSd | vr[td dStd ||d S) Nzdata %sz describe-namez+git archive did not support describe output)rrrwz ref-namesrrwz$FORMATz8unprocessed git archival found (no export subst applied)r)rwr) rFrGrDESCRIBE_UNSUPPORTEDrrrr REF_TAG_REfindallrupper) rrZarchival_describerrrw_refrr(r(r)archival_to_versions.       rcCs"tj|d}t|}t||dS)Nz.git_archival.txtr)rDrLrrr)rrZarchivalrr(r(r)parse_archivals r) r-r.r/rr+r0r,r1r2r3)r9r7r2rs)rr rr:r2r;)NN) rr:rr rrrrr2r)r9rrr rrr2r) rr r9rrrrrr2r)rrCr2r)rrrr r2r)rr:rr r2r)G __future__rrloggingrDrerr\rrrrenumrZos.pathrpathlibrtypingr r r rr rZ_trZ_run_cmdrr3rrrr5Z integrationrZ scm_workdirrrrrrrr getLoggerr rFcompilerrr{rr6r7rrrrr$r%r&r'r__annotations__rr>rrrrrr(r(r(r)sx                           w   4   +