dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z d Z e jjd e jjd iZd ZddZddZddZddZGddej,Zy )z)Utility functions for GCE OS Config APIs.)absolute_import)division)unicode_literals)apis)waiter)base) resourcesosconfigv1betav1ct|SN)_API_CLIENT_VERSION_MAP) release_tracks 5lib/googlecloudsdk/api_lib/compute/os_config/utils.py GetApiVersionr"s  //NcLtjt|xs t|Sr)rGetClientClass_API_CLIENT_NAMErrapi_version_overrides rrr&s(  ,-m, ..rcLtjt|xs t|Sr)rGetClientInstancerrrs rrr,(  ,-m, ..rcLtjt|xs t|Sr)rGetMessagesModulerrrs rGetClientMessagesr2rrctjj}|jt|xs t ||Sr)r REGISTRYCloneRegisterApiByNamerr)rrregistrys r GetRegistryr$8s>    % % '( ,-m,. /rc(eZdZdZdZdZdZdZy)Pollerz+Poller for synchronous patch job execution.cv||_||_|jjjj|jjjj |jjjj |jjjjg|_y)zInitializes poller for patch job execution. Args: client: API client of the OsConfig service. messages: API messages of the OsConfig service. N) clientmessagesPatchJobStateValueValuesEnum SUCCEEDEDCOMPLETED_WITH_ERRORS TIMED_OUTCANCELEDpatch_job_terminal_states)selfr(r)s r__init__zPoller.__init__CsDKDM 33== 33II 33== 33<< &D"rc2|j|jvSz Overrides.)stater0r1 patch_jobs rIsDonez Poller.IsDoneSs ??d<< <rMs|0&',.()H$ 0. . . V # #r