From 20ada50956af6a345516ba95fae01a910c5ff09d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E5=81=A5=E8=BE=89?=
 <40563566+jianhw@users.noreply.github.com>
Date: Tue, 25 Aug 2020 15:58:07 +0800
Subject: [PATCH] add try (#4656)

---
 seahub/api2/endpoints/markdown_lint.py | 35 ++++++++++++++++++--------
 1 file changed, 24 insertions(+), 11 deletions(-)

diff --git a/seahub/api2/endpoints/markdown_lint.py b/seahub/api2/endpoints/markdown_lint.py
index 61c57f9dcc..eba71f11f3 100644
--- a/seahub/api2/endpoints/markdown_lint.py
+++ b/seahub/api2/endpoints/markdown_lint.py
@@ -37,23 +37,36 @@ class MarkdownLintView(APIView):
             logger.error(e)
             error_msg = 'slate invalid.'
             return api_error(status.HTTP_400_BAD_REQUEST, error_msg)
+        try:
+            document_nodes = slate["document"]["nodes"]
+        except Exception as e:
+            logger.error(e)
+            error_msg = 'slate invalid.'
+            return api_error(status.HTTP_400_BAD_REQUEST, error_msg)
 
         issue_list = []
-        document_nodes = slate["document"]["nodes"]
-
         # check h1
-        heading_one_issue_list = check_heading_one(document_nodes)
-        if len(heading_one_issue_list) > 0:
-            issue_list.extend(heading_one_issue_list)
+        try:
+            heading_one_issue_list = check_heading_one(document_nodes)
+            if len(heading_one_issue_list) > 0:
+                issue_list.extend(heading_one_issue_list)
+        except Exception as e:
+            logger.error('check h1 error: %s' % e)
 
         # check heading_end_with
-        heading_end_issue_list = check_heading_end_with(document_nodes)
-        if len(heading_end_issue_list) > 0:
-            issue_list.extend(heading_end_issue_list)
+        try:
+            heading_end_issue_list = check_heading_end_with(document_nodes)
+            if len(heading_end_issue_list) > 0:
+                issue_list.extend(heading_end_issue_list)
+        except Exception as e:
+            logger.error('check heading_end_with error: %s' % e)
 
         # check heading_increase
-        heading_increase_issue_list = check_heading_increase(document_nodes)
-        if len(heading_increase_issue_list) > 0:
-            issue_list.extend(heading_increase_issue_list)
+        try:
+            heading_increase_issue_list = check_heading_increase(document_nodes)
+            if len(heading_increase_issue_list) > 0:
+                issue_list.extend(heading_increase_issue_list)
+        except Exception as e:
+            logger.error('check heading_increase error: %s' % e)
 
         return Response({"issue_list": issue_list}, status=status.HTTP_200_OK)