85 lines
3.2 KiB
Python
85 lines
3.2 KiB
Python
# Copyright (c) 2013 Mirantis Inc.
|
|
#
|
|
# 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.
|
|
|
|
import datetime
|
|
|
|
from climateclient import base
|
|
from climateclient import exception
|
|
from climateclient.openstack.common.gettextutils import _ # noqa
|
|
|
|
|
|
class LeaseClientManager(base.BaseClientManager):
|
|
"""Manager for the lease connected requests."""
|
|
|
|
def create(self, name, start, end, reservations, events):
|
|
"""Creates lease from values passed."""
|
|
values = {'name': name, 'start_date': start, 'end_date': end,
|
|
'reservations': reservations, 'events': events}
|
|
|
|
return self._create('/leases', values, 'lease')
|
|
|
|
def get(self, lease_id):
|
|
"""Describes lease specifications such as name, status and locked
|
|
condition.
|
|
"""
|
|
return self._get('/leases/%s' % lease_id, 'lease')
|
|
|
|
def update(self, lease_id, name=None, prolong_for=None, reduce_by=None):
|
|
"""Update attributes of the lease."""
|
|
values = {}
|
|
if name:
|
|
values['name'] = name
|
|
|
|
lease_length_option = prolong_for or reduce_by
|
|
|
|
if lease_length_option:
|
|
if lease_length_option.endswith('s'):
|
|
coefficient = 1
|
|
elif lease_length_option.endswith('m'):
|
|
coefficient = 60
|
|
elif lease_length_option.endswith('h'):
|
|
coefficient = 60 * 60
|
|
elif lease_length_option.endswith('d'):
|
|
coefficient = 24 * 60 * 60
|
|
else:
|
|
raise exception.ClimateClientException(_("Unsupportable date "
|
|
"format for lease "
|
|
"prolonging."))
|
|
lease = self.get(lease_id)
|
|
cur_end_date = datetime.datetime.strptime(lease['end_date'],
|
|
'%Y-%m-%dT%H:%M:%S.%f')
|
|
coefficient = coefficient * (1 if prolong_for else -1)
|
|
seconds = int(lease_length_option[:-1]) * coefficient
|
|
delta_sec = datetime.timedelta(seconds=seconds)
|
|
new_end_date = cur_end_date + delta_sec
|
|
values['end_date'] = datetime.datetime.strftime(
|
|
new_end_date, '%Y-%m-%d %H:%M'
|
|
)
|
|
if not values:
|
|
return _('No values to update passed.')
|
|
return self._update('/leases/%s' % lease_id, values,
|
|
response_key='lease')
|
|
|
|
def delete(self, lease_id):
|
|
"""Deletes lease with specified ID."""
|
|
self._delete('/leases/%s' % lease_id)
|
|
|
|
def list(self, sort_by=None):
|
|
"""List all leases."""
|
|
leases = self._get('/leases', 'leases')
|
|
if sort_by:
|
|
leases = sorted(leases, key=lambda l: l[sort_by])
|
|
return leases
|