�
{ fc @ s� d Z d d l m Z d d l m Z d d l m Z m Z m Z m Z d � Z
d � Z d � Z d � Z
d � Z d
� Z d e j f d � � YZ d
S( s� Fixer for __metaclass__ = X -> (metaclass=X) methods.
The various forms of classef (inherits nothing, inherits once, inherints
many) don't parse the same in the CST so we look at ALL classes for
a __metaclass__ and if we find one normalize the inherits to all be
an arglist.
For one-liner classes ('class X: pass') there is no indent/dedent so
we normalize those into having a suite.
Moving the __metaclass__ into the classdef can also cause the class
body to be empty so there is some special casing for that as well.
This fixer also tries very hard to keep original indenting and spacing
in all those corner cases.
i ( t
fixer_base( t token( t Namet symst Nodet Leafc C s� x� | j D]� } | j t j k r, t | � S| j t j k r
| j r
| j d } | j t j k r� | j r� | j d } t | t � r� | j d k r� t
Sq� q
q
Wt S( s� we have to check the cls_node without changing it.
There are two possibilities:
1) clsdef => suite => simple_stmt => expr_stmt => Leaf('__meta')
2) clsdef => simple_stmt => expr_stmt => Leaf('__meta')
i t
__metaclass__( t childrent typeR t suitet
has_metaclasst simple_stmtt expr_stmtt
isinstanceR t valuet Truet False( t parentt nodet expr_nodet left_side( ( s3 /usr/lib64/python2.7/lib2to3/fixes/fix_metaclass.pyR
s
c C s� x' | j D] } | j t j k r
d Sq
Wx? t | j � D]" \ } } | j t j k r: Pq: q: Wt d � � t t j g � } xC | j | d r� | j | d } | j | j
� � | j � q� W| j | � | } d S( sf one-line classes don't get a suite in the parse tree so we add
one to normalize the tree
Ns No class suite and no ':'!i ( R R R R t enumerateR t COLONt
ValueErrorR t append_childt clonet remove( t cls_nodeR t iR t move_node( ( s3 /usr/lib64/python2.7/lib2to3/fixes/fix_metaclass.pyt fixup_parse_tree- s
c
C s� x7 t | j � D]" \ } } | j t j k r Pq q Wd S| j � t t j g � } t t j | g � } x; | j | r� | j | } | j
| j � � | j � qn W| j | | � | j d j d } | j d j d } | j
| _
d S( s� if there is a semi-colon all the parts count as part of the same
simple_stmt. We just want the __metaclass__ part so we move
everything after the semi-colon into its own simple_stmt node
Ni ( R R R R t SEMIR R R R R R R t insert_childt prefix(
R R t stmt_nodet semi_indR t new_exprt new_stmtR t new_leaf1t old_leaf1( ( s3 /usr/lib64/python2.7/lib2to3/fixes/fix_metaclass.pyt fixup_simple_stmtG s
c C s: | j r6 | j d j t j k r6 | j d j � n d S( Ni����( R R R t NEWLINER ( R ( ( s3 /usr/lib64/python2.7/lib2to3/fixes/fix_metaclass.pyt remove_trailing_newline_ s "c c s� x3 | j D] } | j t j k r
Pq
q
Wt d � � x� t t | j � � D]� \ } } | j t j k rL | j rL | j d } | j t j k r� | j r� | j d } t | t
� r� | j d k r� t | | | � t
| � | | | f Vq� q� qL qL Wd S( Ns No class suite!i u
__metaclass__( R R R R R t listR R R R
R R R( R* ( R R R t simple_nodeR t left_node( ( s3 /usr/lib64/python2.7/lib2to3/fixes/fix_metaclass.pyt
find_metasd s "
c C s� | j d d d � } x, | rD | j � } | j t j k r Pq q Wxm | r� | j � } t | t � r� | j t j k r� | j r� d | _ n d S| j | j d d d � � qH Wd S( s� If an INDENT is followed by a thing with a prefix then nuke the prefix
Otherwise we get in trouble when removing __metaclass__ at suite start
Ni����u (
R t popR R t INDENTR
R t DEDENTR! t extend( R t kidsR ( ( s3 /usr/lib64/python2.7/lib2to3/fixes/fix_metaclass.pyt fixup_indent{ s ! t FixMetaclassc B s e Z e Z d Z d � Z RS( s
classdef<any*>
c C s' t | � s d St | � d } x- t | � D] \ } } } | } | j � q- W| j d j } t | j � d k r� | j d j t j k r� | j d } q�| j d j
� } t t j | g � } | j d | � n� t | j � d k rt t j g � } | j
d | � n~ t | j � d k r�t t j g � } | j
d t t j d � � | j
d | � | j
d t t j d � � n t d � � | j d j d }
d
|
_ |
j } | j r�| j t t j d � � d |
_ n d
|
_ | j d } | j t j k st � d
| j d _ d
| j d _ | j | � t | � | j s�| j � t | d � }
| |
_ | j |
� | j t t j d � � n� t | j � d k r#| j d j t j k r#| j d j t j k r#t | d � }
| j
d |
� | j
d t t j d � � n d S( Ni i i i i i u )u (s Unexpected class definitiont metaclassu ,u u i u passu
i����i����( R
R t NoneR. R R R t lenR t arglistR R t set_childR R R t RPARt LPARR R R! R t COMMAR t AssertionErrorR4 R) R0 R1 ( t selfR t resultst last_metaclassR R t stmtt text_typeR9 R t meta_txtt orig_meta_prefixR t pass_leaf( ( s3 /usr/lib64/python2.7/lib2to3/fixes/fix_metaclass.pyt transform� s`
( t __name__t
__module__R t
BM_compatiblet PATTERNRG ( ( ( s3 /usr/lib64/python2.7/lib2to3/fixes/fix_metaclass.pyR5 � s N( t __doc__t R t pygramR t
fixer_utilR R R R R
R R( R* R. R4 t BaseFixR5 ( ( ( s3 /usr/lib64/python2.7/lib2to3/fixes/fix_metaclass.pyt <module> s " 