Browse Source

mvp: 关闭数据库连接

Signed-off-by: binren <zhangbr@elab-plus.com>
binren 5 years ago
parent
commit
a09995e9f8
1 changed files with 2 additions and 1 deletions
  1. 2 1
      mvp.py

+ 2 - 1
mvp.py

@@ -327,6 +327,7 @@ class Mvp:
     def __init__(self, path=None):
         self.shangju_db = MysqlDB('shangju')
         self.marketing_db = MysqlDB('bi_report')
+        self.linshi_db = MysqlDB('linshi', db_type=1)
         # self.shangju_db.truncate('mvp_standard_score')
         self.tag_data = ExcelUtil(file_name=path).init_mvp_data()
         self.crowd_info = ExcelUtil(file_name=path, sheet_name='选项-人群分类对应表').init_crowd_info()
@@ -344,6 +345,7 @@ class Mvp:
     def close(self):
         self.shangju_db.close()
         self.marketing_db.close()
+        self.linshi_db.close()
 
     def init_city(self):
         """
@@ -459,7 +461,6 @@ class Mvp:
         :return:
         """
         self.insert_table = []
-        self.linshi_db = MysqlDB('linshi', db_type=1)
         self.ids = self.query_data()
         for city in self.city_list:
             for age in self.age_list: