o )j,@sddlZddlZddlmZmZddlmZmZddlm Z ddl m Z ddl m Z e ZGd d d eZGd d d ed ZdS)N)ListOptional)urlparse urlunparse) get_logger)MASTER_MODEL_BRANCH)GitErrorcs eZdZiZfddZZS) Singletoncs0||jvrtt|j|i||j|<|j|SN) _instancessuperr __call__)clsargskwargs __class__S/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/hub/git.pyrs  zSingleton.__call__)__name__ __module__ __qualname__r r __classcell__rrrrr sr c @seZdZdZdZdAdefddZdejfdd Z d d Z d ed efddZ d efddZ ddZ ddZ dAded ed ededeef ddZddZedfdedeed efd!d"Zded#efd$d%Zded&efd'd(Zded&efd)d*Zdefd+d,Z - .dBded/edefd0d1Z dCded ed ed2ed3ed4ef d5d6Zdefd7d8Zdefd9d:Zefded;ed#edZdefd?d@Z dS)DGitCommandWrapperzSome git operation wrapper gitNpathcCs|p|j|_dSr )default_git_pathgit_path)selfrrrr__init__szGitCommandWrapper.__init__returnc Gstd|tj}d|d<|jg|}dd|D}tj|tj tj |d}z| |WStj yl}z/|j j ddd }|jj ddd }d |vrZtd |WYd }~Std |||ft|d }~ww)a^Run git command, if command return 0, return subprocess.response otherwise raise GitError, message is stdout and stderr. Args: args: List of command args. Raises: GitError: Exception with stdout and stderr. Returns: subprocess.CompletedProcess: the command response  0GIT_TERMINAL_PROMPTcSsg|]}|r|qSrr).0itemrrr 2sz6GitCommandWrapper._run_git_command..)stdoutstderrenvzutf-8replace)errorsznothing to commitz;Nothing to commit, your local repo is upto date with remoteNz8Running git command: %s failed stdout: %s stderr: %s)loggerdebugjoinosenvironcopyr subprocessrunPIPEcheck_returncodeCalledProcessErrorr)decoder*infoerrorr )r rZgit_envcommandresponser;Zstd_outZstd_errrrr_run_git_command!s<  z"GitCommandWrapper._run_git_commandcCsX||}d|vr*|||}d||f}|d}|j|}t|jddSdS)N//oauth2z-C %s remote set-url origin %sr#utf8)get_repo_remote_url _add_tokensplitr>r.r/r)r9)r repo_dirZ auth_tokenurlZauth_urlcmd_argsrsprrrconfig_auth_tokenJs     z#GitCommandWrapper.config_auth_tokentokenrEcCs|s|S|dr |Szt|}Wn ty|YSw|jdvr$|S|jdkr+|S|jp/d}|jr;|d|j}d|d|}t|j|dS) aInject OAuth2 token into an HTTP(S) git URL. Uses ``urllib.parse`` for reliable URL component handling, avoiding naive string replacement that can corrupt URLs containing multiple ``://`` sequences. Args: token: OAuth2 access token. url: Remote URL (HTTP, HTTPS, or SSH). Returns: URL with ``oauth2:@`` injected into the *netloc*, or the original *url* unchanged when: * *token* is falsy, * the URL already carries an ``oauth2`` credential, * the scheme is not HTTP/HTTPS (e.g. ``ssh://``, ``git@``). zgit@)httphttpsoauth2:zoauth2:@)netloc) startswithr Exceptionschemeusernamehostnameportr_replace)r rIrEparsedhostrPrrrrBSs$      zGitCommandWrapper._add_tokencCs@|rd|vr|d}|d}|d|||dd}|S)Nr?rLrOr )find)r rE start_indexZ end_indexrrrremove_token_from_urls   z'GitCommandWrapper.remove_token_from_urlcCs.ddg}z|j|WdStyYdSw)Nlfsr+TFr>r )r cmdrrris_lfs_installeds  z"GitCommandWrapper.is_lfs_installedcCs4d|dddg}z|j|WdStyYdSw)N-Cr]installz--forceTFr^)r rDr_rrrgit_lfs_installs  z!GitCommandWrapper.git_lfs_install repo_base_dir repo_namebranchc Cs|||}|rd||||f}nd||f}t||d}z|j|}t|jd|WStyUtj ||}tj tj |drTt d|YdSw)a git clone command wrapper. For public project, token can None, private repo, there must token. Args: repo_base_dir (str): The local base dir, the repository will be clone to local_dir/repo_name token (str): The git token, must be provided for private project. url (str): The remote url repo_name (str): The local repository path name. branch (str, optional): _description_. Defaults to None. Returns: The popen response. z-C %s clone %s %s --branch %sz-C %s clone %sr#r@z.gitzgit clone exited with non-zero status but the repository was cloned successfully at %s. This is usually caused by a post-clone hook (e.g. core.hooksPath). Continuing.N) rBr.r/rCr>r)r9r r1rr0isdirwarning) r rdrIrErerfZ clone_argsr=rDrrrclones,      zGitCommandWrapper.clonec Csddlm}|\}}|rB|rDd|||f}|j|d}t|jdd|||f}|j|d}t|jddSdSdS)Nr)ModelScopeConfigz-C %s/%s config user.name %sr#r@z-C %s/%s config user.email %s) Zmodelscope.hub.apirjZ get_user_infor>rCr.r/r)r9) r rdrerjZ user_nameZ user_emailZconfig_user_name_argsr=Zconfig_user_email_argsrrr add_user_infos  zGitCommandWrapper.add_user_infoFrDfiles all_filescCsZ|rd|}nt|dkrd|}d||f}|d}|j|}t|jd|S)Nz -C %s add -Arr#z -C %s add %sr@)lenr0rCr>r.r/r)r9)r rDrlrmZadd_argsZ files_strrGrrradds      zGitCommandWrapper.addmessagecCs6dd|ddd|g}|j|}t|jd|S)zRun git commit command Args: repo_dir (str): the repository directory. message (str): commit message. Returns: The command popen response. ra%scommit-mz'%s'r@)r>r.r:r)r9)r rDrpZ commit_argsrGrrrrrs zGitCommandWrapper.commitrevisioncCsdd|dd|g}|j|S)Nrarqcheckoutr>r rDrtcmdsrrrrus zGitCommandWrapper.checkoutcCsdd|dd|g}|j|S)Nrarqruz-brvrwrrr new_branchs zGitCommandWrapper.new_branchcCs|dd|ddg}|j|}dd|jdtjD}t|dkr3d |d d ddgSd d|ddDS) Nrarqrfz-rcSsg|]}|qSr)stripr&linerrrr(sz9GitCommandWrapper.get_remote_branches..r@r /rcSs$g|]}d|dddqS)r}r N)r0rCr{rrrr(s$) r>r)r9rzrCr1lineseprnr0)r rDrxrGr:rrrget_remote_branchess  z%GitCommandWrapper.get_remote_branchesoriginmasterremotecCsd|d||g}|j|S)Nrapullrv)r rDrrfrxrrrrs zGitCommandWrapper.pull local_branch remote_branchforcec CsR|||}d||||f}|r|d7}|d}|j|}t|jd|S)Nz-C %s push %s %s:%sz -fr#r@)rBrCr>r.r/r)r9) r rDrIrErrrZ push_argsrGrrrpushs   zGitCommandWrapper.pushcCs0d|}|d}|j|}|jd}|S)Nz$-C %s config --get remote.origin.urlr#r@)rCr>r)r9rz)r rDrFrGrErrrrA s    z%GitCommandWrapper.get_repo_remote_urlcCsZd|}|d}|j|}|jd}g}|tjD] }||ddq|S)Nz-C %s lfs ls-filesr#r@)rCr>r)r9rzr1r~append)r rDrFrGoutrlr|rrrlist_lfs_files's  z GitCommandWrapper.list_lfs_filestag_namerefcCs6d|d|dd||g}|j|}t|jd|S)Nratagrsz"%s"r@r>r.r/r)r9)r rDrrprrFrGrrrr2s  zGitCommandWrapper.tagcCs.d|dd|g}|j|}t|jd|S)Nrarrr@r)r rDrrFrGrrrpush_tag?s zGitCommandWrapper.push_tagr )rr)F)!rrr__doc__rstrr!r4CompletedProcessr>rHrBr\r`rcrrirklistrboolrorrruryrrrrArrrrrrrrrs) /  .      r) metaclass)r1r4typingrr urllib.parserrZmodelscope.utils.loggerrZutils.constantrr-r r.typer rrrrrs