mvp: 增加数据写入临时库的代码

Signed-off-by: binren <zhangbr@elab-plus.com>

mvp.py

@@ -399,6 +399,7 @@ class Mvp:
             定时更新分值
         :return:
         """
+        self.linshi_db = MysqlDB('linshi', db_type=1)
         for city in self.city_list:
             for age in self.age_list:
                 for crowd in self.crowd:
@@ -433,8 +434,8 @@ class Mvp:
             # 先清空之前的数据
             table_name = self.get_table_name('模块分数')
             if table_name and len(module_insert_data) > 0:
-                self.shangju_db.truncate(table_name)
-            self.shangju_db.add_some(module_insert_sql, module_insert_data)
+                self.linshi_db.truncate(table_name)
+            self.linshi_db.add_some(module_insert_sql, module_insert_data)
             print('模块分数更新完成...')
 
         for b_score in behavior_score:
@@ -459,8 +460,8 @@ class Mvp:
                     if len(insert_data) > 0:
                         table_name = self.get_table_name(key)
                         if table_name:
-                            self.shangju_db.truncate(table_name)
-                        self.shangju_db.add_some(insert_sql, insert_data)
+                            self.linshi_db.truncate(table_name)
+                        self.linshi_db.add_some(insert_sql, insert_data)
                 else:
                     print('未找到对应的表，数据无法插入...')
             print('行为分数更新完成...')
@@ -504,7 +505,7 @@ class Mvp:
         self.shangju_db.add_some(self.sql_11, infos)
 
     def query_data(self):
-        ids = self.shangju_db.select(self.sql_14)
+        ids = self.linshi_db.select(self.sql_14)
         return ids
 
     def shanghai_85_module_score_insert(self):

mysql_db.py

@@ -8,12 +8,14 @@ class MysqlDB:
     con = None
     cursor = None
 
-    def __init__(self, db_name):
+    def __init__(self, db_name, db_type=None):
         self.db_name = db_name
-        self.con = ps.connect(host='172.19.189.136', port=3306, user='bi_etl', password='XPtpswuU5lwGo4kx',
-                              db=self.db_name, charset='utf8')
-        # self.con = ps.connect(host='192.168.0.13', port=3306, user='root', password='elab@123'
-        #                       , db=self.db_name, charset='utf8')
+        if db_type is None:
+            self.con = ps.connect(host='172.19.189.136', port=3306, user='bi_etl', password='XPtpswuU5lwGo4kx',
+                                  db=self.db_name, charset='utf8')
+        else:
+            self.con = ps.connect(host='172.19.189.121', port=3306, user='root', password='elab@123'
+                                  , db=self.db_name, charset='utf8')
         self.cursor = self.con.cursor()
 
     def show_tables(self):