词集接口增加错误处理

api/tag.py

@@ -6,6 +6,7 @@
 import json
 from utils.base import APIView, get_offset_count
 from utils.logger import error_logger
+from alpha_types.venus.error import ERROR
 
 
 class TagListView(APIView):
@@ -48,7 +49,7 @@ class TagUpdateOrCreateView(APIView):
         return {'data': data}
 
     def post(self, request):
-        
+
         id = request.POST.get('id')
         down_tags = list(set(map(lambda x: x.split(":")[0], json.loads((request.POST.get('down_tags', '[]'))))))
         up_tags = list(set(map(lambda x: x.split(":")[0], json.loads((request.POST.get('up_tags', '[]'))))))
@@ -273,8 +274,9 @@ class TagSetDeleteView(APIView):
             error_logger.error(u'删除词集失败%s', e)
             raise
         return {
-            "status": 1
-        }
+                "message": "删除成功",
+                "code": 200
+            }
 
 
 class TagSetUpdateOrCreateView(APIView):
@@ -288,17 +290,31 @@ class TagSetUpdateOrCreateView(APIView):
             try:
                 data = self.rpc['venus/sun/tag_set/update'](tag_set_id=tag_set_id, name=name, tag_ids=tag_ids).unwrap()
             except Exception as e:
-                error_logger.error(u'更新词集失败%s', e)
-                raise
+                if e.error == ERROR.TAG_SET_HAD_EXIST:
+                    return {
+                        'message': e.message,
+                        'code': 500
+                    }
+                else:
+                    error_logger.error(u'更新词集失败%s', e)
+                    raise
             return {
-                "status": 1
+                "message": "更新成功",
+                "code": 200
             }
         else:
             try:
                 data = self.rpc['venus/sun/tag_set/create'](name=name, tag_ids=tag_ids).unwrap()
             except Exception as e:
-                error_logger.error(u'创建词集失败%s', e)
-                raise
+                if e.error == ERROR.TAG_SET_HAD_EXIST:
+                    return {
+                        'message': e.message,
+                        'code': 500
+                    }
+                else:
+                    error_logger.error(u'创建词集失败%s', e)
+                    raise
             return {
-                "status": 1
+                "message": "创建成功",
+                "code": 200
             }