Merge pull request #373 from sirnicolaz/sugar

Add utility for pipelining within 'with' statement.

1 files changed, 10 insertions, 0 deletions

diff --git a/redis/utils.py b/redis/utils.py
index ee681bf..e10d20f 100644
--- a/redis/utils.py
+++ b/redis/utils.py
@@ -1,3 +1,6 @@
+from contextlib import contextmanager
+
+
 try:
     import hiredis
     HIREDIS_AVAILABLE = True
@@ -14,3 +17,10 @@ def from_url(url, db=None, **kwargs):
     """
     from redis.client import Redis
     return Redis.from_url(url, db, **kwargs)
+
+
+@contextmanager
+def pipeline(redis_obj):
+    p = redis_obj.pipeline()
+    yield p
+    p.execute()