B ãxác¸ ã@sZdZddlmZddlmZmZmZmZddlm Z m Z m Z dd„Z Gdd „d ej ƒZd S) zÞFixer for import statements. If spam is being imported from the local directory, this import: from spam import eggs Becomes: from .spam import eggs And this import: import spam Becomes: from . import spam é)Ú fixer_baseé)ÚdirnameÚjoinÚexistsÚsep)Ú FromImportÚsymsÚtokenccs¢|g}x–|rœ| ¡}|jtjkr*|jVq|jtjkrPd dd„|jDƒ¡Vq|jtj krn|  |jd¡q|jtj kr’|  |jddd…¡qt dƒ‚qWdS)zF Walks over all the names imported in a dotted_as_names node. ÚcSsg|] }|j‘qS©)Úvalue)Ú.0Zchr r ú=/opt/alt/python37/lib64/python3.7/lib2to3/fixes/fix_import.pyú sz$traverse_imports..rNéþÿÿÿzunknown node type)ÚpopÚtyper ÚNAMEr r Z dotted_namerÚchildrenZdotted_as_nameÚappendZdotted_as_namesÚextendÚAssertionError)ÚnamesZpendingÚnoder r rÚtraverse_importss     rcs4eZdZdZdZ‡fdd„Zdd„Zdd„Z‡ZS) Ú FixImportTzj import_from< 'from' imp=any 'import' ['('] any [')'] > | import_name< 'import' imp=any > cs"tt|ƒ ||¡d|jk|_dS)NZabsolute_import)ÚsuperrÚ start_treeZfuture_featuresÚskip)ÚselfZtreeÚname)Ú __class__r rr/szFixImport.start_treecCs¼|jr dS|d}|jtjkrZxt|dƒs6|jd}q W| |j¡r¸d|j|_| ¡n^d}d}x$t |ƒD]}| |¡r€d}qld}qlW|r |rœ|  |d¡dSt d|gƒ}|j |_ |SdS)NÚimpr rÚ.FTz#absolute and local imports together) rrr Z import_fromÚhasattrrÚprobably_a_local_importr ZchangedrZwarningrÚprefix)r rZresultsr#Z have_localZ have_absoluteZmod_nameÚnewr r rÚ transform3s,        zFixImport.transformcCsv| d¡rdS| dd¡d}t|jƒ}t||ƒ}ttt|ƒdƒƒsHdSx(dtddd d gD]}t||ƒrZd SqZWdS) Nr$Férz __init__.pyz.pyz.pycz.soz.slz.pydT)Ú startswithÚsplitrÚfilenamerrr)r Zimp_nameÚ base_pathZextr r rr&Us    z!FixImport.probably_a_local_import) Ú__name__Ú __module__Ú __qualname__Z BM_compatibleZPATTERNrr)r&Ú __classcell__r r )r"rr&s  "rN)Ú__doc__r rZos.pathrrrrZ fixer_utilrr r rZBaseFixrr r r rÚ s