import pymysql

class mysql(object):
    def insert_db(self, insert_sql):
        """插入"""
        # 建立数据库连接
        db = pymysql.connect(
            host="127.0.0.1",
            port=3306,
            user="root",
            passwd="root",
            db="test"
        )
        # 通过 cursor() 创建游标对象
        cur = db.cursor()
        try:
            # 使用 execute() 执行sql
            cur.execute(insert_sql)
            # 提交事务
            db.commit()
        except Exception as e:
            print("操作出现错误：{}".format(e))
            # 回滚所有更改
            db.rollback()
        finally:
            # 关闭游标
            cur.close()
            # 关闭数据库连接
            db.close()

    def query_db(self, query_sql, ball):
         # 建立数据库连接
        db = pymysql.connect(
            host="127.0.0.1",
            port=3306,
            user="root",
            passwd="root",
            db="test"
        )
        # 通过 cursor() 创建游标对象
        cur = db.cursor()
        try:
            # 使用 execute() 执行sql
            cur.execute(query_sql)
            # fetchall() 获取所有记录
            if ball:
                result = cur.fetchall()
            else:
                result = cur.fetchone()

        except Exception as e:
            print("操作出现错误：{}".format(e))
            # 回滚所有更改
            db.rollback()
        finally:
            # 关闭游标
            cur.close()
            # 关闭数据库连接
            db.close()

        return result