+setattr(ConstantIncludeNode, LOADED_TEMPLATE_ATTR, property(get_included))
+
+
+### AttributeMappers
+
+
+class AttributeMapper(object, DictMixin):
+ def __init__(self, entity):
+ self.entity = entity
+ self.clear_cache()
+
+ def __getitem__(self, key):
+ if not self._cache_populated:
+ self._populate_cache()
+ return self._cache[key]
+
+ def __setitem__(self, key, value):
+ # Prevent circular import.
+ from philo.models.base import JSONValue, ForeignKeyValue, ManyToManyValue, Attribute
+ old_attr = self.get_attribute(key)
+ if old_attr and old_attr.entity_content_type == ContentType.objects.get_for_model(self.entity) and old_attr.entity_object_id == self.entity.pk:
+ attribute = old_attr
+ else:
+ attribute = Attribute(key=key)
+ attribute.entity = self.entity
+ attribute.full_clean()
+
+ if isinstance(value, models.query.QuerySet):
+ value_class = ManyToManyValue
+ elif isinstance(value, models.Model):
+ value_class = ForeignKeyValue
+ else:
+ value_class = JSONValue
+
+ attribute.set_value(value=value, value_class=value_class)
+ self._cache[key] = attribute.value.value
+ self._attributes_cache[key] = attribute
+
+ def get_attributes(self):
+ return self.entity.attribute_set.all()
+
+ def get_attribute(self, key):
+ if not self._cache_populated:
+ self._populate_cache()
+ return self._attributes_cache.get(key, None)
+
+ def keys(self):
+ if not self._cache_populated:
+ self._populate_cache()
+ return self._cache.keys()
+
+ def items(self):
+ if not self._cache_populated:
+ self._populate_cache()
+ return self._cache.items()
+
+ def values(self):
+ if not self._cache_populated:
+ self._populate_cache()
+ return self._cache.values()
+
+ def _populate_cache(self):
+ if self._cache_populated:
+ return
+
+ attributes = self.get_attributes()
+ value_lookups = {}
+
+ for a in attributes:
+ value_lookups.setdefault(a.value_content_type, []).append(a.value_object_id)
+ self._attributes_cache[a.key] = a
+
+ values_bulk = {}
+
+ for ct, pks in value_lookups.items():
+ values_bulk[ct] = ct.model_class().objects.in_bulk(pks)
+
+ self._cache.update(dict([(a.key, getattr(values_bulk[a.value_content_type].get(a.value_object_id), 'value', None)) for a in attributes]))
+ self._cache_populated = True
+
+ def clear_cache(self):
+ self._cache = {}
+ self._attributes_cache = {}
+ self._cache_populated = False
+
+
+class LazyAttributeMapperMixin(object):
+ def __getitem__(self, key):
+ if key not in self._cache and not self._cache_populated:
+ self._add_to_cache(key)
+ return self._cache[key]
+
+ def get_attribute(self, key):
+ if key not in self._attributes_cache and not self._cache_populated:
+ self._add_to_cache(key)
+ return self._attributes_cache[key]
+
+ def _add_to_cache(self, key):
+ try:
+ attr = self.get_attributes().get(key=key)
+ except Attribute.DoesNotExist:
+ raise KeyError
+ else:
+ val = getattr(attr.value, 'value', None)
+ self._cache[key] = val
+ self._attributes_cache[key] = attr
+
+
+class LazyAttributeMapper(LazyAttributeMapperMixin, AttributeMapper):
+ def get_attributes(self):
+ return super(LazyAttributeMapper, self).get_attributes().exclude(key__in=self._cache.keys())
+
+
+class TreeAttributeMapper(AttributeMapper):
+ def get_attributes(self):
+ from philo.models import Attribute
+ ancestors = dict(self.entity.get_ancestors(include_self=True).values_list('pk', 'level'))
+ ct = ContentType.objects.get_for_model(self.entity)
+ return sorted(Attribute.objects.filter(entity_content_type=ct, entity_object_id__in=ancestors.keys()), key=lambda x: ancestors[x.entity_object_id])
+
+
+class LazyTreeAttributeMapper(LazyAttributeMapperMixin, TreeAttributeMapper):
+ def get_attributes(self):
+ return super(LazyTreeAttributeMapper, self).get_attributes().exclude(key__in=self._cache.keys())
+
+
+class PassthroughAttributeMapper(AttributeMapper):
+ def __init__(self, entities):
+ self._attributes = [e.attributes for e in entities]
+ super(PassthroughAttributeMapper, self).__init__(self._attributes[0].entity)
+
+ def _populate_cache(self):
+ if self._cache_populated:
+ return
+
+ for a in reversed(self._attributes):
+ a._populate_cache()
+ self._attributes_cache.update(a._attributes_cache)
+ self._cache.update(a._cache)
+
+ self._cache_populated = True
+
+ def get_attributes(self):
+ raise NotImplementedError
+
+ def clear_cache(self):
+ super(PassthroughAttributeMapper, self).clear_cache()
+ for a in self._attributes:
+ a.clear_cache()
+
+
+class LazyPassthroughAttributeMapper(LazyAttributeMapperMixin, PassthroughAttributeMapper):
+ def _add_to_cache(self, key):
+ for a in self._attributes:
+ try:
+ self._cache[key] = a[key]
+ self._attributes_cache[key] = a.get_attribute(key)
+ except KeyError:
+ pass
+ else:
+ break
+ return self._cache[key]
\ No newline at end of file