-from UserDict import DictMixin
-
from django import forms
from django.contrib.contenttypes.models import ContentType
from django.contrib.contenttypes import generic
from philo.exceptions import AncestorDoesNotExist
from philo.models.fields import JSONField
from philo.signals import entity_class_prepared
-from philo.utils import ContentTypeRegistryLimiter, ContentTypeSubclassLimiter
+from philo.utils import ContentTypeRegistryLimiter, ContentTypeSubclassLimiter, AttributeMapper, TreeAttributeMapper
from philo.validators import json_validator
def __unicode__(self):
return u'"%s": %s' % (self.key, self.value)
+ def set_value(self, value, value_class=JSONValue):
+ """Given a value and a value class, sets up self.value appropriately."""
+ if isinstance(self.value, value_class):
+ val = self.value
+ else:
+ if isinstance(self.value, models.Model):
+ self.value.delete()
+ val = value_class()
+
+ val.set_value(value)
+ val.save()
+
+ self.value = val
+ self.save()
+
class Meta:
app_label = 'philo'
unique_together = (('key', 'entity_content_type', 'entity_object_id'), ('value_content_type', 'value_object_id'))
-class QuerySetMapper(object, DictMixin):
- def __init__(self, queryset, passthrough=None):
- self.queryset = queryset
- self.passthrough = passthrough
-
- def __getitem__(self, key):
- try:
- value = self.queryset.get(key__exact=key).value
- except ObjectDoesNotExist:
- if self.passthrough is not None:
- return self.passthrough.__getitem__(key)
- raise KeyError
- else:
- if value is not None:
- return value.value
- return value
-
- def keys(self):
- keys = set(self.queryset.values_list('key', flat=True).distinct())
- if self.passthrough is not None:
- keys |= set(self.passthrough.keys())
- return list(keys)
-
-
class EntityOptions(object):
def __init__(self, options):
if options is not None:
return new
-class EntityAttributeMapper(object, DictMixin):
- def __init__(self, entity):
- self.entity = entity
-
- def get_attributes(self):
- return self.entity.attribute_set.all()
-
- def make_cache(self):
- attributes = self.get_attributes()
- value_lookups = {}
-
- for a in attributes:
- value_lookups.setdefault(a.value_content_type, []).append(a.value_object_id)
-
- values_bulk = {}
-
- for ct, pks in value_lookups.items():
- values_bulk[ct] = ct.model_class().objects.in_bulk(pks)
-
- self._cache = dict([(a.key, getattr(values_bulk[a.value_content_type].get(a.value_object_id), 'value', None)) for a in attributes])
-
- def __getitem__(self, key):
- if not hasattr(self, '_cache'):
- self.make_cache()
- return self._cache[key]
-
- def keys(self):
- if not hasattr(self, '_cache'):
- self.make_cache()
- return self._cache.keys()
-
- def items(self):
- if not hasattr(self, '_cache'):
- self.make_cache()
- return self._cache.items()
-
- def values(self):
- if not hasattr(self, '_cache'):
- self.make_cache()
- return self._cache.values()
-
-
class Entity(models.Model):
"""An abstract class that simplifies access to related attributes. Most models provided by Philo subclass Entity."""
__metaclass__ = EntityBase
attribute_set = generic.GenericRelation(Attribute, content_type_field='entity_content_type', object_id_field='entity_object_id')
- @property
- def attributes(self):
+ def get_attribute_mapper(self, mapper=AttributeMapper):
"""
- Property that returns a dictionary-like object which can be used to retrieve related :class:`Attribute`\ s' values directly.
+ Returns a dictionary-like object which can be used to retrieve related :class:`Attribute`\ s' values directly.
Example::
u'eggs'
"""
- return EntityAttributeMapper(self)
+ return mapper(self)
+ attributes = property(get_attribute_mapper)
class Meta:
abstract = True
return meta.register(cls)
-class TreeEntityAttributeMapper(EntityAttributeMapper):
- def get_attributes(self):
- 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 TreeEntity(Entity, TreeModel):
"""An abstract subclass of Entity which represents a tree relationship."""
__metaclass__ = TreeEntityBase
- @property
- def attributes(self):
+ def get_attribute_mapper(self, mapper=None):
"""
- Property that returns a dictionary-like object which can be used to retrieve related :class:`Attribute`\ s' values directly. If an attribute with a given key is not related to the :class:`Entity`, then the object will check the parent's attributes.
+ Returns a dictionary-like object which can be used to retrieve related :class:`Attribute`\ s' values directly. If an attribute with a given key is not related to the :class:`Entity`, then the object will check the parent's attributes.
Example::
u'eggs'
"""
-
- if self.parent:
- return TreeEntityAttributeMapper(self)
- return super(TreeEntity, self).attributes
+ if mapper is None:
+ if self.parent:
+ mapper = TreeAttributeMapper
+ else:
+ mapper = AttributeMapper
+ return super(TreeEntity, self).get_attribute_mapper(mapper)
+ attributes = property(get_attribute_mapper)
class Meta:
abstract = True
\ No newline at end of file
+from UserDict import DictMixin
+
from django.db import models
from django.contrib.contenttypes.models import ContentType
from django.core.paginator import Paginator, EmptyPage
from django.template.loader_tags import ExtendsNode, ConstantIncludeNode
+def fattr(*args, **kwargs):
+ def wrapper(function):
+ for key in kwargs:
+ setattr(function, key, kwargs[key])
+ return function
+ return wrapper
+
+
+### ContentTypeLimiters
+
+
class ContentTypeLimiter(object):
def q_object(self):
return models.Q(pk__in=[])
return models.Q(pk__in=contenttype_pks)
-def fattr(*args, **kwargs):
- def wrapper(function):
- for key in kwargs:
- setattr(function, key, kwargs[key])
- return function
- return wrapper
+### Pagination
def paginate(objects, per_page=None, page_number=1):
return paginator, page, objects
+### Facilitating template analysis.
+
+
LOADED_TEMPLATE_ATTR = '_philo_loaded_template'
BLANK_CONTEXT = Context()
# We ignore the IncludeNode because it will never work in a blank context.
setattr(ExtendsNode, LOADED_TEMPLATE_ATTR, property(get_extended))
-setattr(ConstantIncludeNode, LOADED_TEMPLATE_ATTR, property(get_included))
\ No newline at end of file
+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