change test file

rerank_esmm.py

@@ -157,21 +157,34 @@ def to_data_base(df2):
     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)
+    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")
+    # 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)
 
 
 
@@ -180,7 +193,7 @@ if __name__ == "__main__":
     total_samples = list()
     name_tag = get_searchworlds_to_tagid()
     # TODO 把下面的截断改掉
-    for i in users_list[:6]:
+    for i in users_list[:60]:
         tag_list = get_user_profile(i[0])
         queues = get_queues(i[0],i[1])
         if len(queues) > 0 and len(tag_list) > 0:
@@ -195,6 +208,7 @@ if __name__ == "__main__":
         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:"time"})
+        print(df.head(2))
         to_data_base(df)
         print("good boy")
     # to_data_base(df)