【python】redis select

最新推荐文章于 2025-06-23 19:20:32 发布

原创最新推荐文章于 2025-06-23 19:20:32 发布 · 845 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python

redis切换db 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种通过封装Redis模块以实现select功能的方法，详细展示了如何通过Python创建一个可切换数据库的Redis数据访问类，包括连接池的初始化及数据库切换的实现。

部署运行你感兴趣的模型镜像

说明：redis模块本身不支持select, 所以通过内部封装变向的来实现select功能，具体代码如下：

Github

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""redis数据访问类"""

import redis
from .Env import Env

class Redis():
    _instance = dict()

    def __init__(self, host=None, port=6379, password=None, db=0):
        '''
        构造方法
        :param host: host
        :param port: port
        :param password: 密码
        :param db: 默认db
        '''
        env = Env()
        if host is not None:
            self.host = host
            self.port = port
            self.password = password
            self.db = db
        else:					# 读取配置文件的默认配置
            self.host = env("REDIS_HOST", 'localhost')
            self.port = int(env("REDIS_PORT", 6379))
            self.password = env('REDIS_PASSWORD', None)
            self.db = int(env("REDIS_DB", 0))

        if self._instance.get(self.db) is None:
            pool = redis.ConnectionPool(host=self.host, port=self.port, db=self.db, password=self.password, decode_responses=True)
            self._instance[self.db] = redis.Redis(connection_pool=pool)

    def select(self, db):
        '''
        切换redis库
        :param db: 库index
        :return: redis.Redis
        '''
        if type(db) != int:
            raise Exception('select db must be the type of int')
        if self._instance.get(db) is None:
            pool = redis.ConnectionPool(host=self.host, port=self.port, db=db, password=self.password, decode_responses=True)
            self._instance[db] = redis.Redis(connection_pool=pool)
        return self._instance[db]

    def __getattr__(self, key):
        '''
        默认库则调用
        :param key: redis 执行方法
        :return: result
        '''
        return self._instance.get(self.db).__getattribute__(key)

if __name__ == '__main__':
    redis = Redis(host='127.0.0.1', port=6379)
    name = redis.select(8).get('name')