update

db_helper.py

@@ -246,11 +246,6 @@ class MysqlHelper(object):
         初始化mysql连接信息
         """
         self.mysql_info = config_.MYSQL_INFO
-        # self.host = mysql_info['host']
-        # self.port = mysql_info['port']
-        # self.user = mysql_info['user']
-        # self.password = mysql_info['password']
-        # self.db = mysql_info['db']
 
     def get_data(self, sql):
         """
@@ -260,14 +255,6 @@ class MysqlHelper(object):
         """
         # 连接数据库
         conn = pymysql.connect(**self.mysql_info)
-        # conn = pymysql.connect(
-        #     host=self.host,
-        #     port=self.port,
-        #     user=self.user,
-        #     password=self.password,
-        #     db=self.db,
-        #     charset='utf8'
-        # )
         # 创建游标
         cursor = conn.cursor()
         try:

rov_train.py

@@ -175,16 +175,8 @@ def predict():
 def predict_test():
     """测试环境数据生成"""
     # 获取测试环境中最近发布的40000条视频
-    # mysql_info = {
-    #     'host': 'rm-bp1k5853td1r25g3n690.mysql.rds.aliyuncs.com',
-    #     'port': 3306,
-    #     'user': 'wx2016_longvideo',
-    #     'password': 'wx2016_longvideoP@assword1234',
-    #     'db': 'longvideo'
-    # }
     sql = "SELECT id FROM wx_video ORDER BY id DESC LIMIT 40000;"
     mysql_helper = MysqlHelper()
-    # mysql_helper = MysqlHelper(mysql_info=mysql_info)
     data = mysql_helper.get_data(sql=sql)
     video_ids = [video[0] for video in data]
     # 视频状态过滤

utils.py

@@ -183,8 +183,8 @@ def get_video_w_h_rate(video_ids):
 
     # 更新到redis
     info_data = {}
-    for id, w_h_rate in data:
-        info_data[int(id)] = float(w_h_rate)
+    for video_id, w_h_rate in data:
+        info_data[int(video_id)] = float(w_h_rate)
     redis_helper = RedisHelper()
     # 删除旧数据
     redis_helper.del_keys(key_name=config_.W_H_RATE_UP_1_VIDEO_LIST_KEY_NAME)