Commit 093306bc authored by pengxiong's avatar pengxiong

数据库连接关闭

parent 49378324
...@@ -93,7 +93,8 @@ class TAMP_SQL(object): ...@@ -93,7 +93,8 @@ class TAMP_SQL(object):
def __init__(self, db_engine): def __init__(self, db_engine):
# 创建DBSession类型: # 创建DBSession类型:
self.DBSession = scoped_session(sessionmaker(bind=db_engine)) self.db_engine = db_engine
self.DBSession = scoped_session(sessionmaker(bind=self.db_engine))
self.session = self.DBSession() self.session = self.DBSession()
def __enter__(self): def __enter__(self):
...@@ -105,4 +106,5 @@ class TAMP_SQL(object): ...@@ -105,4 +106,5 @@ class TAMP_SQL(object):
except: except:
self.session.rollback() self.session.rollback()
finally: finally:
self.session.dispose() self.session.close()
self.db_engine.dispose()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment