Commit 81921d86 authored by 张彦钊's avatar 张彦钊

add print

parent d707df38
...@@ -22,7 +22,7 @@ def test_con_sql(device_id): ...@@ -22,7 +22,7 @@ def test_con_sql(device_id):
result = cursor.fetchall() result = cursor.fetchall()
df = pd.DataFrame(list(result)) df = pd.DataFrame(list(result))
if not df.empty: if not df.empty:
data_set_cid = pd.read_csv(DIRECTORY_PATH + "data_set_cid.csv")["cid"].values.tolist()
df = df.rename(columns={0: "native_queue", 1: "nearby_queue", 2: "nation_queue", 3: "megacity_queue"}) df = df.rename(columns={0: "native_queue", 1: "nearby_queue", 2: "nation_queue", 3: "megacity_queue"})
native_queue = df.loc[0, "native_queue"].split(",") native_queue = df.loc[0, "native_queue"].split(",")
...@@ -61,6 +61,7 @@ def get_native_queue(device_id): ...@@ -61,6 +61,7 @@ def get_native_queue(device_id):
if not df.empty: if not df.empty:
native_queue = df.loc[0,0].split(",") native_queue = df.loc[0,0].split(",")
native_queue = list(map(lambda x:"diary|"+str(x),native_queue)) native_queue = list(map(lambda x:"diary|"+str(x),native_queue))
native_queue = list(set(native_queue) & set(data_set_cid))
db.close() db.close()
print("成功获取native_queue") print("成功获取native_queue")
return native_queue return native_queue
...@@ -201,8 +202,7 @@ def update_sql_dairy_queue(queue_name, diary_id): ...@@ -201,8 +202,7 @@ def update_sql_dairy_queue(queue_name, diary_id):
def multi_update(key, name_dict): def multi_update(key, name_dict):
diary_id = predict(key, name_dict[key]) diary_id = predict(key, name_dict[key])
print(get_native_queue(device_id))
print(native_queue_list)
if get_native_queue(device_id) == native_queue_list: if get_native_queue(device_id) == native_queue_list:
update_sql_dairy_queue(key, diary_id) update_sql_dairy_queue(key, diary_id)
print("更新结束") print("更新结束")
...@@ -218,6 +218,7 @@ if __name__ == "__main__": ...@@ -218,6 +218,7 @@ if __name__ == "__main__":
native_queue_list, nearby_queue_list, nation_queue_list, megacity_queue_list = test_con_sql(device_id) native_queue_list, nearby_queue_list, nation_queue_list, megacity_queue_list = test_con_sql(device_id)
name_dict = {"native_queue": native_queue_list, "nearby_queue": nearby_queue_list, name_dict = {"native_queue": native_queue_list, "nearby_queue": nearby_queue_list,
"nation_queue": nation_queue_list, "megacity_queue": megacity_queue_list} "nation_queue": nation_queue_list, "megacity_queue": megacity_queue_list}
data_set_cid = pd.read_csv(DIRECTORY_PATH + "data_set_cid.csv")["cid"].values.tolist()
for key in name_dict.keys(): for key in name_dict.keys():
multi_update(key, name_dict) multi_update(key, name_dict)
# pool = Pool(4) # pool = Pool(4)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment