Python项目Flask框架整合Redis

一、在配置文件中创建Redis连接信息

二、 实现Redis配置类

import redis
from config.config import REDIS_HOST, REDIS_PORT, REDIS_PASSWD, REDIS_DB, EXPIRE_TIME


class RedisDb():
    def __init__(self, REDIS_HOST, REDIS_PORT, REDIS_DB, EXPIRE_TIME, REDIS_PASSWD):
        # 建立数据库连接
        self.r = redis.Redis(
            host=REDIS_HOST,
            port=REDIS_PORT,
            password=REDIS_PASSWD,
            db=REDIS_DB,
            decode_responses=True  # get() 得到字符串类型的数据
        )
    #保存token
    def handle_redis_token(self, key, value=None):
        if value:  # 如果value非空，那么就设置key和value，EXPIRE_TIME为过期时间
            self.r.set(key, value, ex=EXPIRE_TIME)
        else:  # 如果value为空，那么直接通过key从redis中取值
            redis_token = self.r.get(key)
            return redis_token


redis_db = RedisDb(REDIS_HOST, REDIS_PORT, REDIS_DB, EXPIRE_TIME, REDIS_PASSWD)

三、在目标文件中引入并使用

from config.RedisConfig import redis_db

"""
测试专用
"""
@bi.route('/ssss', methods=['GET'])
def get_test():

    redis_db.handle_redis_token("key","value")

    return jsonify({
        "code": 0,
        "msg": "success",
        "data": ""
    })