From 473f2ceaf584774cf829e093a6a62ed7f8bd428f Mon Sep 17 00:00:00 2001
From: superlishunqin <852326703@qq.com>
Date: Sat, 17 May 2025 13:25:40 +0800
Subject: [PATCH] cookie_setting

---
 app/__init__.py | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/app/__init__.py b/app/__init__.py
index a2de2f4..55c1333 100644
--- a/app/__init__.py
+++ b/app/__init__.py
@@ -1,4 +1,4 @@
-from flask import Flask, render_template, session, g, Markup, redirect, url_for
+from flask import Flask, render_template, session, g, Markup, redirect, url_for, request
 from flask_login import LoginManager
 from app.models.database import db
 from app.models.user import User
@@ -111,6 +111,17 @@ def create_app(config=None):
 
         db.session.commit()
 
+    # 添加缓存控制中间件
+    @app.after_request
+    def add_cache_headers(response):
+        # 为HTML页面和主页添加禁止缓存的头
+        if request.path == '/' or 'text/html' in response.headers.get('Content-Type', ''):
+            response.headers["Cache-Control"] = "no-cache, no-store, must-revalidate, max-age=0"
+            response.headers["Pragma"] = "no-cache"
+            response.headers["Expires"] = "0"
+            response.headers['Vary'] = 'Cookie, Authorization'
+        return response
+
     # 其余代码保持不变...
     @app.before_request
     def load_logged_in_user():
@@ -233,10 +244,8 @@ def create_app(config=None):
             get_recent_notifications=get_recent_notifications
         )
 
-    return app
-
     @app.context_processor
     def inject_now():
-        return {'now': datetime.datetime.now()}
-
+        return {'now': datetime.now()}
 
+    return app