change sql for test

diaryQueueUpdate.py

@@ -238,7 +238,8 @@ if __name__ == "__main__":
             old_device_id_list = pd.read_csv(DIRECTORY_PATH + "data_set_device_id.csv")["device_id"].values.tolist()
             # 求活跃用户和老用户的交集，也就是只预测老用户
             predict_list = list(set(device_id_list) & set(old_device_id_list))
-
+            # 只预测尾号是6的ID，这块也可以在数据库取数据时过滤一下
+            predict_list = list(filter(lambda x:str(x)[-1] == "6", predict_list))
             start = time.time()
             warnings.filterwarnings("ignore")
             data_set_cid = pd.read_csv(DIRECTORY_PATH + "data_set_cid.csv")["cid"].values.tolist()

userProfile.py

@@ -7,8 +7,9 @@ def get_active_users():
     now = datetime.now()
     now_start = str(now)[:16] + ":00"
     now_end = str(now)[:16] + ":59"
-    sql = "select device_id from user_active_time " \
-          "where active_time <= '{}' and active_time >= '{}'".format(now_end,now_start)
+    sql = "select device_id from user_active_time order by active_time desc limit 50;"
+    # sql = "select device_id from user_active_time " \
+    #       "where active_time <= '{}' and active_time >= '{}'".format(now_end,now_start)
     device_id_df = con_sql(sql)
     if device_id_df.empty:
         print("当下这一分钟没有活跃用户，不需要预测")