Synnefo » ./kamaki

# Copyright 2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

from kamaki.cli.commissioning import commissioning_cli

from kamaki.clients.quotaholder import QuotaholderClient

class quotaholder_cli(commissioning_cli):

    def __init__(self):

        self.client = QuotaholderClient

        self.add_context = True

        self.description = 'Quotaholder description'

        super(self.__class__, self).__init__()

cli = quotaholder_cli()

cli.generate_all()

_commands = [cli.ctree]

# Copyright 2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

from kamaki.cli.commands import _command_init

from kamaki.cli import command

from kamaki.cli.command_tree import CommandTree

from kamaki.cli.errors import CLIError

from kamaki.clients.commissioning import CanonifyException

class commissioning_cli(object):

    api_spec = None

    appname = None

    client = None

    add_context = False

    ctree = None

    description = None

    def __init__(self):

        self.api_spec = self.client.api_spec

        self.appname = self.client.appname

        self.ctree = CommandTree(self.appname, self.description)

    def generate_all(self):

        for f in self.api_spec.call_names():

            c = self.mkClass(f)

            command(self.ctree)(c)

    def mkClass(self, method):

        class C(_command_init):

            __doc__ = self.api_spec.get_doc(method)

            def init(this):

                this.token = (this.config.get(self.appname, 'token') or

                              this.config.get('global', 'token'))

                this.base_url = (this.config.get(self.appname, 'url') or

                                 this.config.get('global', 'url'))

                this.client = self.client(this.base_url, this.token)

            def call(this, method, args):

                ctx = '=null ' if self.add_context else ''

                arglist = '[' + ctx + ' '.join(args) + ']'

                argdict = self.api_spec.parse(method, arglist)

                f = getattr(this.client, method)

                return f(**argdict)

            def main(this, *args):

                this.init()

                try:

                    r = this.call(method, args)

                    print r

                except CanonifyException, e:

                    params = self.api_spec.show_input_canonical(method)

                    meth = method.replace('_', ' ')

                    m = '%s\n  usage: %s %s' % (e, meth, params)

                    raise CLIError('%s: %s\n' % (self.appname, m))

                except Exception as e:

                    raise CLIError('%s: %s\n' % (self.appname, e))

        C.__name__ = self.appname + '_' + method

        return C

# Copyright 2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

# Import general commission framework

from .exception         import (CallError, CorruptedError, InvalidDataError,

                                ReturnButFail,

				register_exception, register_exceptions)

from .callpoint         import  Callpoint, mkcallargs

from .specificator      import (Specificator, SpecifyException,

                                Canonifier, CanonifyException,

                                Canonical,

                                Null, Nothing, Integer, Serial,

                                Text, Bytes, Tuple, ListOf, Dict, Args)

# Import standard implementations?

# Copyright 2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

from .specificator  import  CanonifyException

from .exception     import  CorruptedError, InvalidDataError, ReturnButFail

from .importing     import  imp_module

from re import compile as re_compile, sub as re_sub

class Callpoint(object):

    api_spec = None

    CorruptedError = CorruptedError

    InvalidDataError = InvalidDataError

    original_calls = None

    def __init__(self, connection=None):

        from json import loads, dumps

        self.json_loads = loads

        self.json_dumps = dumps

        self.init_connection(connection)

        original_calls = {}

        self.original_calls = original_calls

        canonifier = self.api_spec

        if canonifier is None:

            m = "No api spec given to '%s'" % (type(self).__name__,)

            raise NotImplementedError(m)

        for call_name, call_doc in canonifier.call_docs():

            if hasattr(self, call_name):

                # don't crash: wrap the function instead

                #m = (   "Method '%s' defined both in natively "

                #        "in callpoint '%s' and in api spec '%s'" %

                #            (call_name,

                #             type(self).__name__,

                #             type(canonifier).__name__)             )

                #raise ValueError(m)

                call_func = getattr(self, call_name)

                if not callable(call_func):

                    m = (   "api spec '%s', method '%s' is not a "

                            "callable attribute in callpoint '%s'" % 

                            (   type(canonifier).__name__,

                                call_name,

                                type(self).__name       )           )

                    raise ValueError(m)

                original_calls[call_name] = call_func

            def mk_call_func():

                local_call_name = call_name

                def call_func(**data):

                    return self.make_call(local_call_name, data)

                call_func.__name__ = call_name

                call_func.__doc__ = call_doc

                return call_func

            setattr(self, call_name, mk_call_func())

    def init_connection(self, connection):

        pass

    def commit(self):

        pass

    def rollback(self):

        pass

    def do_make_call(self, call_name, data):

        raise NotImplementedError

    def validate_call(self, call_name):

        return hasattr(self, call_name)

    def make_call_from_json_description(self, json_description):

        try:

            description = self.json_loads(json_description)

        except ValueError, e:

            m = "Cannot load json description"

            raise self.InvalidDataError(m)

        data = self.make_call_from_description(description)

        json_data = self.json_dumps(data) if data is not None else None

        return json_data

    def make_call_from_description(self, description):

        try:

            call_name = description['call_name']

            call_data = description['call_data']

        except (TypeError, KeyError), e:

            m = "Invalid description"

            raise self.InvalidDataError(m, e)

        return self.make_call(call_name, call_data)

    def make_call_from_json(self, call_name, json_data):

        if json_data:

            try:

                data = self.json_loads(json_data)

            except ValueError, e:

                m = "Cannot load json data"

                raise self.InvalidDataError(m, e)

        else:

            data = None

        data = self.make_call(call_name, data)

        json_data = self.json_dumps(data)

        return json_data

    def make_call(self, call_name, data):

        if call_name.startswith('_'):

            m = "Invalid call '%s'" % (call_name,)

            raise self.InvalidDataError(m)

        canonifier = self.api_spec

        try:

            data = canonifier.canonify_input(call_name, data)

        except CanonifyException, e:

            m = "Invalid input to call '%s'" % (call_name,)

            raise self.InvalidDataError(m, e)

        if not self.validate_call(call_name):

            m = "Cannot find specified call '%s'" % (call_name,)

            raise self.CorruptedError(m)

        call_func = self.original_calls.get(call_name, None)

        try:

            if call_func is None:

                data = self.do_make_call(call_name, data)

            else:

                data = call_func(**data)

            self.commit()

        except ReturnButFail, e:

            self.rollback()

            data = e.data

        except Exception, e:

            self.rollback()

            raise

        try:

            data = canonifier.canonify_output(call_name, data)

        except CanonifyException, e:

            m = "Invalid output from call '%s'" % (call_name,)

            raise self.CorruptedError(m, e)

        return data

def mkcallargs(**kw):

    return kw

versiontag_pattern = re_compile('[^a-zA-Z0-9_-]')

def mk_versiontag(version):

    if not version or version == 'v':

        return ''

    return '_' + re_sub(versiontag_pattern, '_', version)

def get_callpoint(pointname, version=None, automake=None, **kw):

    versiontag = mk_versiontag(version)

    components = pointname.split('.')

    appname = components[0]

    if len(components) < 2:

        raise ValueError("invalid pointname '%s'" % (pointname,))

    category = components[1]

    if not category or category not in ['clients', 'servers']:

        raise ValueError("invalid pointname '%s'" % (pointname,))

    modname = ('%s.callpoint.API_Callpoint%s' 

                                            % (pointname, versiontag))

    try:

        API_Callpoint = imp_module(modname)

        return API_Callpoint

    except ImportError:

        if not automake:

            raise

    if category != 'clients':

        m = ("Can only auto-make callpoint in 'clients' not '%s'" % (category,))

        raise ValueError(m)

    components = components[1:]

    if not components:

        raise ValueError("invalid pointname '%s'" % (pointname))

    pointname = '.'.join(components)

    if pointname == 'quotaholder':

        apiname = 'quotaholder.api.QuotaholderAPI'

    else:

        apiname = '%s.api.API_Spec%s' % (pointname, versiontag)

    API_Spec = imp_module(apiname)

    basename = 'commissioning.clients.%s.API_Callpoint' % (automake,)

    BaseCallpoint = imp_module(basename)

    stupidpython = (appname,

                    version if version is not None else 'v',

                    pointname,

                    automake)

    class AutoCallpoint(BaseCallpoint):

        appname, version, pointname, automake = stupidpython

        api_spec = API_Spec()

    return AutoCallpoint

# Copyright 2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

from kamaki.clients.commissioning import Callpoint, CallError

from kamaki.clients.commissioning.utils.debug import debug

from kamaki.clients import Client

from json import loads as json_loads, dumps as json_dumps

class CommissioningClient(Callpoint):

    def __init__(self, base_url, token):

        super(CommissioningClient, self).__init__()

        self._kc = Client(base_url, token)

    def do_make_call(self, api_call, data):

        _kc = self._kc

        gettable = ['list', 'get', 'read']

        method = (_kc.get if any(api_call.startswith(x) for x in gettable)

                  else _kc.post)

        path = api_call

        json_data = json_dumps(data)

        debug("%s %s\n%s\n<<<\n", method.func_name, path, json_data)

        resp = method(path, data=json_data, success=(200,450,500))

        debug(">>>\nStatus: %s", resp.status_code)

        body = resp.text

        debug("\n%s\n<<<\n", body[:128] if body else None)

        status = int(resp.status_code)

        if status == 200:

            return json_loads(body)

        else:

            try:

                error = json_loads(body)

            except ValueError, e:

                exc = CallError(body, call_error='ValueError')

            else:

                exc = CallError.from_dict(error)

            raise exc

# Copyright 2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

class CallError(Exception):

    exceptions = {}

    def __new__(cls, *args, **kw):

        call_error = kw.get('call_error', None)

        if call_error is None:

            call_error = cls.__name__

        else:

            call_error = str(call_error)

        cls = CallError.exceptions.get(call_error, cls)

        self = Exception.__new__(cls)

        return self

    def __init__(self, *args, **kw):

        self.call_error = kw.get('call_error', self.__class__.__name__)

        self.args = args

    def __str__(self):

        return '\n--------\n'.join(str(x) for x in self.args)

    def __repr__(self):

        return '%s(%s)' % (self.__class__.__name__,

                           ','.join(str(x) for x in self.args))

    @classmethod

    def from_exception(cls, exc):

        args = None

        try:

            args = tuple(exc.args)

        except (TypeError, AttributeError), e:

            pass

        if args is None:

            args = (str(exc),)

        self = cls(*args, call_error=exc.__class__.__name__)

        return self

    def to_dict(self):

        return {'call_error': self.call_error,

                'error_args': self.args}

    @classmethod

    def from_dict(cls, dictobj):

        args = None

        try:

            if 'error_args' in dictobj and 'call_error' in dictobj:

                args = dictobj['error_args']

                call_error = dictobj['call_error']

        except TypeError, e:

            pass

        if args is None:

            args = (str(dictobj),)

            call_error = 'UnknownError'

        self = cls(*args, call_error=call_error)

        return self

def register_exceptions(*exceptions):

    for exception in exceptions:

        if not issubclass(exception, CallError):

            m = "Registering '%s': is not a CallError subclass" % (exception,) 

            raise ValueError(m)

        CallError.exceptions[exception.__name__] = exception

def register_exception(exc):

    register_exceptions(exc)

    return exc

@register_exception

class CorruptedError(CallError):

    pass

@register_exception

class InvalidDataError(CallError):

    pass

class ReturnButFail(Exception):

    def __init__(self, retval=None):

        self.data = retval

# Copyright 2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

from imp import find_module, load_module

_modules = {}

def imp_module(fullname):

    if fullname in _modules:

        return _modules[fullname]

    components = fullname.split('.')

    if not components:

        raise ValueError('invalid module name')

    module = None

    modulepath = []

    for name in components:

        if not name:

            raise ValueError("Relative paths not allowed")

        modulepath.append(name)

        modulename = '.'.join(modulepath)

        if modulename in _modules:

            module = _modules[modulename]

        elif hasattr(module, name):

            module = getattr(module, name)

        elif not hasattr(module, '__path__'):

            m = find_module(name)

            module = load_module(modulename, *m)

        else:

            try:

                m = find_module(name, module.__path__)

                module = load_module(modulename, *m)

            except ImportError:

                m = "No module '%s' in '%s'" % (name, module.__path__)

                raise ImportError(m)

        _modules[modulename] = module

    return module

def list_modules():

    return sorted(_modules.keys())

# -*- coding: utf-8 -*-

# Copyright 2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

from random import random, choice, randint

from math import log

from inspect import isclass

from .utils.argmap import (argmap_decode, argmap_check, argmap_unpack_dict,

                           argmap_unpack_list)

try:

    from collections import OrderedDict

except ImportError:

    from .utils.ordereddict import OrderedDict

def shorts(s):

    if not isinstance(s, unicode):

        s = str(s)

    if len(s) <= 64:

        return s

    return s[:61] + '...'

class CanonifyException(Exception):

    pass

class SpecifyException(Exception):

    pass

class Canonical(object):

    _random_choice = None

    def __init__(self, *args, **kw):

        self.args = []

        named_args = []

        for a in args:

            if isinstance(a, tuple) and len(a) == 2:

                named_args.append(a)

            else:

                self.args.append(a)

        ordered_dict = OrderedDict(named_args)

        self.name = kw.pop('classname', self.__class__.__name__)

        random_choice = kw.pop('random', None)

        if random_choice is not None:

            self.random_choice = random_choice

        opts = {}

        for k, v in kw.items():

            if not isinstance(v, Canonical):

                if isclass(v) and issubclass(v, Canonical):

                    m = ("argument '%s': value '%s' is a Canonical _class_. "

                         "Perhaps you meant to specify a Canonical _instance_"

                         % (k, v))

                    raise SpecifyException(m)

                opts[k] = v

                del kw[k]

        self.opts = opts

        ordered_dict.update(kw)

        self.kw = ordered_dict

        self.init()

        if 'default' in opts:

            item = opts['default']

            if item is None:

                opts['null'] = 1

            else:

                opts['default'] = self._check(item)

    def init(self):

        return

    def __call__(self, item):

        return self.check(item)

    def check(self, item):

        if argmap_check(item):

            item = self._unpack(item)

        opts = self.opts

        if item is None and 'default' in opts:

            item = opts['default']

        can_be_null = opts.get('null', False)

        if item is None and can_be_null:

            return None

        return self._check(item)

    def _check(self, item):

        return item

    def _unpack(self, item):

        return argmap_unpack_list(item)

    def create(self):

        return None

    def random(self, **kw):

        random_choice = self._random_choice

        if random_choice is None:

            return None

        if callable(random_choice):

            return random_choice(kw)

        if isinstance(random_choice, str):

            return getattr(self, random_choice)(kw)

        return choice(random_choice)

    def tostring(self, depth=0, showopts=0, multiline=0):

        depth += 1

        if not multiline:

            argdepth = ''

            owndepth = ''

            joinchar = ','

            padchar = ''

        else:

            argdepth = '    ' * depth

            owndepth = '    ' * (depth - 1)

            joinchar = ',\n'

            padchar = '\n'

        args = [a.tostring( depth=depth,

                            showopts=showopts,

                            multiline=multiline) for a in self.args]

        args += [("%s=%s" %

                    (k, v.tostring( depth=depth,

                                    showopts=showopts,

                                    multiline=multiline)))

                                    for k, v in self.kw.items()]

        if showopts:

            args += [("%s=%s" % (k, str(v))) for k, v in self.opts.items()]

        if len(args) == 0:

            string = "%s(%s)" % (self.name, ','.join(args))

        else:

            string = "%s(%s" % (self.name, padchar)

            for arg in args:

                string += argdepth + arg + joinchar

            string = string[:-1] + padchar

            string += owndepth + ")"

        return string

    __str__ = tostring

    def __repr__(self):

        return self.tostring(multiline=0, showopts=1)

    def show(self):

        showable = self.opts.get('show', True)

        return self._show() if showable else ''

    def _show(self):

        return self.name

class Null(Canonical):

    def _check(self, item):

        return None

Nothing = Null()

class Integer(Canonical):

    def _check(self, item):

        try:

            num = long(item)

        except ValueError, e:

            try:

                num = long(item, 16)

            except Exception:

                m = "%s: cannot convert '%s' to long" % (self, shorts(item))

                raise CanonifyException(m)

        except TypeError, e:

            m = "%s: cannot convert '%s' to long" % (self, shorts(item))

            raise CanonifyException(m)

        optget = self.opts.get

        minimum = optget('minimum', None)

        maximum = optget('maximum', None)

        if minimum is not None and num < minimum:

            m = "%s: %d < minimum=%d" % (self, num, minimum)

            raise CanonifyException(m)

        if maximum is not None and num > maximum:

            m = "%s: %d > maximum=%d" % (self, num, maximum)

            raise CanonifyException(m)

        return num

    def _random_choice(self, kw):

        optget = self.opts.get

        kwget = kw.get

        minimum = kwget('minimum', optget('minimum', -4294967296L))

        maximum = kwget('maximum', optget('maximum', 4294967295L))

        r = random()

        if r < 0.1:

            return minimum

        if r < 0.2:

            return maximum

        if minimum <= 0 and maximum >= 0 and r < 0.3:

            return 0L

        return long(minimum + r * (maximum - minimum))

Serial = Integer(

            classname   =   'Serial',

            null        =   True,

)

class Text(Canonical):

    re = None

    matcher = None

    choices = None

    def init(self):

        opts = self.opts

        if 'regex' in opts:

            pat = opts['regex']

            re = self.re

            if re is None:

                import re

                self.re = re

            self.matcher = re.compile(pat, re.UNICODE)

            self.pat = pat

        if 'choices' in opts:

            opts['choices'] = dict((unicode(x), unicode(x))

                                    for x in opts['choices'])

    def _check(self, item):

        if not isinstance(item, unicode):

            # require non-unicode items to be utf8

            item = str(item)

            try:

                item = item.decode('utf8')

            except UnicodeDecodeError, e:

                item = item.decode('latin1')

                m = "%s: non-unicode '%s' is not utf8" % (self, shorts(item))

                raise CanonifyException(m)

        opts = self.opts

        if 'choices' in opts:

            choices = opts['choices']

            try:

                unknown = item not in choices

            except TypeError, e:

                m = "%s: unhashable type '%s'" % (self.name, shorts(item))

                raise CanonifyException(m, e)

            if unknown:

                m = "%s: '%s' not in choices" % (self.name, shorts(item))

                raise CanonifyException(m)

            return choices[item]

        optget = opts.get

        itemlen = len(item)

        maxlen = optget('maxlen', None)

        if maxlen is not None and itemlen > maxlen:

            m = "%s: len('%s') > maxlen=%d" % (self, shorts(item), maxlen)

            raise CanonifyException(m)

        minlen = optget('minlen', None)

        if minlen is not None and itemlen < minlen:

            m = "%s: len('%s') < minlen=%d" % (self, shorts(item), minlen)

            raise CanonifyException(m)

        matcher = self.matcher

        if matcher is not None:

            match = matcher.match(item)

            if  (       match is None

                    or  (match.start(), match.end()) != (0, itemlen)    ):

                    m = ("%s: '%s' does not match '%s'"

                            % (self, shorts(item), self.pat))

                    raise CanonifyException(m)

        return item

    default_alphabet = '0123456789αβγδεζ'.decode('utf8')

    def _random_choice(self, kw):

        opts = self.opts

        if 'regex' in opts:

            m = 'Unfortunately, random for regex strings not supported'

            raise ValueError(m)

        optget = opts.get

        kwget = kw.get

        minlen = kwget('minlen', optget('minlen', 0))

        maxlen = kwget('maxlen', optget('maxlen', 32))

        alphabet = kwget('alphabet', self.default_alphabet)

        z = maxlen - minlen

        if z < 1:

            z = 1

        g = log(z, 2)

        r = random() * g

        z = minlen + int(2**r)

        s = u''

        for _ in xrange(z):

            s += choice(alphabet)

        return s

class Bytes(Canonical):

    re = None

    matcher = None

    choices = None

    def init(self):

        opts = self.opts

        if 'regex' in opts:

            pat = opts['regex']

            re = self.re

            if re is None:

                import re

                self.re = re

            self.matcher = re.compile(pat)

            self.pat = pat

        if 'choices' in opts:

            opts['choices'] = dict((str(x), str(x))

                                    for x in opts['choices'])

    def _check(self, item):

        if isinstance(item, unicode):

            # convert unicode to utf8

            item = item.encode('utf8')

        opts = self.opts

        if 'choices' in opts:

            choices = opts['choices']

            try:

                unknown = item not in choices

            except TypeError, e:

                m = "%s: unhashable type '%s'" % (self.name, shorts(item))

                raise CanonifyException(m, e)

            if unknown:

                m = "%s: '%s' not in choices" % (self.name, shorts(item))

                raise CanonifyException(m)

            return choices[item]

        optget = opts.get

        itemlen = len(item)

        maxlen = optget('maxlen', None)

        if maxlen is not None and itemlen > maxlen:

            m = "%s: len('%s') > maxlen=%d" % (self, shorts(item), maxlen)

            raise CanonifyException(m)

        minlen = optget('minlen', None)

        if minlen is not None and itemlen < minlen:

            m = "%s: len('%s') < minlen=%d" % (self, shorts(item), minlen)

            raise CanonifyException(m)

        matcher = self.matcher

        if matcher is not None:

            match = matcher.match(item)

            if  (       match is None

                    or  (match.start(), match.end()) != (0, itemlen)    ):

                    m = ("%s: '%s' does not match '%s'"

                            % (self, shorts(item), self.pat))

                    raise CanonifyException(m)

        return item

    default_alphabet = '0123456789abcdef'

    def _random_choice(self, kw):

        opts = self.opts

        if 'regex' in opts:

            m = 'Unfortunately, random for regex strings not supported'

            raise ValueError(m)

        optget = opts.get

        kwget = kw.get

        minlen = kwget('minlen', optget('minlen', 0))

        maxlen = kwget('maxlen', optget('maxlen', 32))

        alphabet = kwget('alphabet', self.default_alphabet)

        z = maxlen - minlen

        if z < 1:

            z = 1

        g = log(z, 2)

        r = random() * g

        z = minlen + int(2**r)

        s = u''

        for _ in xrange(z):

            s += choice(alphabet)

        return s

class ListOf(Canonical):

    def init(self):

        args = self.args

        kw = self.kw

        if not (args or kw):

            raise SpecifyException("ListOf requires one or more arguments")

        if args and kw:

            m = ("ListOf requires either positional "

                 "or keyword arguments, but not both")

            raise SpecifyException(m)

        if args:

            if len(args) > 1:

                self.canonical = Tuple(*args)

            else:

                self.canonical = args[0]

        else:

            self.canonical = Args(**kw)

    def _check(self, item):

        if item is None:

            item = ()

        try:

            items = iter(item)

        except TypeError, e:

            m = "%s: %s is not iterable" % (self, shorts(item))

            raise CanonifyException(m)

        canonical = self.canonical

        canonified = []

        append = canonified.append

        for item in items:

            item = canonical(item)

            append(item)

        if not canonified and self.opts.get('nonempty', False):

            m = "%s: must be nonempty" % (self,)

            raise CanonifyException(m)

        return canonified

    def _random_choice(self, kw):

        z = randint(1, 4)

        get_random = self.canonical.random

        return [get_random() for _ in xrange(z)]

    def _show(self):

        return '[ ' + self.canonical.show() + ' ... ]'

class Args(Canonical):

    def _unpack(self, item):

        arglist = argmap_unpack_dict(item)

        keys = self.kw.keys()

        arglen = len(arglist)

        if arglen != len(keys):

            m = "inconsistent number of parameters: %s != %s" % (

            arglen, len(keys))

            raise CanonifyException(m)

        position = 0

        named_args = OrderedDict()

        for k, v in arglist:

            if k is not None:

                named_args[k] = v

            else:

                # find the right position

                for i in range(position, arglen):

                    key = keys[i]

                    if not key in named_args.keys():

                       position = i + 1

                       break

                else:

                    m = "Formal arguments exhausted"

                    raise AssertionError(m)

                named_args[key] = v

        return named_args

    def _check(self, item):

        try:

            arglist = OrderedDict(item).items()

        except (TypeError, ValueError), e:

            m = "%s: %s is not dict-able" % (self, shorts(item))

            raise CanonifyException(m)

        canonified = OrderedDict()

        try:

            for n, c in self.kw.items():

                t = item[n] if n in item else None

                canonified[n] = c.check(t)

        except KeyError:

            m = ("%s: Argument '%s' not found in '%s'"

                 % (self, shorts(n), shorts(item)))

            raise CanonifyException(m)

        return canonified

    def _show(self):

        strings = [x for x in [c.show() for n, c in self.kw.items()] if x]

        return ' '.join(strings)

    def _random_choice(self, kw):

        args = {}

        for n, c in self.kw.items():

            args[n] = c.random()

        return args

class Tuple(Canonical):

    def _check(self, item):

        try:

            items = list(item)

        except TypeError, e:

            m = "%s: %s is not iterable" % (self, shorts(item))

            raise CanonifyException(m)

        canonicals = self.args

        zi = len(items)

        zc = len(canonicals)

        if zi != zc:

            m = "%s: expecting %d elements, not %d (%s)" % (self, zc, zi, str(items))

            raise CanonifyException(m)

        g = (canonical(element) for canonical, element in zip(self.args, item))

        return tuple(g)

    def __add__(self, other):

        oargs = other.args if isinstance(other, Tuple) else (other,)

        args = self.args + oargs

        return self.__class__(*args)

    def _random_choice(self, kw):

        return tuple(c.random() for c in self.args)

    def _show(self):

        canonicals = self.args

        strings = [x for x in [c.show() for c in canonicals] if x]

        return '[ ' + ' '.join(strings) + ' ]'

class Dict(Canonical):

    def _check(self, item):

        try:

            item = dict(item)

        except TypeError:

            m = "%s: '%s' is not dict-able" % (self, shorts(item))

            raise CanonifyException(m)

        canonified = {}

        canonical = self.kw

        for n, c in canonical.items():

            if n not in item:

                m = "%s: key '%s' not found" % (self, shorts(n))

                raise CanonifyException(m)

            canonified[n] = c(item[n])

        strict = self.opts.get('strict', True)

        if strict and len(item) != len(canonical):

            for k in sorted(item.keys()):

                if k not in canonical:

                    break

            m = "%s: unexpected key '%s' (strict mode)" % (self, shorts(k))

            raise CanonifyException(m)

        return canonified

    def _random_choice(self, kw):

        item = {}

        for n, c in self.kw.items():

            item[n] = c.random()

        return item

class Canonifier(object):

    def __init__(self, name, input_canonicals, output_canonicals, doc_strings):

        self.name = name

        self.input_canonicals = dict(input_canonicals)

        self.output_canonicals = dict(output_canonicals)

        self.doc_strings = dict(doc_strings)

    def call_names(self):

        return self.input_canonicals.keys()

    def call_docs(self):

        get_input_canonical = self.input_canonical

        for call_name, call_doc in self.doc_strings.iteritems():

            if not call_doc:

                canonical = get_input_canonical(call_name)

                call_doc = canonical.tostring(showopts=1, multiline=1)

            yield call_name, call_doc

    def get_doc(self, name):

        doc_strings = self.doc_strings

        if name not in doc_strings:

            m = "%s: Invalid method name '%s'" % (self.name, name)

            raise CanonifyException(m)

        docstring = doc_strings[name]

        if not docstring:

            docstring = self.input_canonical(name).tostring()

        return docstring

    def call_attrs(self):

        for call_name, canonical in self.input_canonicals.iteritems():

            yield call_name, canonical.tostring(showopts=1, multiline=1)

    def input_canonical(self, name):

        input_canonicals = self.input_canonicals

        if name not in input_canonicals:

            m = "%s: Invalid input call '%s'" % (self.name, name)

            raise CanonifyException(m)

        return input_canonicals[name]

    def canonify_input(self, name, the_input):

        return self.input_canonical(name)(the_input)

    def output_canonical(self, name):

        output_canonicals = self.output_canonicals

        if name not in output_canonicals:

            m = "%s: Output canonical '%s' does not exist" % (self.name, name)

            raise CanonifyException(m)

        return output_canonicals[name]

    def canonify_output(self, name, the_output):

        return self.output_canonical(name)(the_output)

    def show_input_canonical(self, name):

        return self.input_canonical(name).show()

    def parse(self, method, arglist):

        args, rest = argmap_decode(arglist)

        return self.input_canonical(method).check(args)

class Specificator(object):

    def __new__(cls):

        if cls is Specificator:

            m = "Specificator classes must be subclassed"

            raise SpecifyException(m)

        import inspect

        canonical_inputs = {}

        canonical_outputs = {}

        doc_strings = {}

        for name in dir(cls):

            f = getattr(cls, name)

            if not inspect.ismethod(f) or f.__name__.startswith('_'):

                continue

            doc_strings[name] = f.__doc__

            argspec = inspect.getargspec(f)

            defaults = argspec.defaults

            args = argspec.args

            if args and args[0] == 'self':

                args = args[1:]

            if not defaults:

                defaults = ()

            arglen = len(args)

            deflen = len(defaults)

            if arglen != deflen:

                a = (f.__name__, args[:arglen-deflen])

                m = "Unspecified arguments in '%s': %s" % a

                raise SpecifyException(m)