o DDi@s|UddlZddlZddlmZmZddlZddlmZm Z dddddddZ ee ee e ffe d<Gd d d e ZdS) N)DictOptional)CodemodContextVisitorBasedCodemodCommand)parameter_countargument_count)blackyapfpresets_per_formatterc seZdZUedZeed<   ddedede e de e d df fd d Z e d e jd dfd dZdejdejd ejfddZdejdejd ejfddZZS)AddTrailingCommasaA Codemod that adds trailing commas to arguments in function headers and function calls. The idea is that both the black and yapf autoformatters will tend to split headers and function calls so that there is one parameter / argument per line if there is a trailing comma: - Black will always separate them by line - Yapf appears to do so whenever there are at least two arguments Applying this codemod (and then an autoformatter) may make it easier to read function definitions and calls DESCRIPTIONr Ncontext formatterrr returncs\t|t|}|durtd|ddt|p#|d|_|p*|d|_dS)NzUnknown formatter z. Presets exist for z, rr ) super__init__r get ValueErrorjoinkeysrr )selfrrrr presets __class__d/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codemod/commands/add_trailing_commas.pyr,s    zAddTrailingCommas.__init__ arg_parsercCsF|jddddtdd|jddd d tdd|jd d d dtdddS)Nz --formatterr FORMATTERz(Formatter to target (e.g. yapf or black)r )destmetavarhelptypedefaultz--paramter-countrPARAMETER_COUNTz9Minimal number of parameters for us to add trailing commaz--argument-countr ARGUMENT_COUNTz8Minimal number of arguments for us to add trailing comma) add_argumentstrint)rrrradd_args=s0 zAddTrailingCommas.add_args original_node updated_nodecCsx|jdupt|j|jkpt|jdko|jdjjdv}|r!|S|jd}|jg|jdd|jtdRdS)Nrr>clsrcomma)params)rlenr1namevalue with_changescstComma)rr+r,skip last_paramrrrleave_ParametersXs    z"AddTrailingCommas.leave_ParameterscCsJt|j|jkr |S|jd}|jg|jdd|jtdRdS)Nr.r/)args)r2r;r r5r6r7)rr+r,last_argrrr leave_Callqs  zAddTrailingCommas.leave_Call)r NN)__name__ __module__ __qualname__textwrapdedentrr(__annotations__rrr)r staticmethodargparseArgumentParserr*r6 Parametersr:Callr= __classcell__rrrrr sD   r )rErAtypingrrlibcstr6libcst.codemodrrr r(r)rCr rrrrs