change test file

rerank_esmm.py

@@ -152,8 +152,26 @@ def tag_boost(cid_str, tag_list):
 
 
 
-def to_data_base(df,table_name = "tag_boost_device_diary_queue"):
-    pass
+def to_data_base(df2):
+    device_count = df2.shape[0]
+    con = pymysql.connect(host='172.16.40.158', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test',
+                          charset='utf8')
+    cur = con.cursor()
+    try:
+        for i in range(0, device_count):
+            query = """INSERT INTO esmm_resort_diary_queue (device_id, city_id, native_queue,nearby_queue,
+            nation_queue,megacity_queue,time) VALUES('%s', '%s', '%s', '%s',%s', '%s', '%s') \
+                ON DUPLICATE KEY UPDATE device_id='%s', city_id='%s', native_queue='%s', 
+                nearby_queue='%s',nation_queue='%s', megacity_queue='%s',time='%s'""" % (
+            df2.device_id[i], df2.city_id[i],df2.native_queue[i], df2.nearby_queue[i],df2.nation_queue[i],
+            df2.megacity_queue[i],df2.time[i],df2.device_id[i], df2.city_id[i],df2.native_queue[i], df2.nearby_queue[i],df2.nation_queue[i],
+            df2.megacity_queue[i],df2.time[i])
+            cur.execute(query)
+            con.commit()
+        con.close()
+        print("insert or update sucess")
+    except Exception as e:
+        print(e)
 
 
 
@@ -161,8 +179,6 @@ if __name__ == "__main__":
     users_list = get_esmm_users()
     total_samples = list()
     name_tag = get_searchworlds_to_tagid()
-    print(users_list[:6])
-    print(name_tag)
     # TODO 把下面的截断改掉
     for i in users_list[:6]:
         tag_list = get_user_profile(i[0])
@@ -171,15 +187,16 @@ if __name__ == "__main__":
             new_native = tag_boost(queues[0], tag_list)
             new_nearby = tag_boost(queues[1], tag_list)
 
-            sample = [i[0],i[1],new_native,new_nearby,queues[2],queues[3],int(time.time())]
+            insert_time = str(datetime.datetime.now().strftime('%Y%m%d%H%M'))
+            sample = [i[0],i[1],new_native,new_nearby,queues[2],queues[3],insert_time]
             total_samples.append(sample)
 
     if len(total_samples) > 0:
         df = pd.DataFrame(total_samples)
         df = df.rename(columns={0: "device_id", 1: "city_id",2:"native_queue",
-                                3:"nearby_queue",4:"nation_queue",5:", megacity_queue",6:"insert_time"})
-        print(total_samples[:2])
-        print(df.head(2))
+                                3:"nearby_queue",4:"nation_queue",5:", megacity_queue",6:"time"})
+        to_data_base(df)
+        print("good boy")
     # to_data_base(df)
     # to kv