a !f<@sddlZddlZz ddlZWney6edYn0ddlZddlZddlmZm Z m Z m Z m Z m Z e egefe ee e ee efdddZGdddejZed ejejd d d Zed krejddZejddddddejddddddeZeejZeejZesNededejdedesxededejdede erededejdedeeedS) Nz@Run `python -m pip install "libcst >= 0.2.5"` to install libcst.)AnyCallableDictListSequenceTuple) predicateiteratorreturncCs8ggf}|D]}|t|||q |d|dfS)z!A stable, out-of-place partition.r)intappend)rr resultsiro/var/www/html/python-backend/venv/lib/python3.9/site-packages/../../../bin/fixup_firestore_admin_v1_keywords.py partitionsrc @sheZdZUdZeeed<ddddddddddd d d Zeeeefed <e j e j e j d ddZ dS)firestore_adminCallTransformer)retrytimeoutmetadata CTRL_PARAMS)parentindex)name)rcollection_idsZoutput_uri_prefix)rrZinput_uri_prefix)r)rfilterZ page_sizeZ page_token)Zdatabase update_mask)fieldr) Z create_indexZ delete_indexZexport_documentsZ get_database get_fieldZ get_indexZimport_documentsZlist_databasesZ list_fieldsZ list_indexesZupdate_databaseZ update_fieldMETHOD_TO_PARAMS)originalupdatedr c sz|jjj}j|}Wnttfy2|YS0tdd|j\}}tdd|Dr^|Stfdd|\}}|dt ||t |d}}| ddt |j Dt jt ddt |||Dt d d } |j| g|d S) NcSs t|j SN)boolkeywordarrrHz;firestore_adminCallTransformer.leave_Call..css|]}|jjdkVqdS)requestN)r%value).0krrr Ir)z.cs|jjjvSr#)r%r+rr&selfrrr(Nr)css*|]"\}}tj|jtj|ddVqdS)r+r+r%N)cstArgr+Name)r,r'Zctrlrrrr.Ssc Ss2g|]*\}}ttd|tj|jdqS)z'{}'r1)r3Z DictElementZ SimpleStringformatZElementr+)r,rargrrr Ws  z=firestore_adminCallTransformer.leave_Call..r*r2)args)funcattrr+r AttributeErrorKeyErrorrr9anylenextendziprr3r4rr5Z with_changes) r0r!r"keyZ kword_paramsr9kwargsZ ctrl_kwargsZ ctrl_argsZ request_argrr/r leave_Call>s2   "      z)firestore_adminCallTransformer.leave_CallN) __name__ __module__ __qualname__rrstr__annotations__r rr3CallZCSTNoderDrrrrr-s r) transformer)in_dirout_dirc Csddt|D}|D]}t|d}|}Wdn1sD0Yt|}||}|||} | j j dddt| d}| |j Wdq1s0YqdS)zDuplicate the input dir to the output dir, fixing file method calls. Preconditions: * in_dir is a real directory * out_dir is a real, empty directory cssF|]>\}}}|D].}tj|ddkrttj||VqqdS)r z.pyN)ospathsplitextpathlibPathjoin)r,root_filesfrrrr.tszfix_files..rNT)parentsexist_okw) rNwalkopenreadr3Z parse_modulevisitjoinpath relative_tormkdirwritecode) rLrMrKZ pyfile_genZfpathrWsrctreer"Z updated_pathrrr fix_fileshs  &   rg__main__aFix up source that uses the firestore_admin client library. The existing sources are NOT overwritten but are copied to output_dir with changes made. Note: This tool operates at a best-effort level at converting positional parameters in client method calls to keyword based parameters. Cases where it WILL FAIL include A) * or ** expansion in a method call. B) Calls via function or method alias (includes free function calls) C) Indirect or dispatched calls (e.g. the method is looked up dynamically) These all constitute false negatives. The tool will also detect false positives when an API method shares a name with another method. ) descriptionz-dz--input-directoryT input_dirz6the input directory to walk for python files to fix up)requireddesthelpz-oz--output-directory output_dirz5the directory to output files fixed via un-flatteningzinput directory 'z&' does not exist or is not a directory)filezoutput directory 'z' is not empty)!argparserNZlibcstr3 ImportErrorrQsystypingrrrrrrr$rZCSTTransformerrrRrgrEArgumentParserparser add_argument parse_argsr9rjrnis_dirprintstderrexitlistdirrrrrst     ? #