Add @cyql.connected() decorator to make functions that operate with database connecti...

[cyql.git] / __init__.py

diff --git a/__init__.py b/__init__.py
index 0c29a582215770b31c8ed2cffdc820e5fa49c525..0f11db4e73652355bfcfe9684e669ac847598fea 100644 (file)
--- a/__init__.py
+++ b/__init__.py
@@ -80,6 +80,14 @@ def connect(dsn):
     finally:
         driver.close()
 
+def connected(dsn):
+    def wrapped(method):
+        def replaced(*args, **kw):
+            with connect(dsn) as connection:
+                return method(connection, *args, **kw)
+        return replaced
+    return wrapped
+
 @contextmanager
 def transact(dsn, **args):
     with connect(dsn) as connection: