Yf @ 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 Gd d d e j Z
d S) a 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)
)
fixer_base)CallCommaNameNode BlankLinesymsz0(n='!=' | '==' | 'is' | n=comp_op< 'is' 'not' >)z(power< 'type' trailer< '(' x=any ')' > >c sn e Z d Z d Z d e e e e f Z f d d Z d d Z d d Z d d
Z
d d Z S)
FixIdiomsTa
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 ')' > >
>
'\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 sJ t t | j | } | rF d | k rF | d | d k rB | Sd S| S)NsortedZid1Zid2)superr match)selfnoder) __class__ =/opt/alt/python35/lib64/python3.5/lib2to3/fixes/fix_idioms.pyr O s zFixIdioms.matchc C sd d | k r | j | | Sd | k r8 | j | | Sd | k rT | j | | St d d S)N
isinstancewhiler
z
Invalid match)transform_isinstancetransform_whiletransform_sort RuntimeError)r
r resultsr r r transformZ s zFixIdioms.transformc C s | d j } | d j } d | _ d | _ t t d | t | g } d | k r d | _ t t j t d | g } | j | _ | S)NxT r nnot)cloneprefixr r r r r Znot_test)r
r r r r Ztestr r r r d s ! z FixIdioms.transform_isinstancec C s* | d } | j t d d | j d S)Nr Truer" )replacer r" )r
r r Zoner r r r p s
zFixIdioms.transform_whilec C s: | 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 r6| r| j d d
| d
j f } d j | | d
_ n/ t
}
| j j |
| j d d
|
_ d S) Nsortnextlistexprr
r" r z should not have reached here
)
getr$ r r" r! r r remove
rpartitionjoinr parentZ append_child) r
r r Z sort_stmtZ next_stmtZ list_callZ simple_exprnewZbtwnZ prefix_linesZend_liner r r r t s*
zFixIdioms.transform_sort) __name__
__module__ __qualname__ZexplicitTYPECMPZPATTERNr r r r r r r )r r r % s '
r N)__doc__r r Z
fixer_utilr r r r r r r5 r4 ZBaseFixr r r r r s
.