ó
è¾bc           @   sm   d  Z  d d l m Z d d l m Z m Z m Z m Z m Z m	 Z	 d Z
 d Z d e j f d „  ƒ  YZ d S(	   sÜ  Adjust some old Python 2 idioms to their modern counterparts.

* Change some type comparisons to isinstance() calls:
    type(x) == T -> isinstance(x, T)
    type(x) is T -> isinstance(x, T)
    type(x) != T -> not isinstance(x, T)
    type(x) is not T -> not isinstance(x, T)

* Change "while 1:" into "while True:".

* Change both

    v = list(EXPR)
    v.sort()
    foo(v)

and the more general

    v = EXPR
    v.sort()
    foo(v)

into

    v = sorted(EXPR)
    foo(v)
i   (   t
   fixer_base(   t   Callt   Commat   Namet   Nodet	   BlankLinet   symss0   (n='!=' | '==' | 'is' | n=comp_op< 'is' 'not' >)s(   power< 'type' trailer< '(' x=any ')' > >t	   FixIdiomsc           B   sQ   e  Z e Z d  e e e e f Z d „  Z d „  Z d „  Z	 d „  Z
 d „  Z RS(   sº  
        isinstance=comparison< %s %s T=any >
        |
        isinstance=comparison< T=any %s %s >
        |
        while_stmt< 'while' while='1' ':' any+ >
        |
        sorted=any<
            any*
            simple_stmt<
              expr_stmt< id1=any '='
                         power< list='list' trailer< '(' (not arglist<any+>) any ')' > >
              >
              '\n'
            >
            sort=
            simple_stmt<
              power< id2=any
                     trailer< '.' 'sort' > trailer< '(' ')' >
              >
              '\n'
            >
            next=any*
        >
        |
        sorted=any<
            any*
            simple_stmt< expr_stmt< id1=any '=' expr=any > '\n' >
            sort=
            simple_stmt<
              power< id2=any
                     trailer< '.' 'sort' > trailer< '(' ')' >
              >
              '\n'
            >
            next=any*
        >
    c         C   sJ   t  t |  ƒ j | ƒ } | rF d | k rF | d | d k rB | Sd  S| S(   Nt   sortedt   id1t   id2(   t   superR   t   matcht   None(   t   selft   nodet   r(    (    s.   /usr/lib/python2.7/lib2to3/fixes/fix_idioms.pyR   O   s    c         C   sd   d | k r |  j  | | ƒ Sd | k r8 |  j | | ƒ Sd | k rT |  j | | ƒ St d ƒ ‚ d  S(   Nt
   isinstancet   whileR   s   Invalid match(   t   transform_isinstancet   transform_whilet   transform_sortt   RuntimeError(   R   R   t   results(    (    s.   /usr/lib/python2.7/lib2to3/fixes/fix_idioms.pyt	   transformZ   s    c         C   s™   | d j  ƒ  } | d j  ƒ  } d | _ d | _ t t d ƒ | t ƒ  | g ƒ } d | k r‰ d | _ t t j t d ƒ | g ƒ } n  | j | _ | S(   Nt   xt   Tu    u    u
   isinstancet   nu   not(   t   clonet   prefixR   R   R   R   R   t   not_test(   R   R   R   R   R   t   test(    (    s.   /usr/lib/python2.7/lib2to3/fixes/fix_idioms.pyR   d   s    		!	!c         C   s*   | d } | j  t d d | j ƒƒ d  S(   NR   u   TrueR   (   t   replaceR   R   (   R   R   R   t   one(    (    s.   /usr/lib/python2.7/lib2to3/fixes/fix_idioms.pyR   p   s    
c         C   sv  | d } | d } | j  d ƒ } | j  d ƒ } | rW | j t d d | j ƒƒ nR | r | j ƒ  } d | _ | j t t d ƒ | g d | j ƒƒ n t d ƒ ‚ | j ƒ  | j } d	 | k rr| r| j d	 ƒ d
 | d
 j f }	 d	 j	 |	 ƒ | d
 _ qr| j
 st ‚ | j d  k s+t ‚ t ƒ  }
 | j
 j |
 ƒ | j |
 k sYt ‚ | j d	 ƒ d
 |
 _ n  d  S(   Nt   sortt   nextt   listt   expru   sortedR   u    s   should not have reached hereu   
i    (   t   getR    R   R   R   R   R   t   removet
   rpartitiont   joint   parentt   AssertionErrort   next_siblingR   R   t   append_child(   R   R   R   t	   sort_stmtt	   next_stmtt	   list_callt   simple_exprt   newt   btwnt   prefix_linest   end_line(    (    s.   /usr/lib/python2.7/lib2to3/fixes/fix_idioms.pyR   t   s0    

	
	 	(   t   __name__t
   __module__t   Truet   explicitt   TYPEt   CMPt   PATTERNR   R   R   R   R   (    (    (    s.   /usr/lib/python2.7/lib2to3/fixes/fix_idioms.pyR   %   s   '		
		N(   t   __doc__t    R    t
   fixer_utilR   R   R   R   R   R   R;   R:   t   BaseFixR   (    (    (    s.   /usr/lib/python2.7/lib2to3/fixes/fix_idioms.pyt   <module>   s
   .