add test

diaryUpdateOnlineOffline.py

@@ -291,7 +291,12 @@ def multi_proecess_update(device_id, city_id, data_set_cid,total_video_id):
 
 if __name__ == "__main__":
     warnings.filterwarnings("ignore")
-
+    flag = True
+    path = DIRECTORY_PATH
+    # 下面这个ip是本地电脑ip
+    if socket.gethostbyname(socket.gethostname()) == '172.30.8.160':
+        flag = False
+        path = LOCAL_DIRCTORY
     # 增加缓存日记视频列表
     cache_video_id = []
     cache_device_city_list = []

local/testCases.py

@@ -59,26 +59,30 @@ def save_queue():
         df.to_csv("/data/models/{}.csv".format(i), index=None)
         print("end")
 
-# def insert():
-#     queue_name_list = ["native_queue", "nearby_queue", "nation_queue", "megacity_queue"]
-#     a = {}
-#     for i in queue_name_list:
-#         a[i] = pd.read_csv("/data/models/native_queue.csv")["0"].values.tolist()[0]
-#
-#         db = pymysql.connect(host='rm-m5e842126ng59jrv6.mysql.rds.aliyuncs.com', port=3306, user='doris',
-#                              passwd='o5gbA27hXHHm',
-#                              db='doris_prod')
-#         sql = "update device_diary_queue set {}='{}' where device_id = '{}' and city_id = '{}'".format \
-#             (i, a[i], device_id, city_id)
-#         cursor = db.cursor()
-#         cursor.execute(sql)
-#         db.commit()
-#         db.close()
-#         print("end")
+def insert(id,device_id,city_id):
+    queue_name_list = ["native_queue", "nearby_queue", "nation_queue", "megacity_queue"]
+    a = {}
+    for i in queue_name_list:
+        a[i] = pd.read_csv("/data/models/native_queue.csv")["0"].values.tolist()[0]
+    # columns = ["native_queue", "nearby_queue", "nation_queue", "megacity_queue","id","device_id","city_id"]
+    db = pymysql.connect(host='rm-m5e842126ng59jrv6.mysql.rds.aliyuncs.com', port=3306, user='doris',
+                         passwd='o5gbA27hXHHm',
+                         db='doris_prod')
+    sql = "INSERT INTO device_diary_queue (native_queue, nearby_queue, nation_queue, " \
+          "megacity_queue,id,device_id,city_id) VALUES ('{}','{}','{}','{}','{}','{}','{}');".format \
+        (a["native_queue"], a["nearby_queue"],a["nation_queue"], a["megacity_queue"],id,device_id, city_id)
+    cursor = db.cursor()
+    cursor.execute(sql)
+    db.commit()
+    db.close()
+    print("end")
 
 
 if __name__=="__main__":
-    save_queue()
+    id = 334
+    device_id = '00CA20EB-2719-4518-85CC-60E765AC526F'
+    city_id = 'beijing'
+    insert(id, device_id, city_id)