modify delete py file

delete_temp_csv.py → delete_temp_files.py

-
-
 import os
 import time
 from config import *
@@ -8,7 +6,9 @@ from config import *
 # 定期删除特定文件夹内特征的文件
 def remove_files(fileDir):
     for eachFile in os.listdir(fileDir):
-        if (os.path.isfile(fileDir + "/" + eachFile)) and ("DiaryTop3000.csv" in eachFile):
+        condition_a = os.path.isfile(fileDir + "/" + eachFile)
+        condition_b = ("DiaryTop3000.csv" in eachFile) or ("output.txt" in eachFile) or ("feed" in eachFile)
+        if condition_a and condition_b:
             ft = os.stat(fileDir + "/" + eachFile)
             ltime = int(ft.st_mtime)
             # 删除5分钟前的文件

predictDiary.py

@@ -52,7 +52,7 @@ def predict(user_profile):
     param = {"log": "/data2/models/result"}
     ffm_model.predict(param,DIRECTORY_PATH + "model.out",
                       DIRECTORY_PATH + "result/{0}_output.txt".format(user_profile['device_id']))
-    print("预测结束")
+    print("该用户预测结束")
     predict_save_to_local(user_profile, instance)
 
     #TODO 没有提供生产环境的redis地址，所以这个函数先不运行