o b²“iÚ*ã@sddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z dZ e  ¡Z dZe jdkr>> _dirmatch('/home/foo/bar', '/home/foo/bar') True >>> _dirmatch('/home/foo/bar/', '/home/foo/bar') True >>> _dirmatch('/home/foo/bar/etc', '/home/foo/bar') True >>> _dirmatch('/home/foo/bar2', '/home/foo/bar') False >>> _dirmatch('/home/foo/bar2/etc', '/home/foo/bar') False éÚTF)ÚlenÚ startswithÚosÚsep)ÚpathÚ matchwithÚmatchlenr r r Ú _dirmatch!s $rcCs‚tj |td¡}trtj ¡}ni}tj|ddg|tj d}|  ¡\}}|j s)|s+J‚|  d¡  ¡}|dttt|dd…ƒƒfS) z/obtain version and path info from a virtualenv.Úpythonz-czjimport sys;print ("%d.%d" % (sys.version_info.major, sys.version_info.minor));print ("\n".join(sys.path));)ÚenvÚstdoutúutf-8rr N)rrÚjoinÚ env_bin_dirÚ_WIN32ÚenvironÚcopyÚ subprocessÚPopenÚPIPEÚ communicateÚ returncodeÚdecodeÚ splitlinesÚlistÚfilterÚbool)Ú venv_pathÚ executablerÚprÚerrÚlinesr r r Ú_virtualenv_sys4s  ý÷ r.csÔtj ˆ¡s tdˆƒ‚tj |¡rtd|ƒ‚t dˆ|f¡tjˆ|dt d¡dt |ƒ\}}t d¡t ˆ||ƒ‡fdd „}||ƒrSt d ¡t |ˆ|ƒt |ƒ}||d ƒ}|rcJ|ƒ‚t ˆ|ƒdS) Nzsrc dir %r does not existzdest dir %r existsz"cloning virtualenv '%s' => '%s'...Tz*.pyc)ÚsymlinksÚignorezfixing scripts in bin...cst‡fdd„|DƒƒS)Nc3s|] }t|ˆƒr|VqdS©N)r)Ú.0Úi©Úsrc_dirr r Ú \s€z5clone_virtualenv....)Úany)Úsr4r r Ú\sz"clone_virtualenv..zfixing paths in sys.path...r ) rrÚexistsrÚloggerÚinfoÚshutilÚcopytreeÚignore_patternsr.Ú fixup_scriptsÚfixup_syspath_itemsÚfix_symlink_if_necessary)r5Údst_dirÚversionÚsys_pathÚhas_oldÚv_sysÚ remainingr r4r Úclone_virtualenvNs&    ÿ        rIc Cs˜t d¡t |¡D]?\}}}t ||¡D]3}tj ||¡}tj |¡rHtj  |¡}|  |¡rH|  ||¡}t  d|f¡t  |¡t ||¡qq dS)NzEscanning for internal symlinks that point to the original virtual envzfixing symlink in %s)r;r<rÚwalkÚ itertoolsÚchainrrÚislinkÚrealpathrÚreplaceÚdebugÚremoveÚsymlink) r5rCÚdirpathÚdirnamesÚ filenamesÚa_fileÚfull_file_pathÚtargetÚ new_targetr r r rBis       €øÿrBc Csâtj |t¡}tt |¡ƒ\}}}t d¡}|D]V} tj || ¡} | dd|dfvr+q|  d¡r6|  | ¡r6q|   d¡r.rewrite_shebangrz#!r1)rrÚnormcaseÚabspathrrtÚreadÚseekÚ readlinesr$ÚstripÚUnicodeDecodeErrorrr) rhrlrerfrDr[Ú old_shebangÚ env_shebangrxr{ÚbangÚ short_versionr ryr rdšsN ý  û  þ   ÿþ   ÿrdcCs’t d|¡t|dƒ}| ¡ d¡}Wdƒn1swY| ||¡}t|dƒ}| | d¡¡WdƒdS1sBwYdS)Nrprnrrr)r;rPrtr~r$rOrurs)rmrerfrxÚdatar r r raÖs ÿ  "ÿracCs¬t d|¡|durt |¡}tj tj |¡¡ ||¡}tj |¡s2tj tj  ||¡¡}d}nd}t ||ƒrO|rH|t |ƒd…  tj ¡}n| ||d¡}t||ƒdS)NrpTFr )r;rPrÚreadlinkrÚdirnamer}rOÚisabsrrrÚlstriprÚ_replace_symlink)rmrerfrXÚorigdirÚrellinkr r r rbàs   rbcCs$d|}t ||¡t ||¡dS)Nz%s.new)rrRÚrename)rmÚ newtargetÚtmpfnr r r rŒùs rŒc Cs¾|D]Z}tj |¡s qtj tj |¡¡}t||ƒr)| ||d¡}tj |¡s(qnt||ƒs/qtt  |¡ƒ\}}}|D] }tj  ||¡}|  d¡rPt |||ƒq;|  d¡r[t |||ƒq;qdS)Nr z.pthz .egg-link)rrÚisdirr|r}rrOr:r\rJrr`Úfixup_pth_fileÚfixup_egglink_file) Úsyspathrerfrrhrirjrlrmr r r rAÿs*   ÿ    €ûõrAc Cs t d|¡t|dƒ }| ¡}Wdƒn1swYd}t|ƒD]/\}}t|dƒr4| d¡n| ¡}|rD| d¡sD| d¡rEq&t ||ƒrU|  ||d¡||<d }q&|rƒt|d ƒ}t j   d d „|Dƒ¡t j }| |¡WdƒdS1s|wYdSdS) Nzfixup_pth_file %sÚrFr$rú#zimport r TÚwcSsg|]}| ¡‘qSr )r)r2Úliner r r Ú &sz"fixup_pth_file..)r;rPrtr€Ú enumerateÚhasattrr$rrrrOrÚlineseprru) rmrerfrxr-Ú has_changeÚnumr™Úpayloadr r r r“s&  ÿ €  "þÿr“cCs®t d|¡t|dƒ}| ¡ d¡ ¡}Wdƒn1s wYt||ƒrU| ||d¡}t|dƒ}|d d¡}|  |¡WdƒdS1sNwYdSdS)Nrprnrr rrrq) r;rPrtr~r$rrrOrsru)rmrerfrxÚlinkr r r r”*s ÿ   "þþr”cCsät d¡}|jdddddd| ¡\}}z|\}}Wnty0tdtfƒ| d ¡Ynwtj   |¡}tj   |¡}t j t j t jftd |jƒ}t j|d d zt||ƒWdStyqt ¡d }| t|ƒ¡YdSw)NzBusage: %prog [options] /path/to/existing/venv /path/to/cloned/venvz-vÚcountÚverboseFÚ verbosity)ÚactionÚdestÚdefaultÚhelpzvirtualenv-clone %sznot enough arguments given.roz %(message)s)ÚlevelÚformatr )ÚoptparseÚ OptionParserÚ add_optionÚ parse_argsÚ ValueErrorÚprintÚ __version__ÚerrorrrrNÚloggingÚWARNINGÚINFOÚDEBUGÚminr£Ú basicConfigrIrÚsysÚexc_infoÚstr)ÚparserÚoptionsÚargsrerfÚloglevelÚer r r Úmain5s.ÿ ÿ   þ    þrÁÚ__main__)Fr1)!Ú __future__rr³r«rÚos.pathr]r=rr¹rKr±Ú getLoggerr;rÚplatformrÚ Exceptionrrr.rIrBr@rdrarbrŒrAr“r”rÁrr r r r ÚsD    <   ÿ