a
XC?h � @ sr d dl mZ d dlmZ d dlmZmZmZ dd� Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
G dd� dej�ZdS )� )�
fixer_base)�token)�syms�Node�Leafc C sz | j D ]n}|jtjkr"t|� S |jtjkr|j r|j d }|jtjkr|j r|j d }t|t�r|j dkr dS qdS )N� �
__metaclass__TF)
�children�typer �suite�
has_metaclass� simple_stmt� expr_stmt�
isinstancer �value)�parent�node� expr_nodeZ left_side� r �@/opt/alt/python39/lib64/python3.9/lib2to3/fixes/fix_metaclass.pyr s
�r c C s� | j D ]}|jtjkr d S qt| j �D ]\}}|jtjkr( qJq(td��ttjg �}| j |d d � r�| j |d }|� |�
� � |� � qV| � |� |}d S )NzNo class suite and no ':'!� ) r r
r r � enumerater �COLON�
ValueErrorr � append_child�clone�remove)�cls_noder �ir � move_noder r r �fixup_parse_tree- s
r c
C s� t |j�D ]\}}|jtjkr
q(q
d S |�� ttjg �}ttj |g�}|j|d � rz|j| }|�
|� � � |�� qJ| � ||� |jd jd }|jd jd } | j
|_
d S )Nr )r r r
r �SEMIr r r r r
r r � insert_child�prefix)
r r Z stmt_nodeZsemi_indr Znew_exprZnew_stmtr Z new_leaf1Z old_leaf1r r r �fixup_simple_stmtG s
r$ c C s* | j r&| j d jtjkr&| j d �� d S )N�����)r r
r �NEWLINEr )r r r r �remove_trailing_newline_ s r' c c s� | j D ]}|jtjkr q$qtd��tt|j ��D ]t\}}|jtjkr2|j r2|j d }|jtjkr2|j r2|j d }t |t
�r2|j dkr2t |||� t
|� |||fV q2d S )NzNo class suite!r r )r r
r r r �listr r
r r r r r$ r' )r r r Z simple_noder Z left_noder r r �
find_metasd s
� r) c C sz | j d d d� }|r,|�� }|jtjkrq,q|rv|�� }t|t�r^|jtjkr^|jrZd|_d S |� |j d d d� � q,d S )Nr% � )
r �popr
r �INDENTr r �DEDENTr# �extend)r Zkidsr r r r � fixup_indent{ s r/ c @ s e Zd ZdZdZdd� ZdS )� FixMetaclassTz
classdef
c C s8 t |�s d S t|� d }t|�D ]\}}}|}|�� q |jd j}t|j�dkr�|jd jtjkrp|jd }n(|jd � � } t
tj| g�}|� d|� n�t|j�dkr�t
tjg �}|� d|� nZt|j�dk�rt
tjg �}|� dt
tjd�� |� d|� |� dt
tjd�� ntd ��|jd jd }
d
|
_|
j} |j�rZ|�t
tjd �� d |
_nd
|
_|jd } d
| jd _d
| jd _|�|� t|� |j�s�|�� t
|d�}
| |
_|�|
� |�t
tjd�� nbt|j�dk�r4|jd jtjk�r4|jd jtjk�r4t
|d�}
|� d|
� |� dt
tjd�� d S )Nr � � � � r �)�(zUnexpected class definition� metaclass�,� r* r �pass�
�����r% )r r r) r r r
�lenr �arglistr r Z set_childr" r r �RPAR�LPARr r r# r �COMMAr/ r& r, r- )�selfr ZresultsZlast_metaclassr r ZstmtZ text_typer> r Zmeta_txtZorig_meta_prefixr Z pass_leafr r r � transform� sb
��
zFixMetaclass.transformN)�__name__�
__module__� __qualname__Z
BM_compatibleZPATTERNrC r r r r r0 � s r0 N)r* r Zpygramr Z
fixer_utilr r r r r r$ r' r) r/ ZBaseFixr0 r r r r � s