def Run(self, args):
env_ref = args.CONCEPTS.environment.Parse()
if args.airflow_version:
# Converts airflow_version arg to image_version arg
args.image_version = (
image_versions_command_util.ImageVersionFromAirflowVersion(
args.airflow_version))
# Checks validity of image_version upgrade request.
if (args.image_version and
not image_versions_command_util.IsValidImageVersionUpgrade(
env_ref, args.image_version, self.ReleaseTrack())):
raise command_util.InvalidUserInputError(
'Invalid environment upgrade. [Requested: {}]'.format(
args.image_version))
# Checks validity of update_web_server_allow_ip
if (self.ReleaseTrack() == base.ReleaseTrack.BETA and
args.update_web_server_allow_ip):
flags.ValidateIpRanges(
[acl['ip_range'] for acl in args.update_web_server_allow_ip])
field_mask, patch = self._ConstructPatch(
env_ref, args, UpdateBeta.support_environment_upgrades,
UpdateBeta.support_web_server_access_control, True, True)
return patch_util.Patch(
env_ref,
field_mask,
patch,
args.async_,
release_track=self.ReleaseTrack())
def ParseWebServerAccessControlConfigOptions(self, args):
if (args.enable_private_environment and not args.web_server_allow_ip and
not args.web_server_allow_all and not args.web_server_deny_all):
raise command_util.InvalidUserInputError(
'Cannot specify --enable-private-environment without one of: ' +
'--web-server-allow-ip, --web-server-allow-all ' +
'or --web-server-deny-all')
# Default to allow all if no flag is specified.
self.web_server_access_control = (
environments_api_util.BuildWebServerAllowedIps(
args.web_server_allow_ip, args.web_server_allow_all or
not args.web_server_allow_ip, args.web_server_deny_all))
flags.ValidateIpRanges(
[acl['ip_range'] for acl in self.web_server_access_control])
def ParseWebServerAccessControlConfigOptions(self, args):
self.web_server_access_control = environments_api_util.BuildWebServerAllowedIps(
args.web_server_allow_ip, not args.web_server_allow_ip,
args.web_server_deny_all)
flags.ValidateIpRanges(
[acl['ip_range'] for acl in self.web_server_access_control])