Author: justi9 Date: 2010-09-23 10:36:07 -0400 (Thu, 23 Sep 2010) New Revision: 4335 Modified: mgmt/newdata/cumin/python/cumin/database.py Log: Replace closed database connections on demand Modified: mgmt/newdata/cumin/python/cumin/database.py =================================================================== --- mgmt/newdata/cumin/python/cumin/database.py 2010-09-23 13:49:25 UTC (rev 4334) +++ mgmt/newdata/cumin/python/cumin/database.py 2010-09-23 14:36:07 UTC (rev 4335) @@ -20,11 +20,14 @@ #m = re.match(r"^([^:]+)://([^@]+)@([^/]+)/(.+)$", self.uri) def get_connection(self): - if not hasattr(self.thread_local, "connection"): - self.thread_local.connection = psycopg2.connect(self.dsn) + conn = getattr(self.thread_local, "connection", None) - return self.thread_local.connection + if not conn or conn.closed: + conn = psycopg2.connect(self.dsn) + setattr(self.thread_local, "connection", conn) + return conn + def __repr__(self): return self.__class__.__name__