pyMysql封装

该内容为针对pymysql的封装,具体的调用 清纤细见函数的注释即可

import pymysql


class sql_server:
    def __init__(self, link: str = "127.0.0.1:3306 root 123456 cut_picture_server"):
        """数据库处理类
        :param link:  格式见默认 地址:端口号 用户名 密码 数据库名
        """
        link_li = link.split(" ")
        self.link = link_li[0].split(":") + link_li[1:]

    def get_db(self):
        """【非公开方法==>外部不要调用】 获取数据库的方法
        :return: 返回数据库对象,注意返回的对象用完之后要及时的close一下
        """
        return pymysql.connect(host=self.link[0], port=int(self.link[1]), user=self.link[2],
                               password=self.link[3], database=self.link[4], charset="utf8")

    def do(self, sql: str, args=()) -> bool:
        """对数据库进行插入,删除,更新之类的操作
        :param args:  %s 表示的参数,用于图片的上传
        :param sql:  sql语句
        :return: 返回值为操作是否成功
        """
        ret = True
        db = self.get_db()
        cursor = db.cursor()
        try:
            cursor.execute(sql, args=args)
            db.commit()
        except Exception as e:
            db.rollback()
            ret = False
            print("数据库异常", e, sql)
        cursor.close()
        db.close()
        return ret

    def select(self, sql: str) -> list:
        """对数据库查询的方法
        :param sql: sql查询语句
        :return: 返回为二位列表,若为空,则返回为空列表
        """
        ret = []
        db = self.get_db()
        cursor = db.cursor()
        try:
            cursor.execute(sql)
            ret = [list(row) for row in cursor.fetchall()]
        except Exception as e:
            print("数据库报错", e, sql)
        cursor.close()
        db.close()
        return ret

    def title_select(self, title: str, sql: str) -> list:
        """ 查询后,将表格按照title进行转化
        :param title: 标题[字符串类型]通过逗号分割
        :param sql: 查询的sql语句
        :return: 返回值为一个列表,列表中是字典元素 类似于这样 [{"id": 1}, {"id", 2}]
        """
        ret = []
        db = self.get_db()
        cursor = db.cursor()
        try:
            cursor.execute(sql)
            tab = [list(row) for row in cursor.fetchall()]
            title_list = title.split(",")
            for row_li in tab:
                i = min(len(row_li), len(title_list))
                if i:
                    ret.append({key: val for key, val in zip(title_list[: i], row_li[: i])})
        except Exception as e:
            print("数据库报错", e, sql)
        cursor.close()
        db.close()
        return ret

    def insert(self, sql: str, args=()):
        """数据库进行插入
        :param args: %s代表,用于存储文件类型的数据
        :param sql:  插入的语句
        :return:  返回值为主键!
        """
        ret = -1
        db = self.get_db()
        cursor = db.cursor()
        try:
            if args:
                cursor.execute(sql, args=args)
            else:
                cursor.execute(sql)
            ret = db.insert_id()
            db.commit()
        except Exception as e:
            db.rollback()
            print("数据库异常", e, sql)
        cursor.close()
        db.close()
        return ret

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值