dZddlmZddlmZddlmZddlZddlmZddlm Z ddl m Z dd l m Z dd l mZdd Zej dd Zdd Zy)z;Utilities for operating on endpoints for different regions.)absolute_import)division)unicode_literalsN)apis) constants)log) properties)parsectj|\}}}}}}|jdr'|rdj||}ndj||}tj||||||fS)z(Adds region as a prefix of the base url. aiplatformz{}-prediction-{}z{}-{})r urlparse startswithformat urlunparse) endpointregion is_predictionschemenetlocpathparamsqueryfragments 2lib/googlecloudsdk/command_lib/ai/endpoint_util.py DeriveAiplatformRegionalEndpointrsn27..2J/&&$x |$!((8f~~ff-f   664I JJc#Kt|||}tjjdj |t j jjj|dyw)a?Context manager to override the AI Platform endpoints for a while. Raises an error if region is not set. Args: version: str, implies the version that the endpoint will use. region: str, region of the AI Platform stack. is_prediction: bool, it's for prediction endpoint or not. Yields: None )versionrrzUsing endpoint [{}]N) GetEffectiveEndpointrstatusPrintrr VALUESapi_endpoint_overridesr Set)rrr used_endpoints rAiplatformEndpointOverridesr&)s['wv5BD-**(// >? **5599-HsA5A7ctjtjtj|}t |||S)zOReturns regional AI Platform endpoint, or raise an error if the region not set.)r)rGetEffectiveApiEndpointrAI_PLATFORM_API_NAMEAI_PLATFORM_API_VERSIONr)rrrrs rrr?s?  ) )$$''02( *m 55r)F)__doc__ __future__rrr contextlibgooglecloudsdk.api_lib.utilrgooglecloudsdk.command_lib.airgooglecloudsdk.corerr six.moves.urllibr rcontextmanagerr&rrrr4sKB&',3#*"K *5r