2024-03-29 10:22:15 +08:00

27 lines
824 B
Python

import redis
from redis.connection import Connection, SSLConnection
redis_client = redis.Redis()
def init_app(app):
connection_class = Connection
if app.config.get("REDIS_USE_SSL", False):
connection_class = SSLConnection
redis_client.connection_pool = redis.ConnectionPool(
**{
"host": app.config.get("REDIS_HOST", "localhost"),
"port": app.config.get("REDIS_PORT", 6379),
"username": app.config.get("REDIS_USERNAME", None),
"password": app.config.get("REDIS_PASSWORD", None),
"db": app.config.get("REDIS_DB", 0),
"encoding": "utf-8",
"encoding_errors": "strict",
"decode_responses": False,
},
connection_class=connection_class,
)
app.extensions["redis"] = redis_client