2023-简单点-proxyPool源码（四)-dbClient.py

# 导入os模块，该模块提供了与操作系统交互的功能  
import os  
  
# 导入sys模块，该模块提供了与Python解释器交互的功能  
import sys  
  
# 从util.six模块中导入urlparse函数，该函数用于解析URL  
# 注意：Python 3 中 urlparse 已经在 urllib.parse 模块下，这里可能是为了兼容 Python 2 和 3 的写法  
from util.six import urlparse  
  
# 从util.six模块中导入withMetaclass函数，该函数用于创建带有元类的类  
from util.six import withMetaclass  
  
# 从util.singleton模块中导入Singleton类，该类是一个实现单例模式的基类  
from util.singleton import Singleton  
  
# 将当前文件的绝对路径的目录添加到sys.path中，这样可以方便地导入同一目录下的其他模块或包  
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

class DbClient(withMetaclass(Singleton)):
	# 定义DbClient类，使用Singleton作为元类，确保该类只有一个实例 
    """
    DbClient DB工厂类 提供get/put/update/pop/delete/exists/getAll/clean/getCount/changeTable方法


    抽象方法定义：
        get(): 随机返回一个proxy;
        put(proxy): 存入一个proxy;
        pop(): 顺序返回并删除一个proxy;
        update(proxy): 更新指定proxy信息;
        delete(proxy): 删除指定proxy;
        exists(proxy): 判断指定proxy是否存在;
        getAll(): 返回所有代理;
        clean(): 清除所有proxy信息;
        getCount(): 返回proxy统计信息;
        changeTable(name): 切换操作对象


        所有方法需要相应类去具体实现：
            ssdb: ssdbClient.py
            redis: redisClient.py
            mongodb: mongodbClient.py

 	"""
   
    # 初始化方法，接受数据库连接字符串  
    def __init__(self, db_conn):  
        """  
        初始化方法  
        :return:  
        """  
        # 解析数据库连接字符串，得到数据库的类型、主机名、端口、用户名、密码和数据库名  
        self.parseDbConn(db_conn)  
        # 初始化数据库客户端  
        self.__initDbClient()  
  
    # 类方法，用于解析数据库连接字符串  
    @classmethod  
    def parseDbConn(cls, db_conn):  
        # 使用urlparse解析连接字符串  
        db_conf = urlparse(db_conn)  
        # 提取并保存数据库的类型、主机名、端口、用户名、密码和数据库名  
        cls.db_type = db_conf.scheme.upper().strip()  
        cls.db_host = db_conf.hostname  
        cls.db_port = db_conf.port  
        cls.db_user = db_conf.username  
        cls.db_pwd = db_conf.password  
        cls.db_name = db_conf.path[1:]  
        # 返回类本身，允许链式调用  
        return cls  
  
    # 初始化数据库客户端的方法  
    def __initDbClient(self):  
        """  
        初始化数据库客户端  
        :return:  
        """  
        # 根据数据库类型确定要使用的客户端类型  
        __type = None  
        if "SSDB" == self.db_type:  
            __type = "ssdbClient"  
        elif "REDIS" == self.db_type:  
            __type = "redisClient"  
        else:  
            pass  
        # 确保找到了对应的客户端类型  
        assert __type, '类型错误，不支持的数据库类型: {}'.format(self.db_type)  
        # 动态导入对应的客户端模块，并创建客户端实例  
        self.client = getattr(__import__(__type), "%sClient" % self.db_type.title())(host=self.db_host,  
                                                                                     port=self.db_port,  
                                                                                     username=self.db_user,  
                                                                                     password=self.db_pwd,  
                                                                                     db=self.db_name)  
  
    # 数据库get操作的方法，接受key和其他可选参数，返回get的结果  
    def get(self, https, **kwargs):  
        return self.client.get(https, **kwargs)  
  
    # 数据库put操作的方法，接受key和value以及其他可选参数，返回put的结果  
    def put(self, key, **kwargs):  
        return self.client.put(key, **kwargs)  
  
    # 数据库update操作的方法，接受key和value以及其他可选参数，返回update的结果  
    def update(self, key, value, **kwargs):  
        return self.client.update(key, value, **kwargs)  
  
    # 数据库delete操作的方法，接受key和其他可选参数，返回delete的结果  
    def delete(self, key, **kwargs):  
        return self.client.delete(key, **kwargs)  
  
    # 检查key是否存在的方法，接受key和其他可选参数，返回存在的结果  
    def exists(self, key, **kwargs):  
        return self.client.exists(key, **kwargs)  
  
    # 数据库pop操作的方法，接受key和其他可选参数，返回pop的结果  
    def pop(self, https, **kwargs):  
        return self.client.pop(https, **kwargs)  
  
    # 获取所有数据的方法，接受https参数，返回所有数据  
    def getAll(self, https):  
        return self.client.getAll(https) 
        def clear(self):
        return self.client.clear()

    def changeTable(self, name):
        self.client.changeTable(name)

    def getCount(self):
        return self.client.getCount()

    def test(self):
        return self.client.test()