From 649ea30cb4c05ae45045cacb098328717b31eb4a Mon Sep 17 00:00:00 2001
From: Mr Chen <midoks@163.com>
Date: Sat, 19 Oct 2024 22:17:16 +0800
Subject: [PATCH] Update __init__.py

---
 web/admin/__init__.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/web/admin/__init__.py b/web/admin/__init__.py
index d2087bca3a..084098e65f 100644
--- a/web/admin/__init__.py
+++ b/web/admin/__init__.py
@@ -15,7 +15,7 @@
 from flask import Flask, abort, request, current_app, session, url_for
 from werkzeug.local import LocalProxy
 from flask import Blueprint, render_template
-
+from flask import render_template_string
 import setting
 
 socketio = SocketIO(manage_session=False, async_mode='threading',
@@ -44,6 +44,18 @@
 app.logger.debug("Python syspath: %s", sys.path)
 
 
+@app.after_request
+def requestAfter(response):
+    response.headers['soft'] = setting.APP_NAME
+    response.headers['mw-version'] = setting.APP_VERSION
+    return response
+
+
+@app.errorhandler(404)
+def page_unauthorized(error):
+    return render_template_string('404 not found', error_info=error), 404
+
+
 # def create_app(app_name = None):
 #     
 #     if not app_name: