External/agentic-soc-platform
1
0
Fork 0
mirror of https://github.com/FunnyWolf/agentic-soc-platform.git synced 2026-08-22 13:12:56 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix missing default value when create record

Browse Source
This commit is contained in:
funnywolf
2026-05-11 20:00:24 +08:00
parent 2ff3d13c52
commit 1d8f00ba4d
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
PLUGINS/Mock/SIRP/delete_all.py
+1 -1
View File
@@ -19,7 +19,7 @@ if __name__ == "__main__":
# },
# ]
}
worksheet_id_list = ["artifact", "alert", "case", "enrichment", "ticket", "playbook", "message"]
worksheet_id_list = ["artifact", "alert", "case", "enrichment", "ticket", "playbook"]
for worksheet_id in worksheet_id_list:
row_ids = []
rows = WorksheetRow.list(worksheet_id, filter=filter)
PLUGINS/SIRP/sirpapi.py
+5 -5
View File
@@ -18,9 +18,9 @@ from PLUGINS.SIRP.sirpcoremodel import Severity, Confidence, EnrichmentModel, Ti
from PLUGINS.SIRP.sirpextramodel import PlaybookType, PlaybookJobStatus, KnowledgeAction, PlaybookModel, KnowledgeModel
def model_to_fields(model_instance: BaseModel) -> List[Dict[str, Any]]:
def model_to_fields(model_instance: BaseModel, exclude_unset: bool = True) -> List[Dict[str, Any]]:
fields = []
model_data = model_instance.model_dump(mode='json', exclude_unset=True)
model_data = model_instance.model_dump(mode='json', exclude_unset=exclude_unset)
for key, value in model_data.items():
field_info = model_instance.model_fields.get(key)
field_item = {
@@ -232,7 +232,7 @@ class BaseWorksheetEntity(ABC, Generic[T]):
"""
model = cls._prepare_for_save(model)
fields = model_to_fields(model)
fields = model_to_fields(model, exclude_unset=False)
row_id = WorksheetRow.create(cls.WORKSHEET_ID, fields)
return row_id
@@ -270,11 +270,11 @@ class BaseWorksheetEntity(ABC, Generic[T]):
"""
model = cls._prepare_for_save(model)
fields = model_to_fields(model)
if model.row_id is None:
fields = model_to_fields(model, exclude_unset=False)
row_id = WorksheetRow.create(cls.WORKSHEET_ID, fields)
else:
fields = model_to_fields(model)
row_id = WorksheetRow.update(cls.WORKSHEET_ID, model.row_id, fields)
return row_id