Flask中redis的配置与使用

2023-12-13 23:15:01
?注意点:

???????? 在__init__.py中需要将redis_store设置成全局变量,这样方便其他文件导入

一、config.py
import logging
import os
from datetime import timedelta

from redis import StrictRedis


class Config:
    # 调试信息
    DEBUG = True
    SECRET_KEY = os.urandom(32)
    JWT_SECRET = os.urandom(32)
    JWT_EXPIRY_SECOND = 5

    # 数据库信息
    MYSQL_DIALECT = 'mysql'
    MYSQL_DIRVER = 'pymysql'
    MYSQL_NAME = 'root'
    MYSQL_PWD = '123456'
    MYSQL_HOST = 'localhost'
    MYSQL_PORT = 3306
    MYSQL_DB = 'manger'
    MYSQL_CHARSET = 'utf8mb4'
    SQLALCHEMY_DATABASE_URI = f'{MYSQL_DIALECT}+{MYSQL_DIRVER}://{MYSQL_NAME}:{MYSQL_PWD}@{MYSQL_HOST}:{MYSQL_PORT}/{MYSQL_DB}?charset={MYSQL_CHARSET}'
    SQLALCHEMY_TRACK_MODIFICATIONS = True

    # redis配置
    REDIS_HOST = '127.0.0.1'
    REDIS_PORT = 6379

    # session配置
    # SESSION_TYPE = 'filesystem'  # 设置session的存储类型
    # SESSION_REDIS = StrictRedis(host=REDIS_HOST, port=REDIS_PORT)  # 指定session存储的服务器
    # SESSION_USE_SIGNER = True  # 设置签名存储
    # PERMANENT_SESSION_LIFETIME = timedelta(days=1)  # 设置签名过期时间

    # 配置默认的log等级
    LEVEL_NAME = logging.DEBUG


# 开发环境配置信息
class DevelopConfig(Config):
    pass


# 生产(线上)环境配置信息
class ProductConfig(Config):
    DEBUG = False


# 测试环境配置信息
class TestConfig(Config):
    pass


# 提供一个统一的访问入口
config_dict = {
    "develop": DevelopConfig,
    "product": ProductConfig,
    "test": TestConfig
}
二、__init__.py
import logging
from logging.handlers import RotatingFileHandler

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from redis import StrictRedis
from flask_session import Session

from config import Config, config_dict

db = SQLAlchemy()
redis_store = None


def create_app(config_name):
    app = Flask(__name__)

    # 获取config配置
    config = config_dict.get(config_name)

    app.config.from_object(config)

    # 调用日志方法,记录程序运行信息
    log_file(config.LEVEL_NAME)

    # 创建数据库关联对象并关联app
    db.init_app(app)

    # 创建redis对象
    # 当 decode_responses 设置为 True 时,Redis 返回的字符串数据将会被解码为 Python 字符串类型。这样可以方便地处理 Redis 中存储的文本数据。
    # 而当 decode_responses 设置为 False(默认值)时,Redis 返回的字符串数据将会以字节字符串(bytes)的形式返回。
    # 这在处理二进制数据或者需要与其他 Redis 客户端进行交互时可能更为合适
    global redis_store
    redis_store = StrictRedis(host=config.REDIS_HOST, port=config.REDIS_PORT, decode_responses=True)

    # # 创建session对象
    # Session(app)

    # 注册蓝图
    from manger.user import user_blue
    app.register_blueprint(user_blue)

    return app
三、app.py
from manger import create_app, db
from flask_migrate import Migrate
from manger import models

# 传入settings参数,开发版本“develop”,线上版本“product”
app = create_app('develop')

# 数据库设置
migrate = Migrate(app, db)


@app.route('/')
def hello_world():  # put application's code here
    return 'Hello World!'


if __name__ == '__main__':
    app.run()
四、views中使用,登录
@user_blue.route('/login', methods=['POST'])
def login():
    username = request.json.get('username')
    password = request.json.get('password')
    if not all([username, password]):
        return JsonResponse.error(msg='缺少必要参数!').to_response()
    try:
        user_obj = models.User.query.filter(models.User.name == username).first()
    except Exception as e:
        current_app.logger.error(e)
    if not user_obj.check_password(password):
        return JsonResponse.error(msg='密码错误!').to_response()

    token = encode_token(user_obj.id, username, password)
    # 将token存入redis中
    redis_store.set('token', token)
    # 设置过期时间
    redis_store.expire('token', 20)  
    data = {
        "data": user_obj.to_dict(),
        "token": token
    }

    return JsonResponse(code=200, data=data, msg='登录成功!').to_response()

文章来源:https://blog.csdn.net/qq_44906497/article/details/134849933
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。