Merge branch 'main' of https://github.com/dataease/SQLBot

Author: dataeaseShu

backend/apps/datasource/models/datasource.py

@@ -99,7 +99,7 @@ def to_dict(self):
 class TableSchema:
     def __init__(self, attr1, attr2):
         self.tableName = attr1
-        self.tableComment = attr2 if isinstance(attr2, str) else attr2.decode("utf-8")
+        self.tableComment = attr2 if attr2 is None or isinstance(attr2, str) else attr2.decode("utf-8")
 
     tableName: str
     tableComment: str
@@ -109,7 +109,7 @@ class ColumnSchema:
     def __init__(self, attr1, attr2, attr3):
         self.fieldName = attr1
         self.fieldType = attr2
-        self.fieldComment = attr3 if isinstance(attr3, str) else attr3.decode("utf-8")
+        self.fieldComment = attr3 if attr3 is None or isinstance(attr3, str) else attr3.decode("utf-8")
 
     fieldName: str
     fieldType: str