*dZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z d Z d Z d Z dd ZdZdZdZy )zHCommands for interacting with Immersive Stream for XR service instances.)absolute_import)division)unicode_literals)api_util) exceptions)log) propertiesc&dj||S)Nzprojects/{}/locations/{})format)projectlocations ;lib/googlecloudsdk/api_lib/immersive_stream/xr/instances.pyProjectLocationrs # * *7H ==c\|<|jj}|Dcic]}|j|jc}ni|r7t fd|Dr t j jdy|}nx|r>t fd|Dr t j jdyd|D}n8|r6t fd|Dr t j jdy|}tj|}|jj} D]} | d } t| d } | jd d } |j| | | }| r&t| d|_t| d|_| jj#|jjj%| || jD]S}|r6|jj&dk(rj)|jd;|j|j<U|jj}t+j-D]E\}}|jj#|jjj%||G|Scc}w)a>Generates the target location configs. Args: release_track: ALPHA or GA release track add_region_configs: List of region config dicts of the form: [{'region': region1, 'capacity': capacity1, 'enable_autoscaling': enable_autoscaling1, 'autoscaling_buffer': autoscaling_buffer1, 'autoscaling_min_capacity': autoscaling_min_capacity1}] that specifies the regions to add to the service instance update_region_configs: List of region config dicts of the form: [{'region': region1, 'capacity': capacity1}] that specifies the regions to update to the service instance remove_regions: List of regions to remove current_instance: instance object - current state of the service instance before update Returns: A LocationConfigsValue, with entries sorted by location Nc3,K|] }|dv ywregionN.0 region_configlocation_configss r z0GenerateTargetLocationConfigs..Ds$ /M h#33/zOnly new regions can be added.c3&K|]}|v yw)Nr)rrrs rrz0GenerateTargetLocationConfigs..Ms Gf6) )sz%Only existing regions can be removed.c3(K|] }|ddd yw)rF)rcapacityenable_autoscalingNr)rrs rrz0GenerateTargetLocationConfigs..Rs$FqF$sc3,K|] }|dv ywrrrs rrz0GenerateTargetLocationConfigs..Xs$ 2M h'772rz%Only existing regions can be updated.rrrF)r renableAutoscalingautoscaling_bufferautoscaling_min_capacity)keyvaluer)locationConfigsadditionalPropertiesr$r%anyrstatusPrintr GetMessagesStreamInstanceLocationConfigsValueintgetLocationConfigautoscalingBufferautoscalingMinCapacityappendAdditionalPropertyrpopsorteditems) release_trackadd_region_configsupdate_region_configsremove_regionscurrent_instanceadditonal_propertieslocation_configregion_configs_diffmessageslocation_configs_diffrrrrtarget_location_configsr$rs @rGenerateTargetLocationConfigsrCs6!+;;PP 43O _2223   /  jj78 , G GG jj>? $   2  jj>? 0  ! !- 0("11FFH*m 8 $F=,-H&**+?G--,.O *- , -+o'03 2 30o,..5544GGo H #+0/CCo///88A=?..5/>.C.C**+ D%33HHJ%%5%;%;%=>c?007744GG? H ? ! gsJ)ctj|}tj|}|j|}|j |j |S)zGet resource details of an Immersive Stream for XR service instance. Args: release_track: ALPHA or GA release track instance_relative_name: string - canonical resource name of the instance Returns: A service instance resource object. )name)r GetClientr+'ProjectsLocationsStreamInstancesServiceGet0StreamProjectsLocationsStreamInstancesGetRequest)r8instance_relative_nameclientr@services rrHrHs\   m ,&  ! !- 0(  : :6 B' ??%@ rNc tj|} tj|} | j|} | j || ||||} |r| j |} | | _| j| }|j| jttjjjj|| |S)anCreate a new Immersive Stream for XR service instance. Args: release_track: ALPHA or GA release track instance_name: string - name of the service instance content: string - resource path of the content resource that is served by the instance location: string - location where the resource will be created version: string - content build version tag target_location_configs: A LocationConfigsValue proto message represents the target location configs to achieve fallback_url: string - A url to redirect users to when the instance is unable to provide the streaming experience mode: string - The rendering mode supported by the service instance gpu_class: string - The class of GPU used by this instance Returns: An Operation object which can be used to check on the progress of the service instance creation. contentVersionTag)contentcontentBuildVersionrEr&modegpuClass fallbackUri)parentstreamInstancestreamInstanceId)rrFr+ BuildVersionr, StreamConfig streamConfigrGCreate3StreamProjectsLocationsStreamInstancesCreateRequestrr VALUEScorer rH)r8 instance_namerPr versionrB fallback_urlrR gpu_classrKr@ build_versioninstance stream_configrLs rr\r\s>   m ,&  ! !- 0(''''B-  $ $' -  %())l)CM)H  : :6 B' BB $$,,002H "( C rcD|r |jstjdtj|}tj |}|j |}|j|}|j|j|j|dS)aUpdates the location configs for a service instance. Args: release_track: ALPHA or GA release track instance_ref: resource object - service instance to be updated target_location_configs: A LocationConfigsValue proto message represents the target location configs to achieve Returns: An Operation object which can be used to check on the progress of the service instance update. z(Target location configs must be provided)r&rrErW updateMask) r'rErrorrrFr+r,rGPatch2StreamProjectsLocationsStreamInstancesPatchRequest RelativeName)r8 instance_refrBrKr@rerLs rUpdateLocationConfigsros " $ 9 9   E FF   m ,&  ! !- 0(  $ $5L $ M(  : :6 B' AA((*!'B rc"tj|}tj|}|j|}|j |}|j |}|j |j|j|dS)abUpdate content build version of an Immersive Stream for XR service instance. Args: release_track: ALPHA or GA release track instance_ref: resource object - service instance to be updated version: content build version tag Returns: An Operation object which can be used to check on the progress of the service instance update. rN)rQcontent_build_versionrh) rrFr+rYr,rGrkrlrm)r8rnrarKr@rdrerLs rUpdateContentBuildVersionrrs   m ,&  ! !- 0(''''B-  $ $ $ G(  : :6 B' AA((*!,B rc,tj|}tj|}|j|}|j |}|j }||_|j|j|j|dS)aUpdate fallback url of an Immersive Stream for XR service instance. Args: release_track: ALPHA or GA release track instance_ref: resource object - service instance to be updated fallback_url: string - fallback url to redirect users to when the instance is not available Returns: An Operation object which can be used to check on the progress of the service instance update. rTrfrh) rrFr+rGrZr,r[rkrlrm)r8rnrbrKr@rLrfres rUpdateFallbackUrlrts   m ,&  ! !- 0(  : :6 B'''L'A-  $ $ &('( AA((*!$B r)NNN)__doc__ __future__rrr*googlecloudsdk.api_lib.immersive_stream.xrrgooglecloudsdk.corerrr rrCrHr\rorrrtrrrrysRO&'?*#*>p!f8 8vD6r