debug

relevant_top_videos.py

@@ -9,7 +9,8 @@ config_, _ = set_config()
 
 def get_relevant_videos_with_excel():
     filepath = './data/relevant.xlsx'
-    df = pd.read_excel(filepath)
+    # startTime 和 finishTime 的数据类型指定为str
+    df = pd.read_excel(filepath, converters={'startTime': str, 'finishTime': str})
     relevant_videos = {}
     head_videos = []
     for i in range(len(df)):
@@ -17,8 +18,9 @@ def get_relevant_videos_with_excel():
         head_vid = int(df.iloc[i]['headVid'])
         order = int(df.iloc[i]['order'])
         recommend_vid = int(df.iloc[i]['recommendVid'])
-        start_time = int(df.iloc[i]['startTime'].timestamp())
-        finish_time = int(df.iloc[i]['finishTime'].timestamp())
+        # 将时间转换为10位时间戳
+        start_time = int(time.mktime(time.strptime(df.iloc[i]['startTime'].split(r'.')[0], '%Y-%m-%d %H:%M:%S')))
+        finish_time = int(time.mktime(time.strptime(df.iloc[i]['finishTime'].split(r'.')[0], '%Y-%m-%d %H:%M:%S')))
         item = {
             'order': order,
             'recommend_vid': recommend_vid,
@@ -51,6 +53,7 @@ def update_relevant_videos_to_redis(relevant_videos):
         # 存入redis
         redis_helper = RedisHelper()
         redis_helper.set_data_to_redis(key_name=key_name, value=videos_json, expire_time=expire_time)
+        print('head_vid = {} relevant videos update finished!'.format(head_vid))
 
 
 if __name__ == '__main__':