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

Merge pull request #41 from KasperskyLab/dev

Browse Source 
Deleting saved capture folder
This commit is contained in:
Félix Aimé 2021-01-28 19:00:34 +01:00 committed by GitHub
commit 3d52ee83c2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats 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()),