X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-database.git/blobdiff_plain/fe3cf67c45ba77d082eeadfc191d627b43373cc1..9356fabe4ee1e31b768136e5a3e0315135736be6:/python/database/__init__.py

diff --git a/python/database/__init__.py b/python/database/__init__.py
index 5eeb3c5..ce88809 100644
--- a/python/database/__init__.py
+++ b/python/database/__init__.py
@@ -1,7 +1,9 @@
-from invirt.config import structs as config
+from invirt.config import config
 from models import *
 import sqlalchemy
 
-def connect(uri = config.db.uri):
+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)