目录

mysql-toc" style="margin-left:0px;">mysql

redis-toc" style="margin-left:0px;">redis

mongo

---

mysql">mysql

python">class AssertUtil:
    # 连接数据库
    def conn_mysql(self):
        self.conn = pymysql.connect(
            user="user",
            password="pwd",
            host="127.0.0.1",
            database="sd",
            port=3306
        )
        return self.conn

    # 执行sql语句
    def execute_sql(self,sql):
        # 创建数据库链接
        conn = self.conn_mysql()
        # 创建游标
        cs = conn.cursor()
        # 执行SQL
        cs.execute(sql)
        # 取值
        value = cs.fetchone()
        # 关闭资源
        cs.close()
        conn.close()
        # 返回值
        return value
        
if __name__ == "__main__":
    a = AssertUtil().execute_sql("SELECT email FROM pw_user WHERE uid = 1")
    print(a)
    print(type(a))

redis">redis

python">class AssertUtil:
    # 连接redis
    def connect(self, host, port, password, db):
        self.connection = redis.Redis(
            host=host,
            port=port,
            password=password,
            db=db,
            decode_responses=True  # 将字节数据解码为字符串
        )
    # 操作redis
    def execute_command(self, command, *args):
        """
        执行Redis命令
        :param command: Redis命令
        :param args: 命令参数
        :return: 执行结果
        """
        if not hasattr(self, 'connection'):
            raise Exception("redis未连接")
        res = getattr(self.connection, command)(*args)

        # 关闭连接
        self.connection.close()
        return res

if __name__ == "__main__":
    redis_connector = AssertUtil()
    redis_connector.connect(host="192.168.0.1", port=6379, password="111111", db=0)
    # String 操作
    redis_connector.execute_command("set", "name", "4")
    print('string:'+ redis_connector.execute_command('get','name'))
    # list操作
    redis_connector.execute_command("lpush", "mylist", "item3","item4")
    redis_connector.execute_command("lpop", "mylist")
    print(redis_connector.execute_command('lrange', 'mylist',0,-1))
    # Hash 操作
    redis_connector.execute_command("hset", "myhash", "field1", "value1")
    print('hash:'+ redis_connector.execute_command("hget", "myhash", "field1"))
    print(redis_connector.execute_command('hgetall', 'myhash'))
    # Set 操作
    redis_connector.execute_command("sadd", "myset", "value1", "value2")
    print(redis_connector.execute_command("smembers", "myset"))
    # ZSet 操作
    redis_connector.execute_command("zadd", "myzset", {"value1": 1}, {"value2": 2})
    print(redis_connector.execute_command("zrangebyscore", "myzset", 0, 2))
    # 删除操作
    redis_connector.execute_command("delete","myhash")

mongo

python">class AssertUtil:
    def conn_mongo(self, database, host, port, username, password, auth_source):
    self.database = database
    self.client = MongoClient(
        host=host,
        port=port,
        username=username,
        password=password,
        authSource=auth_source
    )
    self.db = self.client[self.database]

def execute_mongo(self,collection, command,reslove, *args):
    """
    执行MongoDB命令
    :param database: 数据库名称
    :param collection: 集合名称
    :param command: MongoDB命令
    :param args: 命令参数
    :return: 执行结果
    """
    if not hasattr(self, 'client'):
        raise Exception("MongoDB未连接")
    self.col = self.db[collection]
    try:
        func = getattr(self.col, command)
        result = func(*args)
        if callable(reslove):
            return reslove(result)
        return result
    finally:
        self.client.close()
if __name__ == "__main__":
    mongo_connector = AssertUtil()
    mongo_connector.conn_mongo(database="test", host="192.168.0.1", port=27017, username="admin", password="1111", auth_source="admin")
    # 插入文档
    mongo_connector.execute_mongo("insert_one",None,{"model_id": "gpt","nickname": "ces","avatar": "1111"})
    # 查询文档
    print(mongo_connector.execute_mongo("test1","find_one",None,{"model_id": "gpt-3.5-turbo"}))
    # 查询多条文档
    def reslove(results):
        for result in results:
            print(result)
    res = mongo_connector.execute_mongo("test1","find", reslove,{"model_id": "gpt"})
    # 更新文档
    mongo_connector.execute_mongo("test1","update_one", None,{"model_id": "gpt11"}, {"$set": {"nickname": "aaaaa"}})
    # 删除文档
    mongo_connector.execute_mongo("test1","delete_one",  None,{"model_id": "model_id"})