infosecn1nja
/
DeTTECT
mirror of https://github.com/infosecn1nja/DeTTECT.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Don't overwrite output files if they already exist, but append a number to the filename as suffix.

master
Ruben Bouman 2019-08-21 10:53:04 +02:00
parent 14852fb24a
commit 6efd04cb3b
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
group_mapping.py
View File

@ -554,13 +554,10 @@ def generate_group_heat_map(groups, overlay, overlay_type, stage, platform, soft
json_string = simplejson.dumps(layer).replace('}, ', '},\n')
if stage == 'pre-attack':
filename = "output/" + stage + '_' + '_'.join(groups_list)
filename = '_'.join(groups_list)
elif overlay:
filename = "output/" + stage + '_' + platform.lower() + '_' + '_'.join(groups_list) + '-overlay_' + '_'.join(overlay_list)
filename = platform.lower() + '_' + '_'.join(groups_list) + '-overlay_' + '_'.join(overlay_list)
else:
filename = "output/" + stage + '_' + platform.lower() + '_' + '_'.join(groups_list)
filename = filename[:255] + '.json'
with open(filename, 'w') as f: # write layer file to disk
f.write(json_string)
print('Written layer: ' + filename)
filename = platform.lower() + '_' + '_'.join(groups_list)
write_file(stage, filename[:255], json_string)