B A[`%@sdZddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZddZddZd d Zd d Zd dZddZGdddejZdS)aFixer for __metaclass__ = X -> (future.utils.with_metaclass(X)) methods. The various forms of classef (inherits nothing, inherits once, inherints 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) NamesymsNodeLeaf touch_importCallStringComma parenthesizecCsxxr|jD]h}|jtjkr t|S|jtjkr|jr|jd}|jtjkr|jr|jd}t|tr|j dkrdSqWdS)z we have to check the cls_node without changing it. There are two possiblities: 1) clsdef => suite => simple_stmt => expr_stmt => Leaf('__meta') 2) clsdef => simple_stmt => expr_stmt => Leaf('__meta') r __metaclass__TF) childrentypersuite has_metaclass simple_stmt expr_stmt isinstancervalue)parentnode expr_nodeZ left_siderP/opt/alt/python37/lib/python3.7/site-packages/libfuturize/fixes/fix_metaclass.pyr&s      rcCsx|jD]}|jtjkrdSqWx,t|jD]\}}|jtjkr,Pq,Wtdttjg}x:|j|ddr|j|d}| | | q\W| ||}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 ':'!) rrrr enumeraterCOLON ValueErrorr append_childcloneremove)cls_noderir move_noderrrfixup_parse_tree9s      r%c Csx(t|jD]\}}|jtjkr Pq WdS|ttjg}ttj |g}x2|j|dr~|j|}| | |qNW| |||jdjd}|jdjd} | 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 efter the semi-colon into its own simple_stmt node Nr)rrrrSEMIr!rrrrrr insert_childprefix) rr#Z stmt_nodeZsemi_indrZnew_exprZnew_stmtr$Z new_leaf1Z old_leaf1rrrfixup_simple_stmtSs     r)cCs*|jr&|jdjtjkr&|jddS)N)rrrNEWLINEr!)rrrrremove_trailing_newlineksr,ccsx$|jD]}|jtjkrPqWtdxtt|jD]t\}}|jtjkr6|jr6|jd}|jtjkr6|jr6|jd}t |t r6|j dkr6t |||t ||||fVq6WdS)NzNo class suite!rr )rrrrrlistrrrrrrr)r,)r"rr#Z simple_noderZ left_noderrr find_metasps       r.cCs|jddd}x|r.|}|jtjkrPqWxL|r||}t|trd|jtjkrd|jr`d|_dS| |jdddq2WdS)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 Nr*) rpoprrINDENTrrDEDENTr(extend)rZkidsrrrr fixup_indents r4c@seZdZdZdZddZdS) FixMetaclassTz classdef c Cst|s dSt|d}x"t|D]\}}}|}|q"W|jdj}t|jdkr|jdjtjkrt|jd}n(|jd } t tj| g}| d|nt|jdkrt tjg}| d|nZt|jdkrt tjg}| dt tjd| d|| dt tjdntd |jdjd} d | _| j} td d ||jdjd } d | _| g} |jrt|jdkr|jd }d|_nVt| }d|_ttdtdt|tt tjt tjdt tjdgddgdd}| t|g|ttd |jd| t||js\|t |d}| |_| || t tj!dnbt|jdkr|jdjtj"kr|jdjtj#krt |d}| d|| dt tj!ddS)Nr)(zUnexpected class definition metaclassz future.utilswith_metaclassr/r rz 'NewBase'{})r(pass r*)$rr%r.r!rrlenrarglistr rZ set_childr'rrRPARLPARrrr(rr r rr r ZatomLBRACERBRACEr3replacer4rr+r1r2)selfrresultsZlast_metaclassrr#stmtZ text_typerFrZmeta_txtZorig_meta_prefixr= argumentsbasebasesZ pass_leafrrr transforms~             zFixMetaclass.transformN)__name__ __module__ __qualname__Z BM_compatibleZPATTERNrRrrrrr5sr5N)__doc__Zlib2to3rZlib2to3.pygramrZlib2to3.fixer_utilrrrrrr r r r rr%r)r,r.r4ZBaseFixr5rrrrs  ,