invirt-quota: more style fixes

[invirt/packages/invirt-database.git] / python / database / __init__.py

diff --git a/python/database/__init__.py b/python/database/__init__.py
index 5eeb3c5..b3dd61c 100644 (file)
--- a/python/database/__init__.py
+++ b/python/database/__init__.py
@@ -4,4 +4,6 @@ import sqlalchemy
 
 def connect(uri = config.db.uri):
     """ Connect to a given database URI"""
-    meta.bind = sqlalchemy.create_engine(uri)
+    engine = sqlalchemy.create_engine(uri, pool_timeout=5)
+    meta.bind = engine
+    session.configure(bind=engine)