0.1.5s

run.py

@@ -6,7 +6,7 @@ from pathlib import Path
 
 import requests
 from flask import (Flask, jsonify, render_template, render_template_string,
-                    request, send_from_directory)
+                    request, send_from_directory, make_response)
 
 from system_module import db, gelbooru, image_processor, logger
 
@@ -30,6 +30,7 @@ THUMB_FOLDER = CONFIG_ROOT["file"]["thumb"]
 
 gb = gelbooru.GB(GB_ID, GB_HASH, GB_HEADERS, FILE_DW_FOLDERS)
 log = logger.Logger()
+log.init_global_exception_hook()
 database = db.DB(True)
 app = Flask(__name__)
 
@@ -59,7 +60,10 @@ def post(post_id):
 @app.route('/file/<path:filename>')
 def serve_image(filename):
     folder_path = find_file(filename, FILE_FOLDERS)
-    return send_from_directory(folder_path, filename)
+    if folder_path == None:
+        return make_response("Not found", 404)
+    else:
+        return send_from_directory(folder_path, filename)
 
 @app.route('/thumb/<path:filename>')
 def serve_thumb(filename):