f d Z ddlmZ ddlmZ ddlmZmZmZ d Z d Z
d Z d Z d Z
d
Z G d d ej Zd
S )a Fixer for __metaclass__ = X -> (metaclass=X) methods.
The various forms of classef (inherits nothing, inherits once, inherits
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.
)
fixer_base)token)symsNodeLeafc P | j D ]}|j t j k rt | c S |j t j k r`|j rY|j d }|j t j k r7|j r0|j d }t |t r|j dk r dS dS )z 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')
__metaclass__TF)
childrentyper suite
has_metaclass simple_stmt expr_stmt
isinstancer value)parentnode expr_node left_sides B/opt/alt/python311/lib64/python3.11/lib2to3/fixes/fix_metaclass.pyr r s 9
" " && & & &
Y$*
*
*t}
*
a(I~ //I4F/ % .q 1 i.. !?:: 44 5 c
| j D ]}|j t j k r dS t | j D ] \ }}|j t
j k r n t d t t j g }| j |dz d r]| j |dz }| |
| | j |dz d ]| | |}dS )zf one-line classes don't get a suite in the parse tree so we add
one to normalize the tree
NzNo class suite and no ':'! ) r r r r
enumerater COLON
ValueErrorr append_childcloneremove)cls_noder ir
move_nodes r fixup_parse_treer$ - s ! 9
" " FF #
X .// 7 74 9 # # E $ 5666
R E
AaCDD
! %ac*
9??,,---
AaCDD
!
% DDDr c n t |j D ] \ }}|j t j k r n dS | t
t j g }t
t j |g }|j |d rW|j | }|
| | |j |d W| || |j d j d }|j d j d } | j
|_
dS )z 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
Nr )r r r r SEMIr r r r r r r insert_childprefix)
r r" stmt_nodesemi_indr new_exprnew_stmtr# new_leaf1 old_leaf1s
r fixup_simple_stmtr/ G s.
$I$677 $ 9
" " E # KKMMMDNB''HD$xj11H
XYY
' &x0 ioo / /000
XYY
' 8$$$!!$-a0I"1%.q1I 'Ir c | j rA| j d j t j k r#| j d d S d S d S )N)r r r NEWLINEr )r s r remove_trailing_newliner3 _ sQ } # r*/5=@@
b """""# #@@r c # K | j D ]}|j t j k r nt d t t
|j D ]\ }}|j t j k r|j r}|j d }|j t j k r[|j rT|j d }t |t r2|j dk r't ||| t | |||fV d S )NzNo class suite!r r
)r r r r
r listr r r r r r r/ r3 )r! r r" simple_noder left_nodes r
find_metasr8 d s ! , , 9
" " E # *+++ y 7 788 1 1; t / / /K4H /#,Q/I~ //I4F/ % .q 1 i.. 1!?::%dA{;;;+K888 K0000 1 1r c p | j ddd }|r,| }|j t j k rn|,|ru| }t |t r%|j t j k r|j rd|_ dS | |j ddd |sdS dS )z 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
Nr1 )
r popr r INDENTr r DEDENTr( extend)r
kidsr s r fixup_indentr@ { s >$$B$ D
xxzz 9 $ $ -xxzz dD ! ! -di5<&?&?{
! F KK
ddd+ , , , - - - - -r c e Zd ZdZdZd ZdS ) FixMetaclassTz
classdef