GetProjectDetail

Author: SpenserCai

dongtai_sdk/DongTai.py

@@ -3,7 +3,7 @@
 Date: 2021-12-23 15:10:01
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-25 15:51:12
+LastEditTime: 2021-12-25 16:09:46
 Description: Main
 '''
 from .base.DongTaiProject import DongTaiProject,DongTaiProjectVersion
@@ -81,6 +81,17 @@ def SearchProject(self,projectId):
             errorObject = DongTaiError(errorMsg)
             return errorObject
 
+    def GetProjectDetail(self,projectId):
+        repData = self.dongTaiApi.GetProjectDetail(projectId)
+        if repData["status"] == 201:
+            tmpObject = DongTaiProject(repData["data"])
+            #TODO:通过参数判断是否加载完整Agent信息
+            return tmpObject
+        else:
+            errorMsg = {"status":repData["status"],"msg":repData["msg"]}
+            errorObject = DongTaiError(errorMsg)
+            return errorObject
+
     #[Agent Function]
     def DeleteAgent(self,agentId):
         repData = self.dongTaiApi.DeleteAgent(agentId)

dongtai_sdk/base/DongTaiProject.py

@@ -3,7 +3,7 @@
 Date: 2021-12-23 14:50:44
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-24 10:49:49
+LastEditTime: 2021-12-25 16:06:03
 Description: 动态项目对象
 '''
 from .BaseObejct import BaseObject
@@ -24,6 +24,18 @@ def Total(self):
     def Name(self):
         return self.TryGetValue("name")
 
+class AgentBaseInfo(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+
+    @property
+    def Id(self):
+        return self.TryGetValue("id")
+
+    @property
+    def Name(self):
+        return self.TryGetValue("name")
+
 class LevelCount(BaseObject):
     def __init__(self,jsonData):
         self.ObjectData = jsonData
@@ -168,4 +180,17 @@ def TypeSummary(self):
             tmpObject = TypeSummary(tmpData)
             returnData.append(tmpObject)
         return returnData
+
+    @property
+    def ScanId(self):
+        return self.TryGetValue("scan_id")
 
+    #通过参数判断是否要加载完整的Agent信息
+    @property
+    def Agents(self):
+        tmpDataList = self.TryGetValue("agents")
+        returnData = []
+        for tmpData in tmpDataList:
+            tmpObject = AgentBaseInfo(tmpData)
+            returnData.append(tmpObject)
+        return returnData