118 lines
2.8 KiB
Python
118 lines
2.8 KiB
Python
# 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 distutils.version import StrictVersion
|
|
import operator
|
|
|
|
|
|
SUPPORTED_VERSIONS = [
|
|
'1.0',
|
|
'1.1',
|
|
'1.2',
|
|
'1.3',
|
|
'1.4',
|
|
'1.5',
|
|
'1.6',
|
|
'1.7',
|
|
'1.8',
|
|
'1.9',
|
|
'1.10',
|
|
'1.11',
|
|
'1.12'
|
|
]
|
|
|
|
|
|
def _op(func, b):
|
|
return lambda a: func(StrictVersion(a), StrictVersion(b))
|
|
|
|
|
|
def lt(b):
|
|
return _op(operator.lt, b)
|
|
|
|
|
|
def le(b):
|
|
return _op(operator.le, b)
|
|
|
|
|
|
def eq(b):
|
|
return _op(operator.eq, b)
|
|
|
|
|
|
def ne(b):
|
|
return _op(operator.ne, b)
|
|
|
|
|
|
def ge(b):
|
|
return _op(operator.ge, b)
|
|
|
|
|
|
def gt(b):
|
|
return _op(operator.gt, b)
|
|
|
|
|
|
def _compare(ver, *predicates, **kwargs):
|
|
func = kwargs.get('op', all)
|
|
return func(p(ver) for p in predicates)
|
|
|
|
|
|
def compare(ver, *predicates, **kwargs):
|
|
"""Validate version satisfies provided predicates.
|
|
|
|
kwargs['exc'] - boolean whether exception should be raised
|
|
kwargs['op'] - (all, any) how predicates should be checked
|
|
|
|
Examples:
|
|
compare('1.1', version.gt('1.2'), exc=False) - False
|
|
compare('1.1', version.eq('1.0'), version.eq('1.1'), op=any) - True
|
|
|
|
"""
|
|
exc = kwargs.get('exc', True)
|
|
if not _compare(ver, *predicates, **kwargs):
|
|
if exc:
|
|
raise ValueError(
|
|
'Operation or argument is not supported with version %s' % ver)
|
|
return False
|
|
return True
|
|
|
|
|
|
def check(*predicates, **check_kwargs):
|
|
"""Decorator for command object method.
|
|
|
|
See `compare`
|
|
|
|
"""
|
|
def wrapped(func):
|
|
def inner(self, *args, **kwargs):
|
|
compare(get_version(self), *predicates, **check_kwargs)
|
|
return func(self, *args, **kwargs)
|
|
return inner
|
|
return wrapped
|
|
|
|
|
|
def get_version(obj):
|
|
"""Extract version from a command object."""
|
|
try:
|
|
version = obj.app.client_manager.placement.api_version
|
|
except AttributeError:
|
|
# resource does not have api_version attr when docs are generated
|
|
# so let's use the minimal one
|
|
version = SUPPORTED_VERSIONS[0]
|
|
return version
|
|
|
|
|
|
class CheckerMixin(object):
|
|
def check_version(self, *predicates, **kwargs):
|
|
return compare(get_version(self), *predicates, **kwargs)
|
|
|
|
def compare_version(self, *predicates, **kwargs):
|
|
return compare(get_version(self), *predicates, exc=False, **kwargs)
|