82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
# -*- coding: utf-8 -*- #
|
|
# Copyright 2017 Google LLC. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
"""Common validation methods for some SQL commands."""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
|
|
import re
|
|
from googlecloudsdk.api_lib.sql import validate as api_validate
|
|
from googlecloudsdk.calliope import arg_parsers
|
|
|
|
|
|
def InstanceNameRegexpValidator():
|
|
"""Returns a function that validates an instance name using predefined rules.
|
|
|
|
Returns:
|
|
function: str -> str, usable as an argparse type
|
|
"""
|
|
|
|
# : and . are not valid characters, but we allow them through this regex so
|
|
# that we can give a better error message using ValidateInstanceName, below.
|
|
pattern = r'^[a-z][a-z0-9-:.]*'
|
|
description = ('must be composed of lowercase letters, numbers, and hyphens; '
|
|
'must start with a letter.')
|
|
|
|
def Parse(value):
|
|
if not re.match(pattern + '$', value):
|
|
raise arg_parsers.ArgumentTypeError(
|
|
'Bad value [{0}]: {1}'.format(value, description))
|
|
api_validate.ValidateInstanceName(value)
|
|
return value
|
|
|
|
return Parse
|
|
|
|
|
|
def IsProjectLevelBackupRequest(backup_id):
|
|
"""Checks if the backup request is project level.
|
|
|
|
Project level requests will have backup_id in string format whearas they will
|
|
be integer values for instance level backup requests.
|
|
|
|
Args:
|
|
backup_id: The id of the requested backup.
|
|
|
|
Returns:
|
|
True if is a project level backup request.
|
|
"""
|
|
try:
|
|
int(backup_id)
|
|
except ValueError:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def IsBackupDrBackupRequest(backup_id: str) -> bool:
|
|
"""Checks if the backup request is a backupdr backup by checking if the backup id contains /backupVaults.
|
|
|
|
A backupdr backup will have the backup in the format of
|
|
projects/{project}/locations/{location}/backupVaults/{backup_vault}/dataSources/{data_source}/backups/{backup}.
|
|
|
|
Args:
|
|
backup_id: The id of the requested backup.
|
|
|
|
Returns:
|
|
True if the request is a backupdr backup request.
|
|
"""
|
|
return backup_id and '/backupVaults' in backup_id
|