Skip to content

Commit ced6164

Browse files
committed
Add a function to generate
JSON objects in the API library.
1 parent 0f15968 commit ced6164

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

contentctl/api.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
#from contentctl import contentctl
44
from contentctl.input.yml_reader import YmlReader
55
from contentctl.objects.config import test_common, test_servers, test
6-
6+
from contentctl.objects.security_content_object import SecurityContentObject
77

88

99
def configFromFile(path:Path=Path("contentctl.yml"), config: dict[str,Any]={},
@@ -122,3 +122,18 @@ def updateConfig(config:Union[test,test_servers], **key_value_updates:dict[str,A
122122

123123
return config_copy
124124

125+
126+
from contentctl.input.director import DirectorOutputDto
127+
def contentToDict(director:DirectorOutputDto)->dict[str,list[dict[str,Any]]]:
128+
output_dict:dict[str,list[dict[str,Any]]] = {}
129+
for contentType in ['detections','stories','baselines','investigations',
130+
'playbooks','macros','lookups','deployments','ssa_detections']:
131+
132+
output_dict[contentType] = []
133+
t:list[SecurityContentObject] = getattr(director,contentType)
134+
135+
for item in t:
136+
print(item.model_dump())
137+
output_dict[contentType].append(item.model_dump())
138+
return output_dict
139+

0 commit comments

Comments
 (0)