From a2d1fedffd3b6684a87b2de4a9601f87f8008709 Mon Sep 17 00:00:00 2001
From: Joe Jevnik <joe@quantopian.com>
Date: Tue, 5 Jan 2016 14:03:07 -0500
Subject: [PATCH] ENH: Adds classlazyval for computed values on a class

---
 zipline/utils/memoize.py | 34 +++++++++++++++++++++++++++++++---
 1 file changed, 31 insertions(+), 3 deletions(-)

diff --git a/zipline/utils/memoize.py b/zipline/utils/memoize.py
index aeedb01c..70401582 100644
--- a/zipline/utils/memoize.py
+++ b/zipline/utils/memoize.py
@@ -7,9 +7,8 @@ from weakref import WeakKeyDictionary
 
 
 class lazyval(object):
-    """
-    Decorator that marks that an attribute should not be computed until
-    needed, and that the value should be memoized.
+    """Decorator that marks that an attribute of an instance should not be
+    computed until needed, and that the value should be memoized.
 
     Example
     -------
@@ -57,6 +56,35 @@ class lazyval(object):
         del self._cache[instance]
 
 
+class classlazyval(lazyval):
+    """ Decorator that marks that an attribute of a class should not be
+    computed until needed, and that the value should be memoized.
+
+    Example
+    -------
+
+    >>> from zipline.utils.memoize import classlazyval
+    >>> class C(object):
+    ...     count = 0
+    ...     @classlazyval
+    ...     def val(cls):
+    ...         cls.count += 1
+    ...         return "val"
+    ...
+    >>> C.count
+    0
+    >>> C.val, C.count
+    ('val', 1)
+    >>> C.val, C.count
+    ('val', 1)
+    """
+    # We don't reassign the name on the class to implement the caching because
+    # then we would need to use a metaclass to track the name of the
+    # descriptor.
+    def __get__(self, instance, owner):
+        return super(classlazyval, self).__get__(owner, owner)
+
+
 def remember_last(f):
     """
     Decorator that remembers the last computed value of a function and doesn't