uj,ddlZddlZddlmZmZddlmZmZddlm Z ddl m Z ddl m Z e ZGd d eZGd d e ZdS)N)ListOptional)urlparse urlunparse) get_logger)MASTER_MODEL_BRANCH)GitErrorc"eZdZiZfdZxZS) Singletonc||jvr(tt|j|i||j|<|j|SN) _instancessuperr __call__)clsargskwargs __class__s b/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/modelscope/hub/git.pyrzSingleton.__call__sV cn $ $#6% (+#-#--5t#G?E#G#GCN3 ~c"")__name__ __module__ __qualname__rr __classcell__)rs@rr r s=J#########rr c eZdZdZdZd-defdZdejfdZ dZ d ed efd Z d efd Z d Z dZ d-ded ed ededeef dZdZedfdedeedefdZdedefdZdedefdZdedefdZdefdZ d.ded!edefd"Z d/ded ed ed#ed$ed%ef d&Zdefd'Zdefd(Zefded)eded*efd+Zdefd,Z dS)0GitCommandWrapperzSome git operation wrapper gitNpathc"|p|j|_dSr)default_git_pathgit_path)selfr s r__init__zGitCommandWrapper.__init__s5 5 rreturnc td|tj}d|d<|jg|}d|D}tj|tj tj |} | |S#tj $r}|j dd}|j dd}d |vr!td |cYd }~Std |d |d|t#|d }~wwxYw)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_PROMPTcg|]}||Sr,).0items r z6GitCommandWrapper._run_git_command..2s444Dt44444r)stdoutstderrenvzutf-8replace)errorsznothing to commitz;Nothing to commit, your local repo is upto date with remoteNzRunning git command: z failed stdout: z stderr: )loggerdebugjoinosenvironcopyr# subprocessrunPIPEcheck_returncodeCalledProcessErrorr0decoder1infoerrorr )r$rgit_envcommandresponserBstd_outstd_errs r_run_git_commandz"GitCommandWrapper._run_git_command!si  SXXd^^$$$*//##),%&=(4(44G444> ??      (  % % ' ' 'O, ( ( (o,,WY,GGGl))'))DDG"g-- Q  ww2333w''' (s%B**E 9AEE 2EE c||}d|vrq|||}d|d|}|d}|j|}t|jddSdS)N//oauth2-C z remote set-url origin r(utf8)get_repo_remote_url _add_tokensplitrHr5r6r0r@)r$repo_dir auth_tokenurlauth_urlcmd_argsrsps rconfig_auth_tokenz#GitCommandWrapper.config_auth_tokenJs&&x00 S z377HH;C88XXNH~~c**H'$'2C LL**622 3 3 3 3 3 ! rtokenrRc<|s|S|dr|S t|}n#t$r|cYSwxYw|jdvr|S|jdkr|S|jpd}|jr |d|j}d|d|}t||S) 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$rWrRparsedhostr_s rrNzGitCommandWrapper._add_tokenSs& J >>& ! ! J c]]FF   JJJ  = 1 1 1J ?h & &J$" ; +**V[**D)5))4))&///88999s - <<c|rFd|vrB|d}|d}|d|||dzdz}|S)NrJr[r^r )find)r$rR start_index end_indexs rremove_token_from_urlz'GitCommandWrapper.remove_token_from_urlsY  ::$$((8,,K Il{l#c)a-..&99C rcFddg} |j|dS#t$rYdSwxYw)Nlfsr2TFrHr )r$cmds ris_lfs_installedz"GitCommandWrapper.is_lfs_installedsGen  !D !3 ' '4   55 s   cLd|dddg} |j|dS#t$rYdSwxYw)N-Croinstallz--forceTFrp)r$rPrqs rgit_lfs_installz!GitCommandWrapper.git_lfs_installsNXui;  !D !3 ' '4   55 s  ## repo_base_dir repo_namebranchcP|||}|rd|d|d|d|}nd|d|}t||d} |j|}t|jd|S#t$r}tj ||}tj tj |drt d|YdSwxYw) 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. rKz clone r(z --branch rLz.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) rNr5r6rOrHr0r@r r8r r7isdirwarning) r$rwrWrRrxry clone_argsrErPs rclonezGitCommandWrapper.clonesH&ooeS))  A A > @ @ :  9 9 9 yyy))%5 !,t,.C.I.I#.N.NOH LL//77 8 8 8 8 yyy**&6 ",t,'--c224H LL//77 8 8 8 8 8 9 9 9 9rFrPfiles all_filesc|rd|z}n0t|dkrd|}d|d|}|d}|j|}t|jd|S)Nz -C %s add -Arr(rKz add rL)lenr7rOrHr5r6r0r@)r$rPrradd_args files_strrUs raddzGitCommandWrapper.adds  >%0HH ZZ!^^II)199=H>>#&&#d#X. SZ&&v../// rmessagecdd|zddd|zg}|j|}t|jd|S)zRun git commit command Args: repo_dir (str): the repository directory. message (str): commit message. Returns: The command popen response. rt%scommit-mz'%s'rL)rHr5rAr0r@)r$rPr commit_argsrUs rrzGitCommandWrapper.commitsSTH_hfw>NO #d#[1 CJ%%f--... rrevisionc.dd|zdd|zg}|j|S)NrtrcheckoutrHr$rPrcmdss rrzGitCommandWrapper.checkouts(dXoz4(?C$t$d++rc*dd|zdd|g}|j|S)Nrtrrz-brrs r new_branchzGitCommandWrapper.new_branchs&dXoz4B$t$d++rcdd|zddg}|j|}d|jdt jD}t|dkr7d|d dddgSd |ddDS) Nrtrryz-rc6g|]}|Sr,)stripr-lines rr/z9GitCommandWrapper.get_remote_branches..s0    JJLL   rrLr rrcng|]2}d|ddd3S)rr N)r7rOrs rr/z9GitCommandWrapper.get_remote_branches..s7GGGdCHHTZZ__QRR011GGGr) rHr0r@rrOr8lineseprr7)r$rPrrUrAs rget_remote_branchesz%GitCommandWrapper.get_remote_branchessdXox6#d#T*   ))&117799?? KK    t99>>HHT!W]]3//3445 5GGd122hGGG Groriginmasterremotec$d|d||g}|j|S)Nrtpullr)r$rPrryrs rrzGitCommandWrapper.pulls$h7$t$d++r local_branch remote_branchforcec|||}d|d|d|d|}|r|dz }|d}|j|}t|jd|S)NrKz push r(r]z -frL)rNrOrHr5r6r0r@) r$rPrWrRrrr push_argsrUs rpushzGitCommandWrapper.pushsooeS))-5XXsssLLL-:]<    IOOC(( #d#Y/ SZ&&v../// rcd|z}|d}|j|}|jd}|S)Nz$-C %s config --get remote.origin.urlr(rL)rOrHr0r@r)r$rPrTrUrRs rrMz%GitCommandWrapper.get_repo_remote_url sO9HD>>#&&#d#X.j''yy{{rcJd|z}|d}|j|}|jd}g}|t jD]0}||dd1|S)Nz-C %s lfs ls-filesr(rL)rOrHr0r@rr8rappend)r$rPrTrUoutrrs rlist_lfs_filesz GitCommandWrapper.list_lfs_files's'(2>>#&&#d#X.j''--//IIbj)) . .D LLC, - - - - rtag_namerefcd|d|dd|z|g}|j|}t|jd|S)Nrttagrz"%s"rLrHr5r6r0r@)r$rPrrrrTrUs rrzGitCommandWrapper.tag2sX (E8T W c $d#X. SZ&&v../// rcd|dd|g}|j|}t|jd|S)NrtrrrLr)r$rPrrTrUs rpush_tagzGitCommandWrapper.push_tag?sH(FHh?#d#X. SZ&&v../// rr)rr)F)!rrr__doc__r"strr%r;CompletedProcessrHrVrNrmrrrvrr~rlistrboolrrrrrrrrMrr rrr,rrrrs66S6666'()D'('('('(R444-:-:#-:-:-:-:^'+ ,, ,,, , sm ,,,,\ 9 9 9" $tvv#   9      s S    ,,,,,,,3,#,,,, HC H H H H$#,,,,,,,,!      $C s    +         rr) metaclass)r8r;typingrr urllib.parserrmodelscope.utils.loggerrutils.constantr r4r r5typer rr,rrrs !!!!!!!!--------......000000 ########jjjjj)jjjjjjr