From 71a25f47740d03931c440c09deda90adb36be069 Mon Sep 17 00:00:00 2001
From: resonate <1650473152@qq.com>
Date: Fri, 14 Jan 2022 12:39:42 +0800
Subject: [PATCH] =?UTF-8?q?session=E5=AD=98=E5=85=A5redis?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 applications/configs/config.py          | 10 +++++++++-
 applications/extensions/__init__.py     |  2 ++
 applications/extensions/init_session.py |  3 +++
 applications/models/__init__.py         |  2 +-
 4 files changed, 15 insertions(+), 2 deletions(-)
 create mode 100644 applications/extensions/init_session.py

diff --git a/applications/configs/config.py b/applications/configs/config.py
index bdae8ec..0ff2057 100644
--- a/applications/configs/config.py
+++ b/applications/configs/config.py
@@ -1,9 +1,10 @@
 import logging
 import os
 
+import redis
+
 
 class BaseConfig:
-
     SYSTEM_NAME = os.getenv('SYSTEM_NAME', 'Pear Admin')
     # 主题面板的链接列表配置
     SYSTEM_PANEL_LINKS = [
@@ -43,6 +44,13 @@ class BaseConfig:
     MYSQL_PORT = int(os.getenv('MYSQL_PORT') or 3306)
     MYSQL_DATABASE = os.getenv('MYSQL_DATABASE') or "PearAdminFlask"
 
+    SESSION_USE_SIGNER = True
+    # 指明保存到redis中
+    SESSION_TYPE = "redis"
+    SESSION_REDIS = redis.StrictRedis(host=REDIS_HOST, port=REDIS_PORT, db=4)
+    # session的有效期，单位：秒
+    PERMANENT_SESSION_LIFETIME = 7200
+
     # mysql 数据库的配置信息
     SQLALCHEMY_DATABASE_URI = f"mysql+pymysql://{MYSQL_USERNAME}:{MYSQL_PASSWORD}@{MYSQL_HOST}:{MYSQL_PORT}/{MYSQL_DATABASE}"
     # 默认日志等级
diff --git a/applications/extensions/__init__.py b/applications/extensions/__init__.py
index 2e7e108..830c545 100644
--- a/applications/extensions/__init__.py
+++ b/applications/extensions/__init__.py
@@ -9,6 +9,7 @@ from .init_mail import init_mail
 from .init_apscheduler import init_scheduler
 from .init_upload import init_upload
 from .init_dotenv import init_dotenv
+from .init_session import session
 
 
 def init_plugs(app: Flask) -> None:
@@ -21,3 +22,4 @@ def init_plugs(app: Flask) -> None:
     init_scheduler(app)
     init_upload(app)
     init_dotenv()
+    session.init_app(app)
diff --git a/applications/extensions/init_session.py b/applications/extensions/init_session.py
new file mode 100644
index 0000000..571c438
--- /dev/null
+++ b/applications/extensions/init_session.py
@@ -0,0 +1,3 @@
+from flask_session import Session
+
+session = Session()
diff --git a/applications/models/__init__.py b/applications/models/__init__.py
index fe78e9c..2d3fd14 100644
--- a/applications/models/__init__.py
+++ b/applications/models/__init__.py
@@ -6,4 +6,4 @@ from .admin_power import Power
 from .admin_role import Role
 from .admin_role_power import role_power
 from .admin_user import User
-from .admin_user_role import user_role
\ No newline at end of file
+from .admin_user_role import user_role