change test file

monitor.py

@@ -189,29 +189,32 @@ def diary_write(device_id,cid):
         cursor = db.cursor()
         cursor.execute(sql)
         result = cursor.fetchall()
-        tags = result[0][0]
-        if tags is not None:
-            sql = "select a.id from src_mimas_prod_api_diary a left join src_mimas_prod_api_diary_tags b " \
-                  "on a.id=b.diary_id left join src_zhengxing_api_tag c on b.tag_id=c.id " \
-                  "where a.is_online = 1 and a.content_level >= '3' " \
-                  "and c.id = {} and c.tag_type = '3'".format(tags)
-            cursor.execute(sql)
-            result = cursor.fetchall()
-            db.close()
-            cids = list(set([i[0] for i in result]))
-            if len(cids) != 0:
-
-                r = redis.Redis(host="172.16.40.135", port=5379, password="")
-                key = str(device_id) + "_dislike_diary"
-                if r.exists(key):
-                    value = eval(r.get(key)).split(",")
-                    value.extend(cids)
-                    cids = json.dumps(list(set(value)))
-                    r.set(key, json.dumps(cids))
-                else:
-                    r.set(key, json.dumps(cids))
-                    r.expire(key, 7*24*60*60)
-        return "diary good"
+        if len(result) > 0:
+            tags = result[0][0]
+            if tags is not None:
+                sql = "select a.id from src_mimas_prod_api_diary a left join src_mimas_prod_api_diary_tags b " \
+                      "on a.id=b.diary_id left join src_zhengxing_api_tag c on b.tag_id=c.id " \
+                      "where a.is_online = 1 and a.content_level >= '3' " \
+                      "and c.id = {} and c.tag_type = '3'".format(tags)
+                cursor.execute(sql)
+                result = cursor.fetchall()
+                db.close()
+                cids = [str(i[0]) for i in result]
+                if len(cids) != 0:
+
+                    r = redis.Redis(host="172.16.40.135", port=5379, password="")
+                    key = str(device_id) + "_dislike_diary"
+                    if r.exists(key):
+                        value = eval(r.get(key)).split(",")
+                        value.extend(cids)
+                        value = ",".join([str(i) for i in set(value)])
+                        cids = json.dumps(value)
+                        r.set(key, cids)
+
+                    else:
+                        r.set(key, json.dumps(",".join(cids)))
+                        r.expire(key, 7 * 24 * 60 * 60)
+            return "diary good"
 
     except Exception as e:
         print("diary insert redis fail")
@@ -237,7 +240,7 @@ def diary_write(device_id,cid):
 # print(tractate_write("hello","10078"))
 
 
-print(question_write("hello",188472))
-print(question_write("hello","188472"))
+print(diary_write("hello",16951048))
+print(diary_write("hello","16951010"))