Skip to content

Commit fcce5ff

Browse files
committed
sca module complete
1 parent a5b88b2 commit fcce5ff

File tree

4 files changed

+96
-9
lines changed

4 files changed

+96
-9
lines changed

README.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
* @Date: 2021-12-23 14:25:35
44
* @version:
55
* @LastEditors: 饕餮
6-
* @LastEditTime: 2021-12-25 21:32:18
6+
* @LastEditTime: 2021-12-27 09:59:59
77
* @Description: How to use
88
-->
99
# DongTai-SDK-Python
@@ -62,8 +62,9 @@ dongTaiSdk.GetAgentDetail(agentId)
6262
dongTaiSdk.GetAgentList(page=1,pageSize=50,projectName=None,state=None,token=None)
6363
```
6464

65-
### Sca
65+
### Sca (Completed)
6666
```python
6767
dongTaiSdk.GetScaDetail(scaId)
6868
dongTaiSdk.GetScaList(page=1,pageSize=50,keyword=None,language=None,level=None,order=None,projectId=None,projectName=None,versionId=None)
69+
dongTaiSdk.GetScaSummary(page=1,pageSize=50,keyword=None,language=None,level=None,order=None,projectId=None,projectName=None,versionId=None)
6970
```

dongtai_sdk/DongTai.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,12 @@
33
Date: 2021-12-23 15:10:01
44
version:
55
LastEditors: 饕餮
6-
LastEditTime: 2021-12-25 21:32:29
6+
LastEditTime: 2021-12-27 09:59:24
77
Description: Main
88
'''
99
from .base.DongTaiProject import DongTaiProject,DongTaiProjectVersion
1010
from .base.DongTaiAgent import DongTaiAgent
11-
from .base.DongTaiSca import DongTaiSca
11+
from .base.DongTaiSca import DongTaiSca, ScaSummary
1212
from .DongTaiApi import DongTaiApi
1313
from .base.BaseObejct import DongTaiError
1414

@@ -193,4 +193,12 @@ def GetScaDetail(self,scaId):
193193
errorObject = DongTaiError(errorMsg)
194194
return errorObject
195195

196-
196+
def GetScaSummary(self,page=1,pageSize=50,keyword=None,language=None,level=None,order=None,projectId=None,projectName=None,versionId=None):
197+
repData = self.dongTaiApi.GetScaSummary(page,pageSize,keyword,language,level,order,projectId,projectName,versionId)
198+
if repData["status"] == 201:
199+
scaSummary = ScaSummary(repData["data"])
200+
return scaSummary
201+
else:
202+
errorMsg = {"status":repData["status"],"msg":repData["msg"]}
203+
errorObject = DongTaiError(errorMsg)
204+
return errorObject

dongtai_sdk/base/BaseObejct.py

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
Date: 2021-12-23 14:49:05
44
version:
55
LastEditors: 饕餮
6-
LastEditTime: 2021-12-23 21:05:50
6+
LastEditTime: 2021-12-27 10:01:53
77
Description: 通用基类
88
'''
99
import json
@@ -61,4 +61,20 @@ def Status(self):
6161

6262
@property
6363
def ErrorMsg(self):
64-
return self.TryGetValue("msg")
64+
return self.TryGetValue("msg")
65+
66+
class ProjectSummary(BaseObject):
67+
def __init__(self,jsonData):
68+
self.ObjectData = jsonData
69+
70+
@property
71+
def ProjectName(self):
72+
return self.TryGetValue("project_name")
73+
74+
@property
75+
def Count(self):
76+
return self.TryGetValue("count")
77+
78+
@property
79+
def Id(self):
80+
return self.TryGetValue("id")

dongtai_sdk/base/DongTaiSca.py

Lines changed: 64 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,11 @@
33
Date: 2021-12-24 10:57:03
44
version:
55
LastEditors: 饕餮
6-
LastEditTime: 2021-12-25 21:21:59
6+
LastEditTime: 2021-12-27 10:02:24
77
Description: Sca Object
88
'''
99
import json
10-
from .BaseObejct import BaseObject
10+
from .BaseObejct import BaseObject,ProjectSummary
1111

1212
class ScaVulnReference(BaseObject):
1313
def __init__(self,jsonData):
@@ -64,6 +64,34 @@ def Reference(self):
6464
returnData.append(tmpObject)
6565
return returnData
6666

67+
class ScaLanguage(BaseObject):
68+
def __init__(self,jsonData):
69+
self.ObjectData = jsonData
70+
71+
@property
72+
def Language(self):
73+
return self.TryGetValue("language")
74+
75+
@property
76+
def Count(self):
77+
return self.TryGetValue("count")
78+
79+
class ScaLevel(BaseObject):
80+
def __init__(self,jsonData):
81+
self.ObjectData = jsonData
82+
83+
@property
84+
def Level(self):
85+
return self.TryGetValue("level")
86+
87+
@property
88+
def Count(self):
89+
return self.TryGetValue("count")
90+
91+
@property
92+
def LevelId(self):
93+
return self.TryGetValue("level_id")
94+
6795
class DongTaiSca(BaseObject):
6896
def __init__(self,jsonData):
6997
self.ObjectData = jsonData
@@ -128,4 +156,38 @@ def Vuls(self):
128156
for tmpData in tmpDataList:
129157
tmpObject = ScaVuln(tmpData)
130158
returnData.append(tmpObject)
159+
return returnData
160+
161+
class ScaSummary(BaseObject):
162+
def __init__(self,jsonData):
163+
self.ObjectData = jsonData
164+
165+
@property
166+
def Language(self):
167+
returnData = []
168+
tmpDataList = self.TryGetValue("language")
169+
if tmpDataList is not None:
170+
for tmpData in tmpDataList:
171+
tmpObject = ScaLanguage(tmpData)
172+
returnData.append(tmpObject)
173+
return returnData
174+
175+
@property
176+
def Level(self):
177+
returnData = []
178+
tmpDataList = self.TryGetValue("level")
179+
if tmpDataList is not None:
180+
for tmpData in tmpDataList:
181+
tmpObject = ScaLevel(tmpData)
182+
returnData.append(tmpObject)
183+
return returnData
184+
185+
@property
186+
def Projects(self):
187+
returnData = []
188+
tmpDataList = self.TryGetValue("projects")
189+
if tmpDataList is not None:
190+
for tmpData in tmpDataList:
191+
tmpObject = ProjectSummary(tmpData)
192+
returnData.append(tmpObject)
131193
return returnData

0 commit comments

Comments
 (0)