From 3692299f0d2740835cc47e69b0c88b1d5d8e752b Mon Sep 17 00:00:00 2001
From: fleaz <mail@felixbreidenstein.de>
Date: Wed, 16 Oct 2024 21:46:03 +0200
Subject: [PATCH] Fix after_request hook to delete files

---
 main.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index 0c251e5..7fddb9c 100644
--- a/main.py
+++ b/main.py
@@ -26,12 +26,18 @@ def do_upload():
 
 @route('/download/<id>')
 def download(id):
-    @hook('after_request')
-    def delFiles():
-        remove(f"./downloads/{id}.pdf")
     filename = f"{id}.pdf"
     return static_file(filename, root='downloads', download=filename)
 
+@hook('after_request')
+def delFiles():
+    if request.path.startswith("/download/"):
+        id = request.path.split("/")[2]
+        try:
+             remove(f"./downloads/{id}.pdf")
+        except FileNotFoundError:
+            return
+
 @route('/static/<filename:path>')
 def send_static(filename):
     return static_file(filename, root='static/')