B 0zf#@sddlmZddlmZddlmZddlmZddlmZddl Z ddl Z dZ Gdd d e Z Gd d d e ZGd d d e Zee dreneZdS))redirect_stdout)Differ) unified_diff)StringIO)PathNz4.2.0c@sTeZdZdZeZdZdZdZdZ dZ dZ dZ d Z d d Zed d ZeddZdS)Flake8IsortBaseZ flake8_isortz0I001 isort found an import in the wrong positionz>I002 no configuration found (.isort.cfg or [isort] in configs)z4I003 isort expected 1 blank line in imports, found 0z4I004 isort found an unexpected blank line in importsz-I005 isort found an unexpected missing importFNTcCs||_||_dS)N)filenamelines)selftreerr r O/home/ankuromar296_gmail_com/.local/lib/python3.7/site-packages/flake8_isort.py__init__'szFlake8IsortBase.__init__cCs|jddddddS)Nz--isort-show-traceback store_trueTz(Show full traceback with diff from isort)actionZparse_from_confighelp) add_option)clsparserr r r add_options+s zFlake8IsortBase.add_optionscCs|j|_|j|_dS)N)stdin_display_nameZisort_show_tracebackshow_traceback)roptionsr r r parse_options4szFlake8IsortBase.parse_options)__name__ __module__ __qualname__name __version__versionisort_unsortedZ no_config_msgisort_blank_reqisort_blank_unexpisort_add_unexprrZsearch_currentr classmethodrrr r r r rs rc@s@eZdZdZddZddZddZedd Zed d Z d S) Flake8Isort4zclass for isort <5c cs|j|jk r|j}nd}t}t| tj|d|jddd}WdQRX||}x6| |D](\}}|j rx||7}|d|t |fVqbWdS)NT) file_path file_contentscheckZ show_diffr) rrrrisortZ SortImportsjoinr _format_isort_outputsortimports_linenum_msgrtype)r r'buffer sort_result tracebackline_nummessager r r run=s    zFlake8Isort4.runccs|jr dS||||t}||j|j}d}dd|jD}x|D]}|dddrj|d7}qN|dddr|d7}| d kr||j fVq||j fVqN| d kr|d|j fVqN| |krN|d|j fVqNWdS) aParses isort.SortImports for line number changes and message Uses a diff.Differ comparison of SortImport `in_lines`:`out_lines` to yield the line numbers of import lines that have been moved or blank lines added. Args: sort_imports (isort.SortImports): The isorts results object. Yields: tuple: A tuple of the specific isort line number and message. NrcSsh|]}d|qS)z+ {})format).0Z add_importr r r hsz7Flake8Isort4.sortimports_linenum_msg..z z- -+)Zskipped_fixup_sortimports_wrapped_fixup_sortimports_eofrcomparein_lines out_lines add_imports startswithstripr"r r!r#)r r0Zdifferdiffr2 additionsliner r r r-Qs*         z$Flake8Isort4.sortimports_linenum_msgcsTddg}|fdd|D7}t|dkr@|dd|dd|S)N)z+++z---z@@zERROR:r&cs(g|] }|dddkr|qS) r9r)rCsplit)r6rF) filtering_outr r sz5Flake8Isort4._format_isort_output..r9 )getvalue splitlinesleninsertappendr+)r Z isort_bufferZ valid_linesr )rIr r,zs    z!Flake8Isort4._format_isort_outputcCs\dht|jB}xFt|jD]8}||krFt|jdkrT|jq|jdPqWdS)agEnsure single end-of-file newline in `isort.SortImports.in_lines` isort fixes EOF blank lines but this change should be suppressed as Flake8 will also flag them. Args: sort_imports (isort.SortImports): The isorts results object. Returns: isort.SortImports: The modified isort results object. r&r9N)setrAreversedr?rCrNpoprP) sort_importsZ to_removerFr r r r=s    z#Flake8Isort4._fixup_sortimports_eofcCsXxRt|jD]D\}}d|kr x2t|j|D]\}}|j|||q2Wq WdS)aSplit-up wrapped imports newlines in `SortImports.out_lines` isort combines wrapped lines into a single list entry string in `out_lines` whereas `in_lines` are separate strings so for diff comparison these need to be comparable. Args: sort_imports (isort.SortImports): The isorts results object. Returns: isort.SortImports: The modified isort results object. rKN) enumerater@rSrMrO)rTidxrFZnew_idxnew_liner r r r<s z'Flake8Isort4._fixup_sortimports_wrappedN) rrr__doc__r4r-r, staticmethodr=r<r r r r r%:s ) r%c@s eZdZdZddZddZdS) Flake8Isort5zclass for isort >=5c csn|j|jk r(t|j}tjj|jd}nd}tjjtd}d|j }d}d}t |}t }t }y,t |tj j ||||d}WdQRXWnFtjjk rYn0tjjk r} zt| Wdd} ~ XYnX|rj|} dt|jdd| jddd|jd|jd } |d | }x:|| D],\} } |jrR| |7} | d | t|fVq:WdS) N)Z settings_pathr&F) input_stream output_streamconfigr'T)keependsz {}:beforez{}:after)fromfiletofilerKr)rrrr*settingsZConfigparentcwdr+r rrapiZ sort_stream exceptionsZ FileSkippedZ ISortErrorwarningswarnrLrrMr5isort_linenum_msgrr.)r r'Z isort_configZ input_stringr1Z isort_changedr[r\Z isort_stdouteZoutlinesZ diff_deltar2r3r r r r4sH         zFlake8Isort5.runccs6d}g}g}x|D]}|dddrPt|dddddd}qn|sVq|dddrn|d7}q|d ddr|d kr||jfV|d7}q||dd||jfV|d7}q|d ddr|d kr||jfVq|||fqWx0|D](\}}|dd|kr||j fVqWdS) zParse unified diff for changes and generate messages Args ---- udiff : unified diff delta Yields ------ tuple: A tuple of the specific isort line number and message. rz@@r8NrG,r9r:r;) rMrBintrHrCr"rPr r!r#)r udiffr2rEmovesrFr r r rhs2 $       zFlake8Isort5.isort_linenum_msgN)rrrrXr4rhr r r r rZs'rZrd) contextlibrdifflibrriorpathlibrr*rfrobjectrr%rZhasattrZ Flake8Isortr r r r s     *}U