a !f"@sddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z eege fe ee e ee efdddZGdddejZedejejd d d Zed krejd dZejddddddejddddddeZeejZeejZes.ededejdedesXededejdedeerededejdedeeedS)N)AnyCallableDictListSequenceTuple) predicateiteratorreturncCs8ggf}|D]}|t|||q |d|dfS)z!A stable, out-of-place partition.r)intappend)rr resultsirf/var/www/html/python-backend/venv/lib/python3.9/site-packages/../../../bin/fixup_pubsub_v1_keywords.py partitionsrc'@seZdZUdZeeed<ddddddd d d d d d dd d d dddddddddddd dddddddddd d!&Zeeeefed"<e j e j e j d#d$d%Z d&S)'pubsubCallTransformer)retrytimeoutmetadata CTRL_PARAMS) subscriptionack_ids)nameschema)parentrZ schema_id)rrlabels)rtopic push_configZbigquery_configZcloud_storage_configack_deadline_secondsZretain_acked_messagesmessage_retention_durationrZenable_message_orderingZexpiration_policyfilterZdead_letter_policyZ retry_policyZdetachedZenable_exactly_once_deliveryZ topic_message_retention_durationstate) rrZmessage_storage_policyZ kms_key_nameZschema_settingsZ satisfies_pzsr!r#Zingestion_data_source_settings)r)rZ revision_id)snapshot)r)r)rview)rr% page_size page_token)rr%r&r')projectr&r')rr&r')rrr )rr)rmessages)rZ max_messagesZreturn_immediately)rtimer$)rZstream_ack_deadline_secondsrZmodify_deadline_secondsZmodify_deadline_ack_idsZ client_idZmax_outstanding_messagesZmax_outstanding_bytes)r$ update_mask)rr+)rr+)rrrmessageencoding)rr)resourceoptions)r.policy)r.Z permissions)&Z acknowledgeZ commit_schemaZ create_schemaZcreate_snapshotZcreate_subscriptionZ create_topicZ delete_schemaZdelete_schema_revisionZdelete_snapshotZdelete_subscriptionZ delete_topicZdetach_subscriptionZ get_schemaZ get_snapshotZget_subscriptionZ get_topicZlist_schema_revisionsZ list_schemasZlist_snapshotsZlist_subscriptionsZ list_topicsZlist_topic_snapshotsZlist_topic_subscriptionsZmodify_ack_deadlineZmodify_push_configpublishpullZrollback_schemaseekZstreaming_pullZupdate_snapshotZupdate_subscriptionZ update_topicZvalidate_messageZvalidate_schemaZget_iam_policyZset_iam_policyZtest_iam_permissionsMETHOD_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)boolkeywordarrr\z2pubsubCallTransformer.leave_Call..css|]}|jjdkVqdS)requestN)r9value).0krrr ]r=z3pubsubCallTransformer.leave_Call..cs|jjjvSr7)r9r?rr:selfrrr<br=css*|]"\}}tj|jtj|ddVqdS)r?r?r9N)cstArgr?Name)r@r;ZctrlrrrrBgsc Ss2g|]*\}}ttd|tj|jdqS)z'{}'rE)rGZ DictElementZ SimpleStringformatZElementr?)r@rargrrr ks  z4pubsubCallTransformer.leave_Call..r>rF)args)funcattrr?r4AttributeErrorKeyErrorrrManylenextendziprrGrHrrIZ with_changes) rDr5r6keyZ kword_paramsrMkwargsZ ctrl_kwargsZ ctrl_argsZ request_argrrCr leave_CallRs2   "      z pubsubCallTransformer.leave_CallN) __name__ __module__ __qualname__rrstr__annotations__r4rrGCallZCSTNoderXrrrrr'sR )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_filesfrrrrBszfix_files..rNT)parentsexist_okw) rbwalkopenreadrGZ parse_modulevisitjoinpath relative_tormkdirwritecode) r`rar_Z pyfile_genZfpathrksrctreer6Z updated_pathrrr fix_files|s  &   r{__main__aFix up source that uses the pubsub 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) argparserbZlibcstrGresystypingrrrrrrr8rZCSTTransformerrrfr{rYArgumentParserparser add_argument parse_argsrMr~ris_dirprintstderrexitlistdirrrrrsn   Y #