dZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z d Z Gd d e jZ y ) zUtilities for Recommendation.)absolute_import)division)unicode_literals)encoding) list_pager)base) flag_utilscBtj|}t|S)zCreates Client. Args: release_track: release_track value, can be ALPHA, BETA, GA Returns: The versioned client. )r GetApiVersionRecommendation) release_track api_versions 8lib/googlecloudsdk/api_lib/recommender/recommendation.py CreateClientrs((7+  $$cTeZdZdZfdZdZdZd dZdZdZ dZ d Z d Z xZ S) r z,Base Recommendation client for all versions.cbtt| ||jj|_yN)superr __init___client/projects_locations_recommenders_recommendations_service)selfr __class__s rrzRecommendation.__init__+s# .$(5LLPPDMrc`dj|}|j||}|r8tj||j|jd}||_d|t j|j|z|i}|jdj|di|S)z-Creates MarkRequest with the specified state.zMarkRecommendation{}Request)etagT) sort_itemsnamezDRecommenderProjectsLocationsRecommendersRecommendationsMark{}Request) format_GetVersionedMessagerDictToAdditionalPropertyMessageStateMetadataValue stateMetadatar ToCamelCase_message_prefix _GetMessage) rrstatestate_metadatar request_name mark_requestmetadatakwargss r_CreateMarkRequestz!Recommendation._CreateMarkRequest/s177>L:4,,\:EL99   # #L 1 D Dh$,l   t33lBC  F 4  N   "  ""rcp|jj|}|jj|S)zGets a Recommendation. Args: name: str, the name of the recommendation being retrieved. Returns: The Recommendation message. )r) _messagesARecommenderProjectsLocationsRecommendersRecommendationsGetRequestrGet)rrrequests rr3zRecommendation.GetIs6nn^^ _G ==  W %%rc|jj|}tj|j|d||dS)aList Recommendations. Args: parent_name: str, the name of the parent. page_size: int, The number of items to retrieve per request. limit: int, The maximum number of records to yield. Returns: The Recommendation messages. )parentpageSizerecommendations)batch_size_attribute batch_sizelimitfield)r1BRecommenderProjectsLocationsRecommendersRecommendationsListRequestr YieldFromListr)r parent_name page_sizer;r4s rListzRecommendation.ListWsJnn__`G  # # '  !!rc`|j|dd|}|jj|S)a,Mark a recommendation's state as ACTIVE. Args: name: str, the name of the recommendation being updated. etag: Fingerprint of the Recommendation. Provides optimistic locking when updating states. Returns: The result recommendations after being marked as active ActiveN)r/r MarkActiverrrr4s rrDzRecommendation.MarkActivems/%%dHdDAG == # #G ,,rc`|j|dd|}|jj|S)a2Mark a recommendation's state as DISMISSED. Args: name: str, the name of the recommendation being updated. etag: Fingerprint of the Recommendation. Provides optimistic locking when updating states. Returns: The result recommendations after being marked as dismissed DismissedN)r/r MarkDismissedrEs rrHzRecommendation.MarkDismissed{s/%%dKtDG == & &w //rc`|j|d||}|jj|S)aMark a recommendation's state as CLAIMED. Args: name: str, the name of the recommendation being updated. state_metadata: A map of metadata for the state, provided by user or automations systems. etag: Fingerprint of the Recommendation. Provides optimistic locking when updating states. Returns: The result recommendations after being marked as accepted Claimed)r/r MarkClaimedrrr*rr4s rrKzRecommendation.MarkClaimeds/%%dI~tLG == $ $W --rc`|j|d||}|jj|S)aMark a recommendation's state as SUCCEEDED. Args: name: str, the name of the recommendation being updated. state_metadata: A map of metadata for the state, provided by user or automations systems. etag: Fingerprint of the Recommendation. Provides optimistic locking when updating states. Returns: The result recommendations after being marked as accepted Succeeded)r/r MarkSucceededrLs rrOzRecommendation.MarkSucceededs/%%dKNG == & &w //rc`|j|d||}|jj|S)aMark a recommendation's state as FAILED. Args: name: str, the name of the recommendation being updated. state_metadata: A map of metadata for the state, provided by user or automations systems. etag: Fingerprint of the Recommendation. Provides optimistic locking when updating states. Returns: The result recommendations after being marked as accepted Failed)r/r MarkFailedrLs rrRzRecommendation.MarkFaileds/%%dHndKG == # #G ,,rr)__name__ __module__ __qualname____doc__rr/r3rArDrHrKrOrR __classcell__)rs@rr r (s44Q"4 &!, - 0. 0 -rr N)rV __future__rrrapitools.base.pyrr"googlecloudsdk.api_lib.recommenderrr r ClientBaser r rrr\s6$&'%'39 %O-T__O-r