ó ~ˆ]c@`sdZddlmZmZmZmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZmZddlmZddlmZddlmZdd lmZdd lmZdd lmZd efd „ƒYZdd„ZdS(uàisort.py. Exposes a simple library to sort through imports within Python code usage: SortImports(file_name) or: sorted = SortImports(file_contents=file_contents).output Copyright (C) 2013 Timothy Edmund Crosley Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(t OrderedDictt namedtuple(tdatetime(t unified_diffi(tsettings(tFindersManager(tnsorted(tinputt SortImportsc B`s|eZeZeZdddeeedededd„ Zed„ƒZ d„Z e d„ƒZ d„Z d„Ze d„ƒZd„Ze eedd „ƒZdd „Zd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z e dd„ƒZ!e d„ƒZ"e d„ƒZ#d„Z$d„Z%d„Z&RS( uc " K`së | r.|r.tjjtjj|ƒƒ}n|p=tjƒ}tj|ƒjƒ|_xÖ| j ƒD]È\} }| j ddƒj ƒ}|dkr t |jj |ƒƒttfkr | jdƒrôtt|j|ƒj|ƒƒ|j|n9d |kr d |_>n!d!|kr8d!|_>n d"|_>|j?|j>ƒ|_@tA|j@ƒ|_B|jBdks |j@d gdgfks |jd#rÇx$|jD]}|j@jC|ƒqªWntA|j@ƒ|_D| s|rø|j?d$ƒd%nd&|_En | |_Eg|_Fiid'6id(6id)6iid(6id'6d*6|_GtHƒ|_Ii|_J|jd}tKd+|ƒg|D] }|^q~Œ|_LxEtMjN|jL|jd,ƒD]'}itHƒd(6tHƒd'6|jI|jY|jFƒ|_:|jd/r• y/tZ|j[|jF|j>ƒ|j"d0ddƒWq• t\k r‘ ||_:t|_]yEtZ|j[|j@|j>ƒ|j"d0ddƒt%d1j&|j"ƒƒWn't\k rŒ t%d2j&|j"ƒƒnXdSXn|rx |j:}|}|jd3r |j |j>dƒj d dƒj d4dƒ}|j |j>dƒj d dƒj d4dƒ}n|jƒ|jƒkrV |jdrR t%d5j&|j"ƒƒndSt%d6j&|j"ƒƒt|_]n|s‹ |jd7r› |j^|ƒnL|rå t;j_dd8krÏ |j:j`|j!ƒ|_:nt;j<j=|j:ƒn|rç | rç |j:|kr dS|r~ |j^|ƒd} x]| dEkrz tad?j&|j"ƒƒj ƒ} | dFkr[ dS| dGkr! t;jbdƒq! q! Wnt(j)|j"d|j!d@dAddƒ=}!|jdBsÎ t%dCj&|j"ƒƒn|!j=|j:ƒWdQXndS(HNunot_uusectionsuforce_alphabetical_sortu'force_alphabetical_sort_within_sectionsu no_sectionsiulines_between_typesu from_firstuindentu u'u"utabu ucomment_prefixuremove_importsu add_importsuimport_headingu# uutf-8uverboseubWARNING: {0} was skipped as it's listed in 'skip' setting or matches a glob in 'skip_glob' settingurbtencodingtnewlineucWARNING: {} was skipped as it couldn't be opened with the given {} encoding or {} fallback encodingiuisort:u skip_fileu line_endingu u u u force_addsu.iÿÿÿÿupyufromustraightunesteduaboveuSectionsuforced_separatetconfigtsectionsuatomicuexecuSERROR: {0} isort would have introduced syntax errors, please report to the project!u'ERROR: {0} File contains syntax errors.uignore_whitespaceu u#SUCCESS: {0} Everything Looks Good!u*ERROR: {0} Imports are incorrectly sorted.u show_diffiuyesuyunounuquituqu*Apply suggested changes to '{0}' [y/n/q]? tmodeuwuquietu Fixing {0}uisort:skip_file(uyesuyunounuquituq(unoun(uquituq(ctostpathtdirnametabspathtgetcwdRt from_pathtcopyRtitemstreplacetlowerttypetgettlistttuplet startswithtsett differencetuniontupdatetTruetstrtisdigittinttstript place_importstimport_placementst_format_simplifiedtremove_importst_format_naturalt add_importst_section_commentst file_encodingt file_patht should_skiptskippedtprinttformattNonetiotopent coding_checktreadtUnicodeDecodeErrortFalseR tseektUnsupportedOperationtIOErrortcodecst getreadert basestringtbytest NameErrort isinstancetdecodet splitlinestoutputtsyststdouttwritetline_separatortsplittin_linestlentoriginal_lengthtappendtnumber_of_linest extensiont out_linestcommentsRtimportstas_mapRRt itertoolstchainR tfindertindext import_indext_first_comment_index_startt_first_comment_index_endt_parset_add_formatted_importst length_changetpoptjointcompilet_strip_top_commentst SyntaxErrortincorrectly_sortedt _show_difft version_infotencodeR texit("tselfR2t file_contentstfile_twrite_to_stdouttcheckt show_difft settings_patht ask_to_applytrun_patht check_skipRTtsetting_overridestkeytvaluet access_keytindenttremovaltadditiont file_nametfR1tfile_to_import_sorttencoding_successtreadert need_decodet add_importt section_namestnametsectiont check_outputt check_againsttanswert output_file((s*lib/python2.7/site-packages/isort/isort.pyt__init__2sf !0,,    &  ,,(                 #   !         8%  3   +#%      % /   +    -0      * cC`s|j S(N(Rh(Rm((s*lib/python2.7/site-packages/isort/isort.pytcorrectly_sorted sc C`s¨x¡t|jdƒ|jjdƒd|jdd|jddt|jrhtjtjj |jƒƒn tj ƒƒdttj ƒƒƒD]}t j j |ƒqŠWdS(Nitfromfileu:beforettofileu:aftert fromfiledatet tofiledate(RRHRIR2R&Rt fromtimestampRRtgetmtimetnowRJRKRL(RmRntline((s*lib/python2.7/site-packages/isort/isort.pyRi s   6cC`sFtj|ƒ}x'|r8|djdƒr8|d}qW|j|ƒS(u:Strips # comments that exist at the top of the given linesiu#i(RR Rd(tlinesRM((s*lib/python2.7/site-packages/isort/isort.pyRfscC`s|jj|ƒS(u¤Tries to determine if a module is a python std import, third party import, or project code: if it can't determine - it assumes it is project code (R[tfind(Rmt module_name((s*lib/python2.7/site-packages/isort/isort.pyt place_module!scC`s#|j|j}|jd7_|S(uDReturns the current line from the file while incrementing the index.i(ROR\(RmR•((s*lib/python2.7/site-packages/isort/isort.pyt _get_line)scC`s:d|krdS|jdƒr#dS|jdƒr6dSdS(uPIf the current line is an import line it will return its type (from or straight)u isort:skipNuimport ustraightufrom ufrom(R (R•((s*lib/python2.7/site-packages/isort/isort.pyt _import_type/s  cC`s|j|jkS(u.returns True if we are at the end of the file.(R\RS(Rm((s*lib/python2.7/site-packages/isort/isort.pyt_at_end9sc C`s€tjd|ƒ}|rF|dr(dnd}|j|jƒƒ}nd}|rgt|ƒjƒ}n t|ƒ}|rÏ|drÏ|jƒrªt|ƒdkrªd}qÏ|d d!jƒrÆd }qÏd }n|d sè|jƒ}n|dksd t|ƒjƒ|kr|d}n|d t|ƒjƒ}dj ||dkrSdpVd ||rytt|ƒƒd|p||ƒS(Nu ^(\.+)\s*(.*)ureverse_relativeu u_uu order_by_typeiuAiuBuCucase_sensitiveu length_sort_u length_sortu {0}{1}{2}u force_to_topu:( tretmatchRdtgroupsR&RtisupperRPR7R6( R˜Rt sub_importst ignore_caset section_nameRžtseptprefixt length_sort((s*lib/python2.7/site-packages/isort/isort.pyt _module_key=s*     ( "cC`s`|jdsI|rEdj|j|ƒd|jddj|ƒƒpH|S|o_|j|ƒdS(uY Returns a string with comments added if ignore_comments is not set. uignore_commentsu {0}{1} {2}iucomment_prefixu; (RR6t_strip_commentsRd(RmRVtoriginal_string((s*lib/python2.7/site-packages/isort/isort.pyt _add_commentsZs   c C`s#|jd}t|ƒ|jdkrÎ|tjjkrÎ|}d}d|krk|jddƒ\}}nx±dD]U}dtj|ƒd}tj ||ƒrr|j ƒj |ƒ rrtj||ƒ}|rìd j |d |ƒ|d |j|ƒn|j| ƒqqWqWdS( Nufrom {0} import ufromuno_inline_sortuforce_single_lineRxc`sˆj|ˆjtˆdˆƒS(NR£(R§RR%(Rx(R¢R‡Rm(s*lib/python2.7/site-packages/isort/isort.pyt®tu{0}.{1}c`sAi|]7\}}|ˆjkrdj|ˆj|ƒ|“qS(u {0} as {1}(RXR6(t.0t from_importt sub_module(Rm(s*lib/python2.7/site-packages/isort/isort.pys µs  ucombine_as_importsu*u combine_staru{0}*iustraightu{}.{}unestedu{0} {1}u;ucomment_prefixuaboveu, uuforce_grid_wrapu line_lengthiumulti_line_outputics`s|]}t|ƒVqdS(N(RP(RÅtx((s*lib/python2.7/site-packages/isort/isort.pys su ((ii(R-R6RRWRR tzipRRcR\RVR¯RªR7RRRRMRdR¼R=tremoveR%RPt_multi_line_reformattmaxRN(Rmt from_modulesR‡R¾R¢R¿t import_startt from_importsR•RÆt sub_modulest as_importsRVtimport_statementtimport_statementst from_commentstsingle_import_lineR¶tabove_commentst star_importtfrom_import_sectiontdo_multiline_reformattforce_grid_wraptother_import_statement((R¢R‡Rms*lib/python2.7/site-packages/isort/isort.pyt_add_from_imports¦sÊ $ % ' & $    "%  " "   %      + +   #  cC`s¨tjj|jdjƒ}t|d||jƒ}dt|ƒd}|jd}|jdpn|jd}||tj|ƒ||||ƒ} |jdr| j |j ƒ} t| ƒ} t| ƒdkrït d „| d Dƒƒ} nd } | } xt| d ƒ| kr{t| ƒ| kr{|d kr{| } |d8}||tj|ƒ||||ƒ} | j |j ƒ} qþWn| j |j ƒd kr¤|j | ƒS| S( Numulti_line_outputu_output_u iuindentu wrap_lengthu line_lengthubalanced_wrappingcs`s|]}t|ƒVqdS(N(RP(RÅR•((s*lib/python2.7/site-packages/isort/isort.pys -siÿÿÿÿii (RR«t_fieldsRRtgetattrt _output_gridRPRRNRMtmintcountR¯(RmRÎRÏRVt output_modet formattertdynamic_indentR{t line_lengthRÒR–t line_counttminimum_lengthtnew_import_statement((s*lib/python2.7/site-packages/isort/isort.pyRË!s0     c`sþˆjd}tjˆjˆjdƒ}ˆjdrÁigd6id6ˆjd‹sufrom uimport cs`s|]}|jƒVqdS(N(R)(RÅR•((s*lib/python2.7/site-packages/isort/isort.pys ”sufromiu*u(u)u\u importuimport(uimport (u import uu{|u{ u|}u }uasu.ucombine_as_importsustraightuverboseu)from-type place_module for %s returned %su[WARNING: could not place module {0} of line {1} -- Do you need to define a default section?unestedu"""u'''uabovecs`s|]}|dfVqdS(N(R7(RÅR¿((s*lib/python2.7/site-packages/isort/isort.pys ùsu)else-type place_module for %s returned %sN(-R=RìRRœRšRR\RîR0R]R RNtupperR*R+R%R›RURRR¨RR)RPR³RMtrstripR°RdRXRRVRcR™R5R6RWRt setdefaultR¼RÌR_RëRR$R7(Rmtraw_lineR•tstatement_indexRR‡Rt import_typetnested_commentsRRVRt stripped_lineRûRÆtitemRWR\R¿t import_fromt placed_moduletroott import_nametassociated_commenttlast((s*lib/python2.7/site-packages/isort/isort.pyR`ts   #   &  74&&!$!?$&!$$*' 6=+       #/"2 33 2   /"2  - 2  N('t__name__t __module__R=RhR4R7R%RŒtpropertyRRit staticmethodRfR™RšR›RœR§RªR¯RÂRÜRËRaRßRRRR R R R RR¨R,R.RîRR`(((s*lib/python2.7/site-packages/isort/isort.pyR .sH Õ     .  {  €        # uutf-8cC`sltjdƒ}xVt|dƒD]E\}}tj||ƒ}|rT|djdƒS|dkrPqqW|S(Nscoding[:=]\s*([-\w.]+)iiuasciii(RReRítfindallRG(R–tdefaulttpatternt line_numberR•RŸ((s*lib/python2.7/site-packages/isort/isort.pyR:s (t__doc__t __future__RRRRRARR8RYRRRJt collectionsRRRtdifflibRRÄRtfindersR tnaturalR t pie_sliceR tobjectR R:(((s*lib/python2.7/site-packages/isort/isort.pyts("       ÿÿÿí