python-designateclient/designateclient/v2/zones.py

201 lines
6.3 KiB
Python

# Copyright 2015 Hewlett-Packard Development Company, L.P.
#
# Author: Endre Karlson <endre.karlson@hp.com>
#
# 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 designateclient.v2.base import V2Controller
from designateclient.v2 import utils as v2_utils
class ZoneController(V2Controller):
def create(self, name, type_=None, email=None, description=None, ttl=None,
masters=None, attributes=None):
type_ = type_ or 'PRIMARY'
data = {
'name': name,
'type': type_
}
if type_ == 'PRIMARY':
if email:
data['email'] = email
if ttl is not None:
data['ttl'] = ttl
elif type_ == 'SECONDARY' and masters:
data['masters'] = masters
if description is not None:
data['description'] = description
if attributes is not None:
data['attributes'] = attributes
return self._post('/zones', data=data)
def list(self, criterion=None, marker=None, limit=None):
url = self.build_url('/zones', criterion, marker, limit)
return self._get(url, response_key='zones')
def get(self, zone):
zone = v2_utils.resolve_by_name(self.list, zone)
return self._get(f'/zones/{zone}')
def update(self, zone, values):
zone = v2_utils.resolve_by_name(self.list, zone)
url = self.build_url(f'/zones/{zone}')
return self._patch(url, data=values)
def delete(self, zone, delete_shares=False):
zone = v2_utils.resolve_by_name(self.list, zone)
url = self.build_url(f'/zones/{zone}')
if delete_shares:
headers = {'X-Designate-Delete-Shares': 'true'}
_resp, body = self.client.session.delete(url, headers=headers)
else:
_resp, body = self.client.session.delete(url)
return body
def abandon(self, zone):
zone = v2_utils.resolve_by_name(self.list, zone)
url = f'/zones/{zone}/tasks/abandon'
self.client.session.post(url)
def axfr(self, zone):
zone = v2_utils.resolve_by_name(self.list, zone)
url = f'/zones/{zone}/tasks/xfr'
self.client.session.post(url)
class ZoneTransfersController(V2Controller):
def create_request(self, zone, target_project_id, description=None):
zone = v2_utils.resolve_by_name(self.client.zones.list, zone)
data = {
'target_project_id': target_project_id
}
if description is not None:
data['description'] = description
url = f'/zones/{zone}/tasks/transfer_requests'
return self._post(url, data=data)
def get_request(self, transfer_id):
url = f'/zones/tasks/transfer_requests/{transfer_id}'
return self._get(url)
def list_requests(self):
url = '/zones/tasks/transfer_requests'
return self._get(url, response_key='transfer_requests')
def update_request(self, transfer_id, values):
url = f'/zones/tasks/transfer_requests/{transfer_id}'
return self._patch(url, data=values)
def delete_request(self, transfer_id):
url = f'/zones/tasks/transfer_requests/{transfer_id}'
self._delete(url)
def accept_request(self, transfer_id, key):
url = '/zones/tasks/transfer_accepts'
data = {
'key': key,
'zone_transfer_request_id': transfer_id
}
return self._post(url, data=data)
def get_accept(self, accept_id):
url = f'/zones/tasks/transfer_accepts/{accept_id}'
return self._get(url)
def list_accepts(self):
url = '/zones/tasks/transfer_accepts'
return self._get(url, response_key='transfer_accepts')
class ZoneExportsController(V2Controller):
def create(self, zone):
zone_id = v2_utils.resolve_by_name(self.client.zones.list, zone)
return self._post(f'/zones/{zone_id}/tasks/export')
def get_export_record(self, zone_export_id):
return self._get(f'/zones/tasks/exports/{zone_export_id}')
def list(self):
return self._get('/zones/tasks/exports')
def delete(self, zone_export_id):
return self._delete(f'/zones/tasks/exports/{zone_export_id}')
def get_export(self, zone_export_id):
return self._get(f'/zones/tasks/exports/{zone_export_id}/export',
headers={'Accept': 'text/dns'})
class ZoneImportsController(V2Controller):
def create(self, zone_file_contents):
return self._post('/zones/tasks/imports', data=zone_file_contents,
headers={'Content-Type': 'text/dns'})
def get_import_record(self, zone_import_id):
return self._get(f'/zones/tasks/imports/{zone_import_id}')
def list(self):
return self._get('/zones/tasks/imports')
def delete(self, zone_import_id):
return self._delete(f'/zones/tasks/imports/{zone_import_id}')
class ZoneShareController(V2Controller):
def create(self, zone, target_project_id):
zone_id = v2_utils.resolve_by_name(self.client.zones.list, zone)
data = {'target_project_id': target_project_id}
return self._post(f'/zones/{zone_id}/shares', data=data)
def list(self, zone, criterion=None, marker=None, limit=None):
zone_id = v2_utils.resolve_by_name(self.client.zones.list, zone)
url = self.build_url(f'/zones/{zone_id}/shares',
criterion, marker, limit)
return self._get(url, response_key='shared_zones')
def delete(self, zone, shared_zone_id):
zone_id = v2_utils.resolve_by_name(self.client.zones.list, zone)
return self._delete(f'/zones/{zone_id}/shares/{shared_zone_id}')
def get(self, zone, shared_zone_id):
zone_id = v2_utils.resolve_by_name(self.client.zones.list, zone)
return self._get(f'/zones/{zone_id}/shares/{shared_zone_id}')