sca module complete

Author: SpenserCai

README.md

@@ -3,7 +3,7 @@
  * @Date: 2021-12-23 14:25:35
  * @version: 
  * @LastEditors: 饕餮
- * @LastEditTime: 2021-12-25 21:32:18
+ * @LastEditTime: 2021-12-27 09:59:59
  * @Description: How to use
 -->
 # DongTai-SDK-Python
@@ -62,8 +62,9 @@ dongTaiSdk.GetAgentDetail(agentId)
 dongTaiSdk.GetAgentList(page=1,pageSize=50,projectName=None,state=None,token=None)
 ```
 
-### Sca
+### Sca (Completed)
 ```python
 dongTaiSdk.GetScaDetail(scaId)
 dongTaiSdk.GetScaList(page=1,pageSize=50,keyword=None,language=None,level=None,order=None,projectId=None,projectName=None,versionId=None)
+dongTaiSdk.GetScaSummary(page=1,pageSize=50,keyword=None,language=None,level=None,order=None,projectId=None,projectName=None,versionId=None)
 ```

dongtai_sdk/DongTai.py

@@ -3,12 +3,12 @@
 Date: 2021-12-23 15:10:01
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-25 21:32:29
+LastEditTime: 2021-12-27 09:59:24
 Description: Main
 '''
 from .base.DongTaiProject import DongTaiProject,DongTaiProjectVersion
 from .base.DongTaiAgent import DongTaiAgent
-from .base.DongTaiSca import DongTaiSca
+from .base.DongTaiSca import DongTaiSca, ScaSummary
 from .DongTaiApi import DongTaiApi
 from .base.BaseObejct import DongTaiError
 
@@ -193,4 +193,12 @@ def GetScaDetail(self,scaId):
             errorObject = DongTaiError(errorMsg)
             return errorObject
 
-    
+    def GetScaSummary(self,page=1,pageSize=50,keyword=None,language=None,level=None,order=None,projectId=None,projectName=None,versionId=None):
+        repData = self.dongTaiApi.GetScaSummary(page,pageSize,keyword,language,level,order,projectId,projectName,versionId)
+        if repData["status"] == 201:
+            scaSummary = ScaSummary(repData["data"])
+            return scaSummary
+        else:
+            errorMsg = {"status":repData["status"],"msg":repData["msg"]}
+            errorObject = DongTaiError(errorMsg)
+            return errorObject

dongtai_sdk/base/BaseObejct.py

@@ -3,7 +3,7 @@
 Date: 2021-12-23 14:49:05
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-23 21:05:50
+LastEditTime: 2021-12-27 10:01:53
 Description: 通用基类
 '''
 import json
@@ -61,4 +61,20 @@ def Status(self):
 
     @property
     def ErrorMsg(self):
-        return self.TryGetValue("msg")
+        return self.TryGetValue("msg")
+
+class ProjectSummary(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+    
+    @property
+    def ProjectName(self):
+        return self.TryGetValue("project_name")
+
+    @property
+    def Count(self):
+        return self.TryGetValue("count")
+
+    @property
+    def Id(self):
+        return self.TryGetValue("id")

dongtai_sdk/base/DongTaiSca.py

@@ -3,11 +3,11 @@
 Date: 2021-12-24 10:57:03
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-25 21:21:59
+LastEditTime: 2021-12-27 10:02:24
 Description: Sca Object
 '''
 import json
-from .BaseObejct import BaseObject
+from .BaseObejct import BaseObject,ProjectSummary
 
 class ScaVulnReference(BaseObject):
     def __init__(self,jsonData):
@@ -64,6 +64,34 @@ def Reference(self):
                 returnData.append(tmpObject)
         return returnData
 
+class ScaLanguage(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+
+    @property
+    def Language(self):
+        return self.TryGetValue("language")
+
+    @property
+    def Count(self):
+        return self.TryGetValue("count")
+
+class ScaLevel(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+
+    @property
+    def Level(self):
+        return self.TryGetValue("level")
+
+    @property
+    def Count(self):
+        return self.TryGetValue("count")
+
+    @property
+    def LevelId(self):
+        return self.TryGetValue("level_id")
+
 class DongTaiSca(BaseObject):
     def __init__(self,jsonData):
         self.ObjectData = jsonData
@@ -128,4 +156,38 @@ def Vuls(self):
             for tmpData in tmpDataList:
                 tmpObject = ScaVuln(tmpData)
                 returnData.append(tmpObject)
+        return returnData
+
+class ScaSummary(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+
+    @property
+    def Language(self):
+        returnData = []
+        tmpDataList = self.TryGetValue("language")
+        if tmpDataList is not None:
+            for tmpData in tmpDataList:
+                tmpObject = ScaLanguage(tmpData)
+                returnData.append(tmpObject)
+        return returnData
+
+    @property
+    def Level(self):
+        returnData = []
+        tmpDataList = self.TryGetValue("level")
+        if tmpDataList is not None:
+            for tmpData in tmpDataList:
+                tmpObject = ScaLevel(tmpData)
+                returnData.append(tmpObject)
+        return returnData
+
+    @property
+    def Projects(self):
+        returnData = []
+        tmpDataList = self.TryGetValue("projects")
+        if tmpDataList is not None:
+            for tmpData in tmpDataList:
+                tmpObject = ProjectSummary(tmpData)
+                returnData.append(tmpObject)
         return returnData