24 lines
683 B
Python
24 lines
683 B
Python
from flask import current_app as app
|
|
|
|
import json
|
|
from os.path import isfile
|
|
from pathlib import Path
|
|
|
|
from copy import deepcopy
|
|
|
|
def check_file(filename:str) -> bool:
|
|
data_dir = Path(app.config.get('DATA'))
|
|
if isfile(f'./{data_dir}/{filename}'): return True
|
|
return False
|
|
|
|
def load(filename:str) -> dict:
|
|
data_dir = Path(app.config.get('DATA'))
|
|
with open(f'./{data_dir}/{filename}') as file: return json.load(file)
|
|
|
|
def declutter_results(raw_results:dict) -> dict:
|
|
data = deepcopy(raw_results)
|
|
playbook_lists = []
|
|
for item in data['playbooks']:
|
|
playbook_lists.append(list(item.keys())[0])
|
|
data['playbooks'] = playbook_lists
|
|
return data |