change test file

rerank_esmm.py

@@ -76,57 +76,61 @@ def get_queues(device_id,city_id):
 def tag_boost(cid_str, tag_list=[15,21,22,85,86]):
     if cid_str is not None or cid_str != "":
         cids = cid_str.split(",")
-        if len(cids) > 6 and len(tag_list) > 0:
-            db = pymysql.connect(host='172.16.40.158', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='eagle')
-
-            sql = "select id,group_concat(diary_id) from " \
-                  "(select a.diary_id,b.id from src_mimas_prod_api_diary_tags a left join src_zhengxing_api_tag b " \
-                  "on a.tag_id = b.id where b.tag_type < '4' and a.diary_id in {}) tmp " \
-                  "where id in {} group by id".format(tuple(cids), tuple(tag_list))
-            cursor = db.cursor()
-            cursor.execute(sql)
-            result = cursor.fetchall()
-            if len(result) > 0:
-                tag_cids = {}
-                left_cids = []
-                for i in result:
-                    tmp = i[1].split(",")
-                    tag_cids[i[0]] = tmp
-                    left_cids.extend(tmp)
-
-                left_cids = list(set(left_cids))
-                right_cids = [i for i in cids if i not in left_cids]
-                tag_cids["right"] = right_cids
-                tag_list.append("right")
-                sort_cids = []
-                n = 0
-                while n != len(tag_cids)-1:
-                    for i in tag_list:
-                        if i in tag_cids.keys():
-                            if len(tag_cids[i]) > 0:
-                                sort_cids.append(tag_cids[i][0])
-                                value = tag_cids[i]
-                                value.pop(0)
-                                tag_cids[i] = value
-                                if len(value) == 0 and i != "right":
-                                    n = n + 1
-
-                if len(tag_cids["right"]) > 0:
-                    sort_cids.extend(tag_cids["right"])
-
-                news_ids = []
-                for id in sort_cids:
-                    if id not in news_ids:
-                        news_ids.append(id)
-
-                new_str = ",".join([str(i) for i in news_ids])
-                return new_str
-
-
+        try:
+            if len(cids) > 6 and len(tag_list) > 0:
+                db = pymysql.connect(host='172.16.40.158', port=4000, user='root', passwd='3SYz54LS9#^9sBvC',
+                                     db='eagle')
+
+                sql = "select id,group_concat(diary_id) from " \
+                      "(select a.diary_id,b.id from src_mimas_prod_api_diary_tags a left join src_zhengxing_api_tag b " \
+                      "on a.tag_id = b.id where b.tag_type < '4' and a.diary_id in {}) tmp " \
+                      "where id in {} group by id".format(tuple(cids), tuple(tag_list))
+                cursor = db.cursor()
+                cursor.execute(sql)
+                result = cursor.fetchall()
+                if len(result) > 0:
+                    tag_cids = {}
+                    left_cids = []
+                    for i in result:
+                        tmp = i[1].split(",")
+                        tag_cids[i[0]] = tmp
+                        left_cids.extend(tmp)
+
+                    left_cids = list(set(left_cids))
+                    right_cids = [i for i in cids if i not in left_cids]
+                    tag_cids["right"] = right_cids
+                    tag_list.append("right")
+                    sort_cids = []
+                    n = 0
+                    while n != len(tag_cids) - 1:
+                        for i in tag_list:
+                            if i in tag_cids.keys():
+                                if len(tag_cids[i]) > 0:
+                                    sort_cids.append(tag_cids[i][0])
+                                    value = tag_cids[i]
+                                    value.pop(0)
+                                    tag_cids[i] = value
+                                    if len(value) == 0 and i != "right":
+                                        n = n + 1
+
+                    if len(tag_cids["right"]) > 0:
+                        sort_cids.extend(tag_cids["right"])
+
+                    news_ids = []
+                    for id in sort_cids:
+                        if id not in news_ids:
+                            news_ids.append(id)
+
+                    new_str = ",".join([str(i) for i in news_ids])
+                    return new_str
+
+                else:
+                    return cid_str
             else:
                 return cid_str
-        else:
+        except:
             return cid_str
+
     else:
         return cid_str