liqian 3 år sedan
förälder
incheckning
82aa357eed
2 ändrade filer med 32 tillägg och 21 borttagningar
  1. 2 2
      config.py
  2. 30 19
      db_helper.py

+ 2 - 2
config.py

@@ -160,6 +160,6 @@ class ProductionConfig(BaseConfig):
 
 def set_config():
     # return DevelopmentConfig()
-    # return TestConfig()
+    return TestConfig()
     # return PreProductionConfig()
-    return ProductionConfig()
+    # return ProductionConfig()

+ 30 - 19
db_helper.py

@@ -1,3 +1,5 @@
+import traceback
+
 import redis
 import psycopg2
 from psycopg2 import pool as pgpool
@@ -5,10 +7,11 @@ from config import set_config
 from log import Log
 
 config_ = set_config()
-log = Log()
+log_ = Log()
 
 conn_redis = None
 
+
 class RedisHelper(object):
     def __init__(self):
         """
@@ -28,9 +31,9 @@ class RedisHelper(object):
         global conn_redis
         if conn_redis is None:
             pool = redis.ConnectionPool(host=self.host,
-                                    port=self.port,
-                                    password=self.password,
-                                    decode_responses=True)
+                                        port=self.port,
+                                        password=self.password,
+                                        decode_responses=True)
             conn = redis.Redis(connection_pool=pool)
             conn_redis = conn
         return conn_redis
@@ -231,7 +234,10 @@ class RedisHelper(object):
 #hologres_info = config_.HOLOGRES_INFO
 #conn = psycopg2.connect(**hologres_info)
 #cur = conn.cursor()
+
 connectPool = pgpool.SimpleConnectionPool(1, 10, **config_.HOLOGRES_INFO)
+
+
 class HologresHelper(object):
     def __init__(self):
         """初始化hologres连接信息"""
@@ -239,21 +245,26 @@ class HologresHelper(object):
 
     def get_data(self, sql):
         #global conn
-        # 连接Hologres
-        #conn = psycopg2.connect(**self.hologres_info)
-        conn = connectPool.getconn()
-        # 创建游标
-        cur = conn.cursor()
-        # 查询数据
-        cur.execute(sql)
-        data = cur.fetchall()
-        cur.close()
-        connectPool.putconn(conn, close=False)
-        # 提交事务
-        #conn.commit()
-        # 释放资源
-        #cur.close()
-        #conn.close()
+        try:
+            # 连接Hologres
+            #conn = psycopg2.connect(**self.hologres_info)
+            conn = connectPool.getconn()
+            # 创建游标
+            cur = conn.cursor()
+            # 查询数据
+            cur.execute(sql)
+            data = cur.fetchall()
+            cur.close()
+            connectPool.putconn(conn, close=False)
+            # 提交事务
+            #conn.commit()
+            # 释放资源
+            #cur.close()
+            #conn.close()
+        except Exception as e:
+            log_.error('hologress error...')
+            log_.error(traceback.format_exc())
+            return []
         return data