reclass/values/refitem.py

#
# -*- coding: utf-8 -*-
#
# This file is part of reclass
#

from reclass.values import item
from reclass.utils.dictpath import DictPath
from reclass.errors import ResolveError


class RefItem(item.ItemWithReferences):

    type = item.ItemTypes.REFERENCE

    def assembleRefs(self, context={}):
        super(RefItem, self).assembleRefs(context)
        try:
            strings = [str(i.render(context, None)) for i in self.contents]
            value = "".join(strings)
            self._refs.append(value)
        except ResolveError as e:
            self.allRefs = False

    def _resolve(self, ref, context):
        path = DictPath(self._settings.delimiter, ref)
        try:
            return path.get_value(context)
        except (KeyError, TypeError) as e:
            raise ResolveError(ref)

    def render(self, context, inventory):
        if len(self.contents) == 1:
            return self._resolve(self.contents[0].render(context, inventory),
                                 context)
        strings = [str(i.render(context, inventory)) for i in self.contents]
        return self._resolve("".join(strings), context)

    def __str__(self):
        strings = [str(i) for i in self.contents]
        rs = self._settings.reference_sentinels
        return '{0}{1}{2}'.format(rs[0], ''.join(strings), rs[1])