Skip to content

Commit 6aca980

Browse files
committed
add GetScaList,GetScaDetail
1 parent 81b39b5 commit 6aca980

File tree

3 files changed

+157
-3
lines changed

3 files changed

+157
-3
lines changed

README.md

Lines changed: 7 additions & 1 deletion
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 17:24:24
6+
* @LastEditTime: 2021-12-25 21:32:18
77
* @Description: How to use
88
-->
99
# DongTai-SDK-Python
@@ -61,3 +61,9 @@ dongTaiSdk.ModifiedAgentAlias(agentId,alias)
6161
dongTaiSdk.GetAgentDetail(agentId)
6262
dongTaiSdk.GetAgentList(page=1,pageSize=50,projectName=None,state=None,token=None)
6363
```
64+
65+
### Sca
66+
```python
67+
dongTaiSdk.GetScaDetail(scaId)
68+
GetScaList(page=1,pageSize=50,keyword=None,language=None,level=None,order=None,projectId=None,projectName=None,versionId=None)
69+
```

dongtai_sdk/DongTai.py

Lines changed: 27 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,12 @@
33
Date: 2021-12-23 15:10:01
44
version:
55
LastEditors: 饕餮
6-
LastEditTime: 2021-12-25 17:23:57
6+
LastEditTime: 2021-12-25 21:32:29
77
Description: Main
88
'''
99
from .base.DongTaiProject import DongTaiProject,DongTaiProjectVersion
1010
from .base.DongTaiAgent import DongTaiAgent
11+
from .base.DongTaiSca import DongTaiSca
1112
from .DongTaiApi import DongTaiApi
1213
from .base.BaseObejct import DongTaiError
1314

@@ -167,4 +168,29 @@ def GetAgentList(self,page=1,pageSize=50,projectName=None,state=None,token=None)
167168
errorObject = DongTaiError(errorMsg)
168169
return errorObject
169170

171+
#[Sca Function]
172+
def GetScaList(self,page=1,pageSize=50,keyword=None,language=None,level=None,order=None,projectId=None,projectName=None,versionId=None):
173+
returnData = []
174+
repData = self.dongTaiApi.GetScaList(self,page,pageSize,keyword,language,level,order,projectId,projectName,versionId)
175+
if repData["status"] == 201:
176+
tmpDataList = repData["data"]
177+
for tmpData in tmpDataList:
178+
tmpObject = DongTaiSca(tmpData)
179+
returnData.append(tmpObject)
180+
return returnData
181+
else:
182+
errorMsg = {"status":repData["status"],"msg":repData["msg"]}
183+
errorObject = DongTaiError(errorMsg)
184+
return errorObject
185+
186+
def GetScaDetail(self,scaId):
187+
repData = self.dongTaiApi.GetScaDetail(scaId)
188+
if repData["status"] == 201:
189+
scaObject = DongTaiSca(repData["data"])
190+
return scaObject
191+
else:
192+
errorMsg = {"status":repData["status"],"msg":repData["msg"]}
193+
errorObject = DongTaiError(errorMsg)
194+
return errorObject
195+
170196

dongtai_sdk/base/DongTaiSca.py

Lines changed: 123 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,129 @@
33
Date: 2021-12-24 10:57:03
44
version:
55
LastEditors: 饕餮
6-
LastEditTime: 2021-12-24 15:41:42
6+
LastEditTime: 2021-12-25 21:21:59
77
Description: Sca Object
88
'''
9+
import json
910
from .BaseObejct import BaseObject
11+
12+
class ScaVulnReference(BaseObject):
13+
def __init__(self,jsonData):
14+
self.ObjectData = jsonData
15+
16+
@property
17+
def Type(self):
18+
return self.TryGetValue("type")
19+
20+
@property
21+
def Title(self):
22+
return self.TryGetValue("title")
23+
24+
@property
25+
def Url(self):
26+
return self.TryGetValue("url")
27+
28+
class ScaVuln(BaseObject):
29+
def __init__(self,jsonData):
30+
self.ObjectData = jsonData
31+
32+
@property
33+
def SafeVersion(self):
34+
return self.TryGetValue("safe_version")
35+
36+
@property
37+
def VulCve(self):
38+
return self.TryGetValue("vulcve")
39+
40+
@property
41+
def VulCwe(self):
42+
return self.TryGetValue("vulcwe")
43+
44+
@property
45+
def VulName(self):
46+
return self.TryGetValue("vulname")
47+
48+
@property
49+
def OverView(self):
50+
return self.TryGetValue("overview")
51+
52+
@property
53+
def TearDown(self):
54+
return self.TryGetValue("teardown")
55+
56+
@property
57+
def Reference(self):
58+
returnData = []
59+
tmpData = self.TryGetValue("reference")
60+
if tmpData is not None:
61+
tmpDataList = json.dumps(tmpData)
62+
for tmpItem in tmpDataList:
63+
tmpObject = ScaVulnReference(tmpItem)
64+
returnData.append(tmpObject)
65+
return returnData
66+
67+
class DongTaiSca(BaseObject):
68+
def __init__(self,jsonData):
69+
self.ObjectData = jsonData
70+
71+
@property
72+
def Id(self):
73+
return self.TryGetValue("id")
74+
75+
@property
76+
def PackageName(self):
77+
return self.TryGetValue("package_name")
78+
79+
@property
80+
def Version(self):
81+
return self.TryGetValue("version")
82+
83+
@property
84+
def ProjectName(self):
85+
return self.TryGetValue("project_name")
86+
87+
@property
88+
def ProjectId(self):
89+
return self.TryGetValue("project_id")
90+
91+
@property
92+
def ProjectVersion(self):
93+
return self.TryGetValue("project_version")
94+
95+
@property
96+
def Language(self):
97+
return self.TryGetValue("language")
98+
99+
@property
100+
def AgentName(self):
101+
return self.TryGetValue("agent_name")
102+
103+
@property
104+
def SignatureValue(self):
105+
return self.TryGetValue("signature_value")
106+
107+
@property
108+
def Level(self):
109+
return self.TryGetValue("level")
110+
111+
@property
112+
def LevelType(self):
113+
return self.TryGetValue("level_type")
114+
115+
@property
116+
def VulCount(self):
117+
return self.TryGetValue("vul_count")
118+
119+
@property
120+
def Dt(self):
121+
return self.TryGetValue("dt")
122+
123+
@property
124+
def Vuls(self):
125+
returnData = []
126+
tmpDataList = self.TryGetValue("vuls")
127+
if tmpDataList is not None:
128+
for tmpData in tmpDataList:
129+
tmpObject = ScaVuln(tmpData)
130+
returnData.append(tmpObject)
131+
return returnData

0 commit comments

Comments
 (0)