c_meyer/KasperskyLab-TinyCheck
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Deleting saved capture folder

Browse Source
This commit is contained in:
Félix Aime
2021-01-28 18:57:35 +01:00
parent 534f847ea7
commit a6c88b7cea
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server/frontend/app/classes/save.py
View File

@@ -50,11 +50,13 @@ class Save():
if method == "usb": if method == "usb":
cd = datetime.now().strftime("%d%m%Y-%H%M") cd = datetime.now().strftime("%d%m%Y-%H%M")
if shutil.make_archive("{}/TinyCheck_{}".format(self.mount_point, cd), "zip", "/tmp/{}/".format(token)): if shutil.make_archive("{}/TinyCheck_{}".format(self.mount_point, cd), "zip", "/tmp/{}/".format(token)):
shutil.rmtree("/tmp/{}/".format(token))
return jsonify({"status": True, return jsonify({"status": True,
"message": "Capture saved on the USB key"}) "message": "Capture saved on the USB key"})
elif method == "url": elif method == "url":
cd = datetime.now().strftime("%d%m%Y-%H%M") cd = datetime.now().strftime("%d%m%Y-%H%M")
if shutil.make_archive("/tmp/TinyCheck_{}".format(cd), "zip", "/tmp/{}/".format(token)): if shutil.make_archive("/tmp/TinyCheck_{}".format(cd), "zip", "/tmp/{}/".format(token)):
shutil.rmtree("/tmp/{}/".format(token))
with open("/tmp/TinyCheck_{}.zip".format(cd), "rb") as f: with open("/tmp/TinyCheck_{}.zip".format(cd), "rb") as f:
return send_file( return send_file(
io.BytesIO(f.read()), io.BytesIO(f.read()),