Source code for robot.running.arguments.argumentmapper

#  Copyright 2008-2015 Nokia Networks
#  Copyright 2016-     Robot Framework Foundation
#  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
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  See the License for the specific language governing permissions and
#  limitations under the License.

from robot.errors import DataError

[docs]class ArgumentMapper(object): def __init__(self, argspec): """:type argspec: :py:class:`robot.running.arguments.ArgumentSpec`""" self._argspec = argspec
[docs] def map(self, positional, named, replace_defaults=True): template = KeywordCallTemplate(self._argspec) template.fill_positional(positional) template.fill_named(named) if replace_defaults: template.replace_defaults() return template.args, template.kwargs
[docs]class KeywordCallTemplate(object): def __init__(self, argspec): """:type argspec: :py:class:`robot.running.arguments.ArgumentSpec`""" self._argspec = argspec self.args = [None if arg not in argspec.defaults else DefaultValue(argspec.defaults[arg]) for arg in argspec.positional] self.kwargs = []
[docs] def fill_positional(self, positional): self.args[:len(positional)] = positional
[docs] def fill_named(self, named): spec = self._argspec for name, value in named: if name in spec.positional_or_named: index = spec.positional_or_named.index(name) self.args[index] = value elif spec.var_named or name in spec.named_only: self.kwargs.append((name, value)) else: raise DataError("Non-existing named argument '%s'." % name) named_names = {name for name, _ in named} for name in spec.named_only: if name not in named_names: value = DefaultValue(spec.defaults[name]) self.kwargs.append((name, value))
[docs] def replace_defaults(self): is_default = lambda arg: isinstance(arg, DefaultValue) while self.args and is_default(self.args[-1]): self.args.pop() self.args = [a if not is_default(a) else a.value for a in self.args] self.kwargs = [(n, v) for n, v in self.kwargs if not is_default(v)]
[docs]class DefaultValue(object): def __init__(self, value): self.value = value
[docs] def resolve(self, variables): try: return variables.replace_scalar(self.value) except DataError as err: raise DataError('Resolving argument default values failed: %s' % err.message)