B
    	Q[                 @   sj   d Z ddlmZ ddlmZmZmZmZmZm	Z	 ddl
mZ ddlmZmZ dd ZG dd	 d	ejZd
S )zn
Fixer for (metaclass=X) -> __metaclass__ = X
Some semantics (see PEP 3115) may be altered in the translation.    )
fixer_base)NamesymsNodeLeafNewline	find_root)token)indentationsuitifyc       
      C   s  d }x| j D  ]}|j }|jtjkrd|d ttjdkrb|d ttjdkrb|d rb|g| }P q|jtjkrx|j D ]}|rP |jtj	kr|}qxt|t
krxd  } }}xX|j D ]N}	|	ttjdkr|	}q|r|	ttjdkr|	}q|r|r|	}||||f}P qW qxW qW |S )Nr   	metaclass   =   )childrentyper   Zargumentr   r	   NAMEEQUALZarglistCOMMAr   )
parentresultsnodeZkidsZchildZcommametaequalnamearg r   @lib/python3.7/site-packages/libpasteurize/fixes/fix_metaclass.pyhas_metaclass   s6    
 r   c               @   s   e Zd ZdZdd ZdS )FixMetaclassz
    classdef<any*>
    c             C   s   t |}|sd S x|D ]}|  qW ttjd}ttjddd}|}d|_ttj	|||g}t
| x||jD ]r}	|	jtjkrnx`|	jD ]V}
|
jtjkr|	j|
d }ttj|
j}|	|| |	|t  |	|| P qW qnW d S )NZ__metaclass__r    )prefixr   )r   remover   r	   r   r   r!   r   r   Zatomr   r   r   ZsuiteINDENTindexvalueZinsert_childr   )selfr   r   Zmeta_resultsr   targetr   r   Z	stmt_nodeitemZstmtZlocZidentr   r   r   	transform5   s*     
zFixMetaclass.transformN)__name__
__module____qualname__ZPATTERNr)   r   r   r   r   r   /   s   r   N)__doc__Zlib2to3r   Zlib2to3.fixer_utilr   r   r   r   r   r   Zlib2to3.pygramr	   Zlibfuturize.fixer_utilr
   r   r   ZBaseFixr   r   r   r   r   <module>   s    $