o Fh @sddlZddlZzddlZWn eyedwddlZddlZddlmZm Z m Z m Z m Z m Z de egefde ede e ee effddZGd d d ejZed d ejd ejfddZedkrejddZejddddddejddddddeZeejZeejZesededejded esed!edejded e ered!ed"ejded eeedSdS)#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 resultsirg/var/www/html/loop/nvenv/lib/python3.10/site-packages/../../../bin/fixup_firestore_admin_v1_keywords.py partitionsrc@seZdZUdZeeed<iddddddd d d d d d dd dddd dd dd dd dd dddddddddddddd d!Zeeeefed"<d#e j d$e j d%e j fd&d'Z d(S))firestore_adminCallTransformer)retrytimeoutmetadata CTRL_PARAMScreate_backup_schedule)parentbackup_schedulecreate_database)rdatabase database_id create_index)rindex delete_backup)namedelete_backup_scheduledelete_database)r!etag delete_indexexport_documents)r!collection_idsoutput_uri_prefix namespace_ids snapshot_time get_backupget_backup_schedule get_database get_field get_indeximport_documents)r!r'input_uri_prefixr) list_backups)rlist_backup_scheduleslist_databases)rfilter page_size page_token)rrbackup)r update_mask)rr9)fieldr9) list_fields list_indexesrestore_databaseupdate_backup_scheduleupdate_database update_fieldMETHOD_TO_PARAMSoriginalupdatedr c sz |jjj}j|}Wn ttfy|YSwtdd|j\}}tdd|Dr/|Stfdd|\}}|dt ||t |d}}| ddt |j Dt jt ddt |||Dt d d } |j| g|d S) NcSs t|j SN)boolkeywordarrrSs z;firestore_adminCallTransformer.leave_Call..css|] }|jjdkVqdS)requestN)rFvalue).0krrr Tsz.cs|jjjvSrD)rFrKrrGselfrrrIYscss,|]\}}tj|jtj|ddVqdS)rKrKrFN)cstArgrKName)rLrHctrlrrrrN^s c Ss2g|]\}}ttd|tj|jdqS)z'{}'rQ)rS DictElement SimpleStringformatElementrK)rLr!argrrr bs z=firestore_adminCallTransformer.leave_Call..rJrR)args)funcattrrKrAAttributeErrorKeyErrorrr]anylenextendziprrSrTrrU with_changes) rPrBrCkey kword_paramsr]kwargs ctrl_kwargs ctrl_args request_argrrOr leave_CallIs4  "      z)firestore_adminCallTransformer.leave_CallN) __name__ __module__ __qualname__rrstr__annotations__rArrSCallCSTNodermrrrrr-sZ       r) transformerin_dirout_dirc Csddt|D}|D]P}t|d }|}Wdn1s"wYt|}||}|||} | j j dddt| d}| |j Wdn1sWwYq dS)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 cssH|]\}}}|D]}tj|ddkr ttj||Vq qdS)r z.pyN)ospathsplitextpathlibPathjoin)rLroot_filesfrrrrNszfix_files..rNT)parentsexist_okw) rxwalkopenreadrS parse_modulevisitjoinpath relative_tormkdirwritecode) rvrwru pyfile_genfpathrsrctreerC updated_pathrrr fix_filesss       r__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)!argparserxlibcstrS ImportErrorr{systypingrrrrrrrErCSTTransformerrr|rrnArgumentParserparser add_argument parse_argsr]rris_dirprintstderrexitlistdirrrrrs     J #