U
if @ sr d dl mZ d dlmZ d dlmZmZmZ dd Zdd Z dd Z
d
d Z d d
Z dd Z
G dd dejZdS ) )
fixer_base)token)symsNodeLeafc C sz | j D ]n}|jtjkr"t| S |jtjkr|j r|j d }|jtjkr|j r|j d }t|tr|j dkr dS qdS )N
__metaclass__TF)
childrentyper suite
has_metaclass simple_stmt expr_stmt
isinstancer value)parentnode expr_nodeZ left_side r @/opt/alt/python38/lib64/python3.8/lib2to3/fixes/fix_metaclass.pyr s
r c C s | j D ]}|jtjkr d S qt| j D ]\}}|jtjkr( qJq(tdttjg }| j |d d r| j |d }| |
| qV| | |}d S )NzNo class suite and no ':'! ) r r
r r enumerater COLON
ValueErrorr append_childcloneremove)cls_noder ir move_noder r r fixup_parse_tree- s
r c
C s t |jD ]\}}|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_childprefix)
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$qtdtt|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|tr^|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|jdkr|jd jtjkrp|jd }n(|jd } t
tj| g}| d| nt|jdkrt
tjg }| d| nZt|jdkrt
tjg }| dt
tjd | d| | dt
tjd ntd |jd jd }
d
|
_|
j} |jrZ|t
tjd d |
_nd
|
_|jd } d
| jd _d
| jd _|| t| |js| t
|d}
| |
_||
|t
tjd nbt|jdkr4|jd jtjkr4|jd jtjkr4t
|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 RPARLPARr 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