o Fh2 @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 resultsira/var/www/html/loop/nvenv/lib/python3.10/site-packages/../../../bin/fixup_firestore_v1_keywords.py partitionsrc@seZdZUdZeeed<iddddddd d d d d ddddddddddddddddddd d!d"Zeeeefed#<d$e j d%e j d&e j fd'd(Z d)S)*firestoreCallTransformer)retrytimeoutmetadata CTRL_PARAMSbatch_get_documents)database documentsmask transactionnew_transaction read_time batch_write)rwriteslabelsbegin_transaction)roptionscommit)rr rcreate_document)parent collection_iddocument document_idrdelete_document)namecurrent_document get_document)r+rrrlist_collection_ids)r& page_size page_tokenrlist_documents) r&r'r/r0order_byrrr show_missinglisten)r add_target remove_targetr!partition_query)r&structured_querypartition_countr0r/rrollback)rrrun_aggregation_query)r&structured_aggregation_queryrrrexplain_options run_query)r&r8rrrr=update_document)r( update_maskrr,write)r stream_idr stream_tokenr!METHOD_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)boolkeywordarrrLs z5firestoreCallTransformer.leave_Call..css|] }|jjdkVqdS)requestN)rIvalue).0krrr Msz6firestoreCallTransformer.leave_Call..cs|jjjvSrG)rIrNrrJselfrrrLRscss,|]\}}tj|jtj|ddVqdS)rNrNrIN)cstArgrNName)rOrKctrlrrrrQWs c Ss2g|]\}}ttd|tj|jdqS)z'{}'rT)rV DictElement SimpleStringformatElementrN)rOr+argrrr [s z7firestoreCallTransformer.leave_Call..rMrU)args)funcattrrNrDAttributeErrorKeyErrorrr`anylenextendziprrVrWrrX with_changes) rSrErFkey kword_paramsr`kwargs ctrl_kwargs ctrl_args request_argrrRr leave_CallBs4  "      z#firestoreCallTransformer.leave_CallN) __name__ __module__ __qualname__rrstr__annotations__rDrrVCallCSTNoderprrrrr-sH       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)rOroot_filesfrrrrQxszfix_files..rNT)parentsexist_okw) r{walkopenreadrV parse_modulevisitjoinpath relative_tor&mkdirrAcode) ryrzrx pyfile_genfpathrsrctreerF updated_pathrrr fix_filesls       r__main__aFix up source that uses the firestore 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)!argparser{libcstrV ImportErrorr~systypingrrrrrrrHrCSTTransformerrrrrqArgumentParserparser add_argument parse_argsr`rris_dirprintstderrexitlistdirrrrrs     C #