support vuln

Author: SpenserCai

README.md

@@ -3,7 +3,7 @@
  * @Date: 2021-12-23 14:25:35
  * @version: 
  * @LastEditors: 饕餮
- * @LastEditTime: 2021-12-27 09:59:59
+ * @LastEditTime: 2021-12-27 18:36:54
  * @Description: How to use
 -->
 # DongTai-SDK-Python
@@ -68,3 +68,8 @@ 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)
 ```
+
+### Vuln
+```python
+dongTaiSdk.GetVulnSummary(projectId=None,language=None,level=None,order=None,projectName=None,status=None,type=None,url=None,versionId=None,statusId=None)
+```

dongtai_sdk/DongTai.py

@@ -3,9 +3,10 @@
 Date: 2021-12-23 15:10:01
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-27 09:59:24
+LastEditTime: 2021-12-27 18:35:26
 Description: Main
 '''
+from .base.DongTaiVuln import VulnSummary
 from .base.DongTaiProject import DongTaiProject,DongTaiProjectVersion
 from .base.DongTaiAgent import DongTaiAgent
 from .base.DongTaiSca import DongTaiSca, ScaSummary
@@ -202,3 +203,15 @@ def GetScaSummary(self,page=1,pageSize=50,keyword=None,language=None,level=None,
             errorMsg = {"status":repData["status"],"msg":repData["msg"]}
             errorObject = DongTaiError(errorMsg)
             return errorObject
+
+    #[Vuln Function]
+    def GetVulnSummary(self,projectId=None,language=None,level=None,order=None,projectName=None,status=None,type=None,url=None,versionId=None,statusId=None):
+        repData = self.dongTaiApi.GetVulnSummary(projectId,language,level,order,projectName,status,type,url,versionId,statusId)
+        if repData["status"] == 201:
+            vulnSummary = VulnSummary(repData["data"])
+            return vulnSummary
+        else:
+            errorMsg = {"status":repData["status"],"msg":repData["msg"]}
+            errorObject = DongTaiError(errorMsg)
+            return errorObject
+    

dongtai_sdk/DongTaiApi.py

@@ -3,7 +3,7 @@
 Date: 2021-12-23 14:27:00
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-27 13:15:27
+LastEditTime: 2021-12-27 17:03:50
 Description: file content
 '''
 import json,requests
@@ -187,11 +187,12 @@ def GetScaDetail(self,scaId):
 
     #[Vuln Function]
     #漏洞概览
-    def GetVulnSummary(self,projectId,language=None,level=None,order=None,projectName=None,status=None,type=None,url=None,versionId=None,statusId=1):
-        data = {
-            "project_id":projectId,
-            "status_id":statusId
-        }
+    def GetVulnSummary(self,projectId=None,language=None,level=None,order=None,projectName=None,status=None,type=None,url=None,versionId=None,statusId=None):
+        data = {}
+        if projectId is not None:
+            data["project_id"] = projectId
+        if statusId is not None:
+            data["status_id"] = statusId
         if language is not None:
             data["language"] = language
         if level is not None:

dongtai_sdk/base/DongTaiVuln.py

@@ -3,6 +3,105 @@
 Date: 2021-12-24 10:56:10
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-24 10:58:36
+LastEditTime: 2021-12-27 18:34:21
 Description: Vuln Object
 '''
+import json
+from typing import List
+from .BaseObejct import BaseObject,ProjectSummary,LanguageSummary
+
+class VulnType(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+
+    @property
+    def Type(self):
+        return self.TryGetValue("type")
+
+    @property
+    def Count(self):
+        return self.TryGetValue("count")
+
+class VulnLevel(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 Id(self):
+        return self.TryGetValue("level_id")
+
+class VulnSummary(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+
+    @property
+    def Language(self) -> List[LanguageSummary]:
+        returnData = []
+        tmpDataList = self.TryGetValue("language")
+        for tmpData in tmpDataList:
+            tmpObject = LanguageSummary(tmpData)
+            returnData.append(tmpObject)
+        return returnData
+        
+    @property
+    def Level(self) -> List[VulnLevel]:
+        returnData = []
+        tmpDataList = self.TryGetValue("level")
+        for tmpData in tmpDataList:
+            tmpObject = VulnLevel(tmpData)
+            returnData.append(tmpObject)
+        return returnData
+
+    @property
+    def Projects(self) -> List[ProjectSummary]:
+        returnData = []
+        tmpDataList = self.TryGetValue("projects")
+        for tmpData in tmpDataList:
+            tmpObject = ProjectSummary(tmpData)
+            returnData.append(tmpObject)
+        return returnData
+
+    @property
+    def HighRisk(self):
+        tmpData = [tmpLevel for tmpLevel in self.Level if tmpLevel.Level == "高危"]
+        if len(tmpData) > 0:
+            return tmpData[0]
+        else:
+            return None
+
+    @property
+    def MediumRisk(self):
+        tmpData = [tmpLevel for tmpLevel in self.Level if tmpLevel.Level == "中危"]
+        if len(tmpData) > 0:
+            return tmpData[0]
+        else:
+            return None
+
+    @property
+    def LowRisk(self):
+        tmpData = [tmpLevel for tmpLevel in self.Level if tmpLevel.Level == "低危"]
+        if len(tmpData) > 0:
+            return tmpData[0]
+        else:
+            return None
+
+    @property
+    def Count(self):
+        tmpCount = 0
+        for tmpLevel in self.Level:
+            if tmpLevel.Level not in ["无风险","提示"]:
+                tmpCount += tmpLevel.Count
+        return tmpCount
+
+
+    
+
+