SearchProject

Author: CyberGiter

dongtai_sdk/DongTai.py

@@ -3,7 +3,7 @@
 Date: 2021-12-23 15:10:01
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-23 21:08:50
+LastEditTime: 2021-12-24 10:08:23
 Description: Main
 '''
 from .base.DongTaiProject import DongTaiProject,DongTaiProjectVersion
@@ -50,3 +50,13 @@ def AddProjectVersion(self,projectId,verName,description,isEdit=True):
             errorObject = DongTaiError(errorMsg)
             return errorObject
 
+    def SearchProject(self,projectId):
+        repData = self.dongTaiApi.SearchProject(projectId)
+        if repData["status"] == 201:
+            tmpObject = DongTaiProject(repData["data"])
+            return tmpObject
+        else:
+            errorMsg = {"status":repData["status"],"msg":repData["msg"]}
+            errorObject = DongTaiError(errorMsg)
+            return errorObject
+

dongtai_sdk/base/DongTaiProject.py

@@ -3,7 +3,7 @@
 Date: 2021-12-23 14:50:44
 version: 
 LastEditors: 饕餮
-LastEditTime: 2021-12-23 20:54:08
+LastEditTime: 2021-12-24 10:05:20
 Description: 动态项目对象
 '''
 from .BaseObejct import BaseObject
@@ -24,6 +24,34 @@ def Total(self):
     def Name(self):
         return self.TryGetValue("name")
 
+class LevelCount(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+
+    @property
+    def LevelId(self):
+        return self.TryGetValue("level_id")
+
+    @property
+    def LevelName(self):
+        return self.TryGetValue("level_name")
+
+    @property
+    def Number(self):
+        return self.TryGetValue("num")
+
+class DayNumber(BaseObject):
+    def __init__(self,jsonData):
+        self.ObjectData = jsonData
+
+    @property
+    def DayLabel(self):
+        return self.TryGetValue("day_label")
+    
+    @property
+    def DayNum(self):
+        return self.TryGetValue("day_num")
+
 class DongTaiProjectVersion(BaseObject):
     def __init__(self,jsonData):
         self.ObjectData = jsonData
@@ -87,4 +115,33 @@ def AgentLanguage(self):
 
     @property
     def VulValidation(self):
-        return self.TryGetValue("vul_validation")
+        return self.TryGetValue("vul_validation")
+
+    @property
+    def VersionData(self):
+        tmpData = self.TryGetValue("versionData",None)
+        if tmpData is not None:
+            tmpObject = DongTaiProjectVersion(tmpData)
+            return tmpObject
+        else:
+            return None
+
+    @property
+    def LevelCount(self):
+        tmpDataList = self.TryGetValue("level_count")
+        returnData = []
+        for tmpData in tmpDataList:
+            tmpObject = LevelCount(tmpData)
+            returnData.append(tmpObject)
+        return returnData
+
+    @property
+    def DayNumber(self):
+        tmpDataList = self.TryGetValue("day_num")
+        returnData = []
+        for tmpData in tmpDataList:
+            tmpObject = DayNumber(tmpData)
+            returnData.append(tmpObject)
+        return returnData
+
+