def testWriteToYamlFile(self):
with open(self.result_file_path, 'w') as yaml_file:
export_util.WriteToYamlFile(yaml_file, util.GetRecordSetsForExport())
with open(self.result_file_path) as results:
with open(self.yaml_file_path) as expected:
self.assertEqual(expected.read().splitlines(),
results.read().splitlines())
def Run(self, args):
api_version = 'v1'
# If in the future there are differences between API version, do NOT use
# this patter of checking ReleaseTrack. Break this into multiple classes.
if self.ReleaseTrack() == base.ReleaseTrack.BETA:
api_version = 'v1beta2'
if self.ReleaseTrack() == base.ReleaseTrack.ALPHA:
api_version = 'v1alpha2'
dns = util.GetApiClient(api_version)
# Get the managed-zone.
zone_ref = util.GetRegistry(api_version).Parse(
args.zone,
params={
'project': properties.VALUES.core.project.GetOrFail,
},
collection='dns.managedZones')
try:
zone = dns.managedZones.Get(
dns.MESSAGES_MODULE.DnsManagedZonesGetRequest(
project=zone_ref.project,
managedZone=zone_ref.managedZone))
except apitools_exceptions.HttpError as error:
raise calliope_exceptions.HttpException(error)
# Get all the record-sets.
record_sets = []
for record_set in list_pager.YieldFromList(
dns.resourceRecordSets,
dns.MESSAGES_MODULE.DnsResourceRecordSetsListRequest(
project=zone_ref.project, managedZone=zone_ref.Name()),
field='rrsets'):
record_sets.append(record_set)
# Export the record-sets.
try:
with files.FileWriter(args.records_file) as export_file:
if args.zone_file_format:
export_util.WriteToZoneFile(export_file, record_sets,
zone.dnsName)
else:
export_util.WriteToYamlFile(export_file, record_sets)
except Exception as exp:
msg = 'Unable to export record-sets to file [{0}]: {1}'.format(
args.records_file, exp)
raise export_util.UnableToExportRecordsToFile(msg)
log.status.Print('Exported record-sets to [{0}].'.format(
args.records_file))
def Run(self, args):
dns = self.context['dns_client']
messages = self.context['dns_messages']
resources = self.context['dns_resources']
# Get the managed-zone.
zone_ref = resources.Parse(
args.zone,
params={
'project': properties.VALUES.core.project.GetOrFail,
},
collection='dns.managedZones')
try:
zone = dns.managedZones.Get(
dns.MESSAGES_MODULE.DnsManagedZonesGetRequest(
project=zone_ref.project,
managedZone=zone_ref.managedZone))
except apitools_exceptions.HttpError as error:
raise exceptions.HttpException(error)
# Get all the record-sets.
record_sets = []
for record_set in list_pager.YieldFromList(
dns.resourceRecordSets,
messages.DnsResourceRecordSetsListRequest(
project=zone_ref.project, managedZone=zone_ref.Name()),
field='rrsets'):
record_sets.append(record_set)
# Export the record-sets.
try:
with open(args.records_file, 'w') as export_file:
if args.zone_file_format:
export_util.WriteToZoneFile(export_file, record_sets,
zone.dnsName)
else:
export_util.WriteToYamlFile(export_file, record_sets)
except Exception as exp:
msg = 'unable to export record-sets to file [{0}]: {1}'.format(
args.records_file, exp)
raise exceptions.ToolException(msg)
log.status.Print('Exported record-sets to [{0}].'.format(
args.records_file))
def Run(self, args):
dns = self.context['dns_client']
messages = self.context['dns_messages']
resources = self.context['dns_resources']
project_id = properties.VALUES.core.project.Get(required=True)
# Get the managed-zone.
zone_ref = resources.Parse(args.zone, collection='dns.managedZones')
try:
zone = dns.managedZones.Get(zone_ref.Request())
except apitools_base.HttpError as error:
raise exceptions.HttpException(util.GetErrorMessage(error))
# Get all the record-sets.
record_sets = []
for record_set in apitools_base.YieldFromList(
dns.resourceRecordSets,
messages.DnsResourceRecordSetsListRequest(
project=project_id, managedZone=zone_ref.Name()),
field='rrsets'):
record_sets.append(record_set)
# Export the record-sets.
try:
with files.Context(open(args.records_file, 'w')) as export_file:
if args.zone_file_format:
export_util.WriteToZoneFile(export_file, record_sets,
zone.dnsName)
else:
export_util.WriteToYamlFile(export_file, record_sets)
except Exception as exp:
msg = 'unable to export record-sets to file [{0}]: {1}'.format(
args.records_file, exp)
raise exceptions.ToolException(msg)
log.status.Print('Exported record-sets to [{0}].'.format(
args.records_file))
def Run(self, args):
api_version = util.GetApiFromTrackAndArgs(self.ReleaseTrack(), args)
dns = util.GetApiClient(api_version)
# Get the managed-zone.
zone_ref = util.GetRegistry(api_version).Parse(
args.zone,
params=util.GetParamsForRegistry(api_version, args),
collection='dns.managedZones')
try:
get_request = dns.MESSAGES_MODULE.DnsManagedZonesGetRequest(
project=zone_ref.project, managedZone=zone_ref.managedZone)
if api_version == 'v2' and self._IsBetaOrAlpha():
get_request.location = args.location
zone = dns.managedZones.Get(get_request)
except apitools_exceptions.HttpError as error:
raise calliope_exceptions.HttpException(error)
# Get all the record-sets.
record_sets = []
list_request = dns.MESSAGES_MODULE.DnsResourceRecordSetsListRequest(
project=zone_ref.project, managedZone=zone_ref.Name())
if api_version == 'v2' and self._IsBetaOrAlpha():
list_request.location = args.location
for record_set in list_pager.YieldFromList(dns.resourceRecordSets,
list_request,
field='rrsets'):
# Alpha is handled differently as it supports ALIAS records.
if self._IsAlpha():
# For BIND file format, ALIAS record sets must be ignored, as they are
# not DNS standards. A zone will have at most one ALIAS record.
if args.zone_file_format:
if record_set.type == 'ALIAS':
log.warning(
'Skippig ALIAS record found in zone, as ALIAS record are custom'
' to Cloud DNS and do not have a standard BIND format. To '
'export ALIAS records, use YAML format instead.')
continue
else: # beta or GA
# Quietly ignore ALIAS records in beta/GA - they aren't supported yet.
if record_set.type == 'ALIAS':
continue
record_sets.append(record_set)
# Export the record-sets.
try:
with files.FileWriter(args.records_file) as export_file:
if args.zone_file_format:
export_util.WriteToZoneFile(export_file, record_sets,
zone.dnsName)
else:
export_util.WriteToYamlFile(export_file, record_sets)
except Exception as exp:
msg = 'Unable to export record-sets to file [{0}]: {1}'.format(
args.records_file, exp)
raise export_util.UnableToExportRecordsToFile(msg)
log.status.Print('Exported record-sets to [{0}].'.format(
args.records_file))
