o ,]g@sdZddlmZddlmZmZmZmZmZddl m Z m Z m Z dZ dZdZdZd d Zefd d Zefd dZGdddejZdS)zg Fixer for Python 3 function parameter syntax This fixer is rather sensitive to incorrect py3k syntax. ) fixer_base)tokenStringNewlineCommaName) indentationsuitify DoubleStarz=%(name)s = %(kwargs)s['%(name)s']; del %(kwargs)s['%(name)s']z'if '%(name)s' in %(kwargs)s: %(assign)szelse: %(name)s = %(default)s _3to2kwargsccs|djtjkrt|dksJd}t|}||krb||}|j}|jtjkr-|d7}q|dur9|jtjkr9dS|j}|j}|durQ|jtj krQ|j}|d7}nd}||fV|d7}||ksdSdS)z Generator that yields tuples of (name, default_value) for each parameter in the list If no default is given, then it is default_value is None (not Leaf(token.NAME, 'None')) rN) typerSTARlen prev_siblingNAME DOUBLESTARvalue next_siblingEQUAL) raw_paramscurr_idxmax_idx curr_item prev_itemnamenxt default_valuer@/usr/lib/python3/dist-packages/libpasteurize/fixes/fix_kwargs.py gen_paramss*    r!cCs~|djtjks J|djtjkr%|d|d|dd}n|dd}|D]}|jtjkr:|q-dSdS)z Removes all keyword-only args from the params list and a bare star, if any. Does not add the kwargs dict if needed. Returns True if more action is needed, False if not (more action is needed if no kwargs dict exists) rr r NFT)rrrCOMMAremover)rkwargs_default kw_paramsparamrrr remove_params+s     r(cCsd}d}|ddD]/}|jtjkrq |jtjkr|sd}q |jtjkr1|r1|r-|jSdS|jtjkr9d}q |r>|SdS)z Returns string with the name of the kwargs dict if the params after the first star need fixing Otherwise returns empty string Fr NT)rrr#rrr)rr% found_kwargs needs_fixtrrr needs_fixingAs   r-c@seZdZdZdZddZdS) FixKwargsz^funcdef< 'def' NAME parameters< '(' arglist=typedargslist< params=any* > ')' > ':' suite=any >c Cs|d}t|D]\}}|jtjkr||d}nqdSt|}|s%dSt||jd}|jd}t|} t|D]Q\} } | durZ| dt | dt t | |d| dq;| dt | dt t | | d| d| dt | dt tt | |d| |d| dq;| |_d|jd_t|} | r|d } t| jd kr| jd jtjkr| t| td d| t|dSdS) Nparamsr )rkwargs)prefix)rdefault)assignrr2r)arglistr ) enumeraterrrr-r childrenrr! insert_childrr_assign_template_else_template _if_templater3r(rr# append_childrr r)selfnoderesultsparams_rawlistiitem new_kwargssuite first_stmtidentrrmust_add_kwargsr6rrr transform`s@    ,   zFixKwargs.transformN)__name__ __module__ __qualname__ run_orderPATTERNrKrrrr r.Zs r.N)__doc__lib2to3rlib2to3.fixer_utilrrrrrlibfuturize.fixer_utilrr r r<r>r=_kwargs_default_namer!r(r-BaseFixr.rrrr s