add GetVulnDetail

Author: SpenserCai

dongtai_sdk/DongTai.py

@@ -3,10 +3,10 @@
 Date: 2021-12-23 15:10:01
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-27 18:35:26
+LastEditTime: 2022-01-18 18:21:09
 Description: Main
 '''
-from .base.DongTaiVuln import VulnSummary
+from .base.DongTaiVuln import VulnDetail, VulnSummary
 from .base.DongTaiProject import DongTaiProject,DongTaiProjectVersion
 from .base.DongTaiAgent import DongTaiAgent
 from .base.DongTaiSca import DongTaiSca, ScaSummary
@@ -214,4 +214,14 @@ def GetVulnSummary(self,projectId=None,language=None,level=None,order=None,proje
             errorMsg = {"status":repData["status"],"msg":repData["msg"]}
             errorObject = DongTaiError(errorMsg)
             return errorObject
+
+    def GetVulnDetail(self,vulnId):
+        repData = self.dongTaiApi.GetVulnDetail(vulnId)
+        if repData["status"] == 201:
+            vulnDetail = VulnDetail(repData["data"])
+            return vulnDetail
+        else:
+            errorMsg = {"status":repData["status"],"msg":repData["msg"]}
+            errorObject = DongTaiError(errorMsg)
+            return errorObject
 

dongtai_sdk/base/DongTaiVuln.py

@@ -3,7 +3,7 @@
 Date: 2021-12-24 10:56:10
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-27 18:55:58
+LastEditTime: 2022-01-18 18:20:18
 Description: Vuln Object
 '''
 import json
@@ -38,6 +38,166 @@ def Count(self):
     def Id(self):
         return self.TryGetValue("level_id")
 
+class Vuln(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+
+    @property
+    def Url(self):
+        return self.TryGetValue("url")
+
+    @property
+    def Uri(self):
+        return self.TryGetValue("uri")
+
+    @property
+    def AgentName(self):
+        return self.TryGetValue("agent_name")
+
+    @property
+    def HttpMethod(self):
+        return self.TryGetValue("http_method")
+
+    @property
+    def Type(self):
+        return self.TryGetValue("type")
+
+    @property
+    def TaintPosition(self):
+        return self.TryGetValue("taint_position")
+
+    @property
+    def FirstTime(self):
+        return self.TryGetValue("first_time")
+
+    @property
+    def LatestTime(self):
+        return self.TryGetValue("latest_time")
+
+    @property
+    def ProjectName(self):
+        return self.TryGetValue("project_name")
+
+    @property
+    def ProjectVersion(self):
+        return self.TryGetValue("project_version")
+
+    @property
+    def Language(self):
+        return self.TryGetValue("language")
+
+    @property
+    def Level(self):
+        return self.TryGetValue("level")
+
+    @property
+    def LevelType(self):
+        return self.TryGetValue("level_type")
+
+    @property
+    def Counts(self):
+        return self.TryGetValue("counts")
+
+    @property
+    def ReqHeader(self):
+        return self.TryGetValue("req_header")
+
+    @property
+    def Response(self):
+        return self.TryGetValue("response")
+
+    @property
+    def Graph(self):
+        return self.TryGetValue("graph")
+
+    @property
+    def ContextPath(self):
+        return self.TryGetValue("context_path")
+
+    @property
+    def ClientIp(self):
+        return self.TryGetValue("client_ip")
+
+    @property
+    def Status(self):
+        return self.TryGetValue("status")
+
+    @property
+    def TaintValue(self):
+        return self.TryGetValue("taint_value")
+
+    @property
+    def ParamName(self):
+        return self.TryGetValue("param_name")
+
+    @property
+    def MethodPoolId(self):
+        return self.TryGetValue("method_pool_id")
+
+    @property
+    def ProjectId(self):
+        return self.TryGetValue("project_id")
+
+class VulnServer(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+
+    @property
+    def Name(self):
+        return self.TryGetValue("name")
+
+    @property
+    def Hostname(self):
+        return self.TryGetValue("hostname")
+
+    @property
+    def Ip(self):
+        return self.TryGetValue("ip")
+
+    @property
+    def Port(self):
+        return self.TryGetValue("port")
+
+    @property
+    def Container(self):
+        return self.TryGetValue("container")
+
+    @property
+    def ServerType(self):
+        return self.TryGetValue("server_type")
+
+    @property
+    def ContainerPath(self):
+        return self.TryGetValue("container_path")
+
+    @property
+    def Runtime(self):
+        return self.TryGetValue("runtime")
+
+    @property
+    def Environment(self):
+        return self.TryGetValue("environment")
+
+    @property
+    def Command(self):
+        return self.TryGetValue("command")
+
+class VulnDetail(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+
+    @property
+    def Vuln(self) -> Vuln:
+        tmpData = self.TryGetValue("vul")
+        tmpObject = Vuln(tmpData)
+        return tmpObject
+
+    @property
+    def Server(self) -> VulnServer:
+        tmpData = self.TryGetValue("server")
+        tmpObject = VulnServer(tmpData)
+        return tmpObject
+
 class VulnSummary(BaseObject):
     WithOutRiskLevel = ["无风险"]