X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/943e8bc4af0c11b0ace3811199e3b0844c4c3fbc..700615d15881697c10a77ca65c84a84a5dd1e4d6:/philo/contrib/sobol/utils.py?ds=sidebyside

diff --git a/philo/contrib/sobol/utils.py b/philo/contrib/sobol/utils.py
index 5c52c81..50d2113 100644
--- a/philo/contrib/sobol/utils.py
+++ b/philo/contrib/sobol/utils.py
@@ -30,4 +30,25 @@ def make_tracking_querydict(search_arg, url):
 		SEARCH_ARG_GET_KEY, urlquote_plus(search_arg),
  		URL_REDIRECT_GET_KEY, urlquote(url),
 		HASH_REDIRECT_GET_KEY, make_redirect_hash(search_arg, url))
-	)
\ No newline at end of file
+	)
+
+
+class RegistryIterator(object):
+	def __init__(self, registry, iterattr='__iter__', transform=lambda x:x):
+		if not hasattr(registry, iterattr):
+			raise AttributeError("Registry has no attribute %s" % iterattr)
+		self.registry = registry
+		self.iterattr = iterattr
+		self.transform = transform
+	
+	def __iter__(self):
+		return self
+	
+	def next(self):
+		if not hasattr(self, '_iter'):
+			self._iter = getattr(self.registry, self.iterattr)()
+		
+		return self.transform(self._iter.next())
+	
+	def copy(self):
+		return self.__class__(self.registry, self.iterattr, self.transform)
\ No newline at end of file