dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z d Zejfd Zejfd Zd ZdZefdZy)z#Utilities for handling region flag.)absolute_import)division)unicode_literals) exceptions) constants)log) properties) console_ioctjjjj tjjjj k(SN)r VALUEScoreuniverse_domainGetdefault0lib/googlecloudsdk/command_lib/ai/region_util.py_IsDefaultUniversersE,,002      / / 7 78rcfdS)z]Returns a no argument function that prompts available regions and catches the user selection.ctSr )PromptForRegionavailable_regionssrz(GetPromptForRegionFunc..%s !23rrrs`rGetPromptForRegionFuncr#s 33rctjrctrXt|}tj|dd}||}t j jdj||Syy)a`Prompt for region from list of available regions. This method is referenced by the declaritive iam commands as a fallthrough for getting the region. Prompts only in GDU Args: available_regions: list of the available regions to choose from Returns: The region specified by the user, str, or None if not in GDU or cannot prompt. Please specify a region: Tmessage cancel_optionGTo make this the default region, run `gcloud config set ai/region {}`. N) r CanPromptrlist PromptChoicerstatusPrintformat)r all_regionsidxregions rrr(sr 2 4()K  ! !9 OC  FJJ;;A6&>K M!5rctjrqtrfttj }tj |dd}||}tjjdj||Syy)a&Prompt for region from list of online prediction available regions. This method is referenced by the declaritive iam commands as a fallthrough for getting the region. Prompts only in GDU. Returns: The region specified by the user, str, or None if not in GDU or cannot prompt. rTrr"N) r r#rr$rSUPPORTED_OP_REGIONSr%rr&r'r(r)r*r+s rPromptForOpRegionr/@sw 2 4y556K  ! !9 OC  FJJ;;A6&>K M!5rctjrqtrfttj }tj |dd}||}tjjdj||Syy)a-Prompt for region from list of deployment resource pool available regions. This method is referenced by the declaritive iam commands as a fallthrough for getting the region. Prompts only in GDU. Returns: The region specified by the user, str, or None if not in GDU or cannot prompt. rTrr"N) r r#rr$r*SUPPORTED_DEPLOYMENT_RESOURCE_POOL_REGIONSr%rr&r'r(r.s r.PromptForDeploymentResourcePoolSupportedRegionr2Usw 2 4yKKLK  ! !9 OC  FJJ;;A6&>K M!5rc>t|ddr |jStjjjj r2tjjjj S|}|r|Stjdd)alGets the region and prompt for region if not provided. Region is decided in the following order: - region argument; - ai/region gcloud config; - prompt user input (only in GDU). Args: args: Namespace, The args namespace. prompt_func: Function, To prompt for region from list of available regions. Returns: A str representing region. r+Nz--regionzRegion is required) getattrr+r r aiIsExplicitlySetrrRequiredArgumentException)args prompt_funcr+s r GetRegionr:js~ T8T" ;;  002      & & * * ,, =& M ,,Z9MNNrN)__doc__ __future__rrrgooglecloudsdk.calliopergooglecloudsdk.command_lib.airgooglecloudsdk.corerr googlecloudsdk.core.consoler rSUPPORTED_REGIONrrr/r2r:rrrrBs\*&'.3#*2.7-G-G4 '0&@&@0**!0Or