From 00eac2b5e6446ce22cb5b46803fbb5f8c4d5b903 Mon Sep 17 00:00:00 2001
From: "pengxiong@wealthgrow.cn" <18163159080@163.com>
Date: Wed, 2 Dec 2020 18:35:48 +0800
Subject: [PATCH] sql

---
 app/api/engine.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/app/api/engine.py b/app/api/engine.py
index b1f6b38..2c13d96 100644
--- a/app/api/engine.py
+++ b/app/api/engine.py
@@ -74,3 +74,28 @@ logging.basicConfig(level=logging.INFO,
                     datefmt=config[env]['log']['datefmt']
                     )
 
+class TAMP_SQL(object):
+    """[sqlalchemy 封装]
+
+    Args:
+        object ([type]): [description]
+
+    Returns:
+        [type]: [description]
+    """
+
+    def __init__(self, db_engine):
+        # 创建DBSession类型:
+        self.DBSession = scoped_session(sessionmaker(bind=db_engine))
+        self.session = self.DBSession()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        try:
+            self.session.commit()
+        except:
+            self.session.rollback()
+        finally:
+            self.session.close()
\ No newline at end of file
-- 
2.18.1