a
    X%b                     @   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 }| 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| } qq
|jtjkr
|j D ]}|r q
|jtj	kr|}qvt|t
krvd  } }}|j D ]P}	|	ttjdkr|	}q|r|	ttjdkr|	}q|r|r|	}||||f} qvqqvq
|S )Nr   	metaclass   =   )childrentyper   Zargumentr   r	   NAMEEQUALZarglistCOMMAr   )
parentresultsnodeZkidschildcommametaequalnamearg r   q/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/libpasteurize/fixes/fix_metaclass.pyhas_metaclass   s8    



r    c                   @   s   e Zd ZdZdd ZdS )FixMetaclassz
    classdef<any*>
    c                 C   s   t |}|sd S |D ]}|  qttjd}ttjddd}|}d|_ttj	|||g}t
| |jD ]p}	|	jtjkrh|	jD ]X}
|
jtjkr~|	j|
d }ttj|
j}|	|| |	|t  |	||  qhq~qhd S )N__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itemstmtloc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    $