3 _Af@sddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z dZ ddZ ddZd d Zd d Zdd dZdddZdS))print_functionN)ManifestParseError cCs tjjS)N)wrapperWrapperget_gitc_manifest_dirrr4sz*_set_project_revisions..z-FATAL: Failed to retrieve revisionExpr for %s rz&Invalid SHA-1 revision project %s (%s)N) Waitprintsysexitstdoutsplitrrrr)projectsZproject_gitcmdsprojZgitcmdrrrr _set_project_revisions)s   r!cCs*|j}|jjd}|s&dtjj}|S)zReturns the manifest group string that should be synced This is the same logic used by Command.GetProjects(), which is used during repo sync @param manifest: The XmlManifest object zmanifest.groupszdefault,platform-)manifestProjectconfig GetStringplatformsystemlower)manifestmpgroupsrrr _manifest_groupsFs  r+c std|dkrtjj}ddtjdtDfdd|D}fdd|D}|dk rxjjD]x\}}|jsqp|j rt j |j  r|j |_ ||jkr|j |qp||krp|j|}|jrd|_ |j|_qp|j |_ qpWd}x.|t|krt|||t|t7}qW|dk rnxB|jjD]4\}}|jr6||kr6j|}|j |_d|_ q6Wx jjD]\} } | j| _qzWtdS) aGenerate a manifest for shafsd to use for this GITC client. @param gitc_manifest: Current gitc manifest, or None if there isn't one yet. @param manifest: A GitcManifest object loaded with the current repo manifest. @param paths: List of project paths we want to update. zDGenerating GITC Manifest by fetching revision SHAs for each project.NcSsg|] }|r|qSrr)rxrrr rbsz*generate_gitc_manifest..z[,\s]+csg|]}j|qSr)paths)rp)r(rr rescsg|]}|jr|qSr) MatchesGroups)rr.)r*rr rfsr)rlistr-keysrerr+itemsr/Zupstreamr rrappend old_revisionlenr!NUM_BATCH_RETRIEVE_REVISIONIDZremotesZresolvedFetchUrlfetchUrl save_manifest) gitc_manifestr(r-rpathr Z gitc_projindexZ repo_proj_namerr)r*r(r generate_gitc_manifestUsD          r>c CsJ|s |j}ttjj|dd}|j|t|dWdQRXtjddS)zSave the manifest file in the client_dir. @param client_dir: Client directory to save the manifest in. @param manifest: Manifest object to save. z .manifestw)r*N) gitc_client_diropenosr;joinSaver+timesleep)r(Z client_dirfrrr r9s r9)N)N) __future__rrCr%r2rrFrr rerrorrr7rr r!r+r>r9rrrr s   @