158 lines
5.2 KiB
Python
158 lines
5.2 KiB
Python
# Copyright 2014 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# 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.
|
|
|
|
from designate.objects.adapters.api_v2 import base
|
|
from designate import objects
|
|
from designate import exceptions
|
|
|
|
|
|
class RecordSetAPIv2Adapter(base.APIv2Adapter):
|
|
|
|
ADAPTER_OBJECT = objects.RecordSet
|
|
|
|
MODIFICATIONS = {
|
|
'fields': {
|
|
"id": {},
|
|
"zone_id": {},
|
|
"project_id": {
|
|
'rename': 'tenant_id'
|
|
},
|
|
"name": {
|
|
'immutable': True
|
|
},
|
|
"zone_name": {
|
|
'read_only': True,
|
|
},
|
|
"type": {
|
|
'rename': 'type',
|
|
'immutable': True
|
|
},
|
|
"records": {
|
|
'read_only': False
|
|
},
|
|
"description": {
|
|
'read_only': False
|
|
},
|
|
"ttl": {
|
|
'read_only': False
|
|
},
|
|
"status": {},
|
|
"action": {},
|
|
"version": {},
|
|
"created_at": {},
|
|
"updated_at": {},
|
|
},
|
|
'options': {
|
|
'links': True,
|
|
'resource_name': 'recordset',
|
|
'collection_name': 'recordsets',
|
|
}
|
|
}
|
|
|
|
@classmethod
|
|
def _parse_object(cls, new_recordset, recordset, *args, **kwargs):
|
|
# TODO(Graham): Remove this when
|
|
# https://bugs.launchpad.net/designate/+bug/1432842 is fixed
|
|
try:
|
|
recordset.records
|
|
except exceptions.RelationNotLoaded:
|
|
recordset.records = objects.RecordList()
|
|
|
|
original_records = set()
|
|
for record in recordset.records:
|
|
original_records.add(record.data)
|
|
# Get new list of Records
|
|
new_records = set()
|
|
if 'records' in new_recordset:
|
|
if isinstance(new_recordset['records'], list):
|
|
for record in new_recordset['records']:
|
|
new_records.add(record)
|
|
else:
|
|
errors = objects.ValidationErrorList()
|
|
e = objects.ValidationError()
|
|
e.path = ['records']
|
|
e.validator = 'type'
|
|
e.validator_value = ["list"]
|
|
e.message = ("'%(data)s' is not a valid list of records"
|
|
% {'data': new_recordset['records']})
|
|
# Add it to the list for later
|
|
errors.append(e)
|
|
raise exceptions.InvalidObject(
|
|
"Provided object does not match "
|
|
"schema", errors=errors, object=cls.ADAPTER_OBJECT())
|
|
|
|
# Get differences of Records
|
|
records_to_add = new_records.difference(original_records)
|
|
records_to_rm = original_records.difference(new_records)
|
|
|
|
# Update all items except records
|
|
record_update = False
|
|
if 'records' in new_recordset:
|
|
record_update = True
|
|
del new_recordset['records']
|
|
|
|
if record_update:
|
|
# Build a list of the "new" records
|
|
new_recordset_records = objects.RecordList()
|
|
|
|
# Remove deleted records if we have provided a records array
|
|
if record_update:
|
|
for record in recordset.records:
|
|
if record.data not in records_to_rm:
|
|
new_recordset_records.append(record)
|
|
|
|
# Add new records
|
|
for record in records_to_add:
|
|
new_recordset_records.append(objects.Record(data=record))
|
|
|
|
# Do a single assignment, preserves the object change fields
|
|
recordset.records = new_recordset_records
|
|
|
|
return super(RecordSetAPIv2Adapter, cls)._parse_object(
|
|
new_recordset, recordset, *args, **kwargs)
|
|
|
|
@classmethod
|
|
def _get_path(cls, request, obj):
|
|
ori_path = request.path
|
|
path = ori_path.lstrip('/').split('/')
|
|
insert_zones = False
|
|
to_insert = ''
|
|
if 'zones' not in path and obj is not None:
|
|
insert_zones = True
|
|
to_insert = 'zones/{0}'.format(obj.zone_id)
|
|
|
|
item_path = ''
|
|
for part in path:
|
|
if part == cls.MODIFICATIONS['options']['collection_name']:
|
|
item_path += '/' + part
|
|
return item_path
|
|
elif insert_zones and to_insert and part == 'v2':
|
|
item_path += '/v2/{0}'.format(to_insert)
|
|
insert_zones = False # make sure only insert once if needed
|
|
else:
|
|
item_path += '/' + part
|
|
|
|
|
|
class RecordSetListAPIv2Adapter(base.APIv2Adapter):
|
|
|
|
ADAPTER_OBJECT = objects.RecordSetList
|
|
|
|
MODIFICATIONS = {
|
|
'options': {
|
|
'links': True,
|
|
'resource_name': 'recordset',
|
|
'collection_name': 'recordsets',
|
|
}
|
|
}
|