Fix download a file wasnt exists

2019-04-08 23:05:06 +08:00 · 2019-04-08 23:05:06 +08:00 · 86fbf31120
commit 86fbf31120
parent ef7d4c8d3e
1 changed files with 4 additions and 1 deletions
--- a/WebMain.py
+++ b/WebMain.py
@ -161,7 +161,10 @@ def fileDownload(path):
        with open(file, "rb") as f:
            for row in f:
                yield row
-    return Response(generate(path), mimetype='application/octet-stream')
+    if os.path.exists(path):
+        return Response(generate(path), mimetype='application/octet-stream')
+    else:
+        return Response(status=404)


 def SubThread():