67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
|
"""
|
|||
|
|
redis 配置类
|
|||
|
|
"""
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
import redis
|
|||
|
|
|
|||
|
|
from ..configs import default
|
|||
|
|
|
|||
|
|
|
|||
|
|
class RedisCli(object):
|
|||
|
|
|
|||
|
|
def __init__(self, *, host: str, port: str, password: str, db: int, socket_timeout=5):
|
|||
|
|
# redis对象，在 @app.on_event('startup') 中连接创建
|
|||
|
|
self._redis_client = None
|
|||
|
|
self.host = host
|
|||
|
|
self.port = port
|
|||
|
|
self.password = password
|
|||
|
|
self.db = db
|
|||
|
|
self.socket_timeout = socket_timeout
|
|||
|
|
|
|||
|
|
def init_redis_connect(self) -> None:
|
|||
|
|
"""
|
|||
|
|
初始化链接
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
r = redis.ConnectionPool(
|
|||
|
|
host=self.host,
|
|||
|
|
port=self.port,
|
|||
|
|
password=self.password,
|
|||
|
|
db=self.db,
|
|||
|
|
socket_timeout=self.socket_timeout,
|
|||
|
|
decode_responses=True # 解码
|
|||
|
|
)
|
|||
|
|
self._redis_client = redis.StrictRedis(connection_pool=r)
|
|||
|
|
if not self._redis_client.ping():
|
|||
|
|
print('连接超时')
|
|||
|
|
sys.exit()
|
|||
|
|
except (redis.AuthenticationError, Exception) as e:
|
|||
|
|
print('连接redis异常')
|
|||
|
|
sys.exit()
|
|||
|
|
|
|||
|
|
def get_redis(self):
|
|||
|
|
return self._redis_client
|
|||
|
|
|
|||
|
|
# 使实例化后的对象，赋予redis对象的方法和属性
|
|||
|
|
def __getattr__(self, item):
|
|||
|
|
return self._redis_client.has_key(item)
|
|||
|
|
|
|||
|
|
def __getitem__(self, item):
|
|||
|
|
return self._redis_client[item]
|
|||
|
|
|
|||
|
|
def __setitem__(self, key, value):
|
|||
|
|
self._redis_client[key] = value
|
|||
|
|
|
|||
|
|
def __delitem__(self, key):
|
|||
|
|
del self._redis_client[key]
|
|||
|
|
|
|||
|
|
|
|||
|
|
redis_client = RedisCli(
|
|||
|
|
host=default.db['host'],
|
|||
|
|
port=default.db['port'],
|
|||
|
|
password=default.db['password'],
|
|||
|
|
db=default.db['db'],
|
|||
|
|
)
|