a ¬…aµ ã@s¸UdZddlmZddlmZddlmZmZmZm Z m Z ddl m Z m Z mZddlmZddlmZmZd Ze ed fed <Gd d „d eƒZeddfe eeeeeedœdd„ZdS)zVFast stream based import identification. Eventually this will likely replace parse.py é)Úpartial)ÚPath)ÚIteratorÚ NamedTupleÚOptionalÚTextIOÚTuple)Ú_normalize_lineÚ _strip_syntaxÚ skip_lineé)Úparse)ÚDEFAULT_CONFIGÚConfig)zdef zcdef zcpdef zclass ú@z async def.ÚSTATEMENT_DECLARATIONSc@s~eZdZUeed<eed<eed<dZeeed<dZ eeed<dZ eed<dZ ee ed <ed œd d „Z ed œd d„ZdS)ÚImportÚ line_numberÚindentedÚmoduleNÚ attributeÚaliasFÚcimportÚ file_path)ÚreturncCsZ|jr dnd}|jr0d|j›d|›d|j›}n|›d|j›}|jrV|d|j›7}|S)NrÚimportúfrom ú z as )rrrr)ÚselfZ import_cmdÚ import_string©r ú-lib/python3.9/site-packages/isort/identify.pyÚ statementszImport.statementcCs.|jpd›d|j›d|jrdnd›| ¡›S)NÚú:rz indented )rrrr")rr r r!Ú__str__#s  ÿÿÿzImport.__str__)Ú__name__Ú __module__Ú __qualname__ÚintÚ__annotations__ÚboolÚstrrrrrrrr"r%r r r r!rs   rNF)Ú input_streamÚconfigrÚtop_onlyrc csÀd}t|ƒ}|D]¨\}}t||||jd\}}|rH|sH| t¡rHq¼|rNq| ¡ d¡d} |  d¡sv|  d¡r| dkrÆ| rŠ| dkrÆzt|ƒ\}} Wnty°YqÆYn0|  ¡ d¡d} q~|   d¡rzt|ƒ\}} WntyöYqYn0|  ¡ d¡d} qÆq| dd¡^} } d d „|  d ¡Dƒ} | rR| d ›d| d›| d <| D]`}t |ƒ\} }|  d ¡rzd}n|  d¡rVd}nqVt | ƒ\}}|  dd¡  dd¡  dd¡}d|vpÊ| d¡}t t|d| d¡||d}d|  dd¡dvrd|  d¡d ¡  d¡szt|ƒ\}} WntyBYqYn0t | ƒ\} }|d| 7}qþn,|  ¡  d¡rzt|ƒ\}} WntyžYqYn0t | ƒ\} }d|  d¡dvrDd|  d¡dvrD|d| 7}|  d¡d ¡  d¡sŒzt|ƒ\}} Wnty$YqŒYn0t | ƒ\} }|d| 7}qànH| ¡  d¡sd|  ¡ d ¡rr|d| 7}n| ¡ d¡d|  ¡}qd|dkr|  dd¡  dd¡  dd¡}| |rÄdnd¡}|d d¡}|râdnd |ddd |dd…¡g|dd…¡}dd „t|ƒ ¡Dƒ}|dd…}d}d|vrl| d¡dt|ƒkrld|vrld}| d¡}|dkr||d}|d}|d |}||d}| |¡| |¡| d¡||dd…<||krô|jrô|||ƒVn||||d!Vnb||d}||d}| |¡| d¡| |¡||krZ|jrZ||ƒVn|||d!Vq\|rV|dkr | d¡}|D]}|||ƒVqŠn|D]}||ƒVq¤qVqdS)"z9Parses a python file taking out and categorizing imports.r#)Úin_quoteÚindexÚsection_commentsú#rÚraiseÚyieldú\r cSsg|] }| ¡‘qSr )Ústrip)Ú.0Úliner r r!Ú Rózimports..ú;éÿÿÿÿ)zimport zcimport ZstraightrÚfromzimport(zimport (rÚ z cimport r)rú )rrú(ú))z importz cimportz import NcSs g|]}| dd¡ dd¡‘qS)z{|z{ z|}z })Úreplace)r8Úitemr r r!r:£sÿÚasÚ.)r)Ú enumerater r2Ú startswithrr7ÚsplitÚnextÚ StopIterationÚendswithr Úparse_commentsrCrrÚrstripÚlstripÚjoinr r1ÚlenÚremoveZremove_redundant_aliasesÚpop)r-r.rr/r0Z indexed_inputr1Zraw_lineZ skipping_lineZ stripped_lineZ next_liner9Zend_of_line_commentZ statementsr"Z _raw_lineZtype_of_importrÚ_Znormalized_import_stringZcimportsZidentified_importÚpartsZ from_importZ just_importsZdirect_importsZtop_level_modulerZas_indexrrr r r!Úimports*s ÿ            ÿ þû    (   ÿ þÿ  ÿþÿ*ÿ þ "                  rV)Ú__doc__Ú functoolsrÚpathlibrÚtypingrrrrrZ isort.parser r r Zcommentsr rMZsettingsrrrr,r*rr+rVr r r r!Ús$   üû