U $FZhj@sddlZddlZz ddlZWnek r8edYnXddlZddlZddlmZm Z m Z m Z m Z m Z e egefe ee e ee efdddZGdddejZed ejejd d d Zed krejddZejddddddejddddddeZeejZeejZesPededejdedeszededejdede 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 resultsiru/home/aprabhat/apps/x.techxrdev.in/venv/lib/python3.8/site-packages/../../../bin/fixup_firestore_admin_v1_keywords.py partitionsrc @seZdZUdZeeed<dddddddd ddddd ddddddd d d dddd ddddddZeeeefed<e j e j e j dddZ dS)firestore_adminCallTransformer)retrytimeoutmetadata CTRL_PARAMS)namecollection_ids namespace_ids)parentbackup_schedule)rdatabase database_id)rindex)rZ user_credsZ user_creds_id)r)retag)rrZoutput_uri_prefixrZ snapshot_time)rrZinput_uri_prefixr)rfilter)r)rZ show_deleted)rr!Z page_sizeZ page_token)rrbackupZencryption_config)r update_mask)rr#)fieldr#)Zbulk_delete_documentsZcreate_backup_scheduleZcreate_databaseZ create_indexZcreate_user_credsZ delete_backupZdelete_backup_scheduleZdelete_databaseZ delete_indexZdelete_user_credsZdisable_user_credsZenable_user_credsZexport_documentsZ get_backupZget_backup_scheduleZ get_database get_field get_indexZget_user_credsZimport_documentsZ list_backupsZlist_backup_schedulesZlist_databasesZ list_fieldsZ list_indexesZlist_user_credsZreset_user_passwordZrestore_databaseZupdate_backup_scheduleZupdate_databaseZ update_fieldMETHOD_TO_PARAMS)originalupdatedr c sz|jjj}j|}Wnttfk r4|YSXtdd|j\}}tdd|Dr`|Stfdd|\}}|dt ||t |d}}| ddt |j Dt jt ddt |||Dt d d } |j| g|d S) NcSs t|j SN)boolkeywordarrr[z;firestore_adminCallTransformer.leave_Call..css|]}|jjdkVqdS)requestN)r,value).0krrr \sz.cs|jjjkSr*)r,r2rr-selfrrr/ar0css*|]"\}}tj|jtj|ddVqdS)r2r2r,N)cstArgr2Name)r3r.Zctrlrrrr5fsc Ss2g|]*\}}ttd|tj|jdqS)z'{}'r8)r:Z DictElementZ SimpleStringformatZElementr2)r3rargrrr js  z=firestore_adminCallTransformer.leave_Call..r1r9)args)funcattrr2r'AttributeErrorKeyErrorrr@anylenextendziprr:r;rr<Z with_changes) r7r(r)keyZ kword_paramsr@kwargsZ ctrl_kwargsZ ctrl_argsZ request_argrr6r leave_CallQs2   "      z)firestore_adminCallTransformer.leave_CallN) __name__ __module__ __qualname__rrstr__annotations__r'rr:CallZCSTNoderKrrrrr-sD "r) transformer)in_dirout_dirc Csddt|D}|D]x}t|d}|}W5QRXt|}||}|||} | j j dddt| d}| |j W5QRXqdS)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)r3root_filesfrrrr5s zfix_files..rT)parentsexist_okwN) rUwalkopenreadr:Z parse_modulevisitjoinpath relative_tormkdirwritecode) rSrTrRZ pyfile_genZfpathr^srctreer)Z updated_pathrrr fix_files{s     rn__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)!argparserUZlibcstr: ImportErrorrXsystypingrrrrrrr+rZCSTTransformerrrYrnrLArgumentParserparser add_argument parse_argsr@rqruis_dirprintstderrexitlistdirrrrrst    R #