Python3操作MySQL数据库

傷城~ 2022-04-22 15:36 296阅读 0赞

# 核心内容 #

（1）使用到的模块：PyMySQL

安装：

pip3 install pymysql

（2）连接数据库

# 连接数据库,地址：localhost,账号：root,密码：123,数据库：test,编码：UTF-8
    db = pymysql.connect('localhost', 'root', '123', 'test', charset='utf8')

（3）创建游标对象

# 创建一个游标对象
    cursor = db.cursor()

（4）执行SQL语句

cursor.execute(SQL)

（5）事务提交、回滚

#提交事务
    db.commit()
    #事务回滚
    db.rollback()

（6）获取结果（**返回的类型为元组**）

*  获取一条数据

cursor.fechone()

*  获取所有的数据

cursor.fechall()

*  查看执行SQL语句后影响到的行数

cursor.rowcount

（7）关闭游标和连接

#关闭游标
    cursor.close()
    #关闭数据库连接
    db.close()

# 增删改查 #

### 1、添加数据 ###

# insertData_mysql.py
    import pymysql
    
    # 连接数据库,地址：localhost,账号：root,密码：123,数据库：test,
    # 编码：UTF-8
    db = pymysql.connect('localhost', 'root', '123', 'test', charset='utf8')
    
    # 创建一个游标对象
    cursor = db.cursor()
    
    # SQL语句
    SQL = '''
          INSERT INTO STUDENT VALUES('2016081111','张三'),('2016081112','李四'),('2016081113','王五');
          '''
    try:
        # 插入数据
        cursor.execute(SQL)
        # 提交到数据库
        db.commit()
        print('插入成功')
    except:
        # 插入失败
        db.rollback()
        print('插入失败')
    
    #关闭游标
    cursor.close()
    #关闭数据库连接
    db.close()

### 2、查询数据 ###

# selectData_mysql.py
    import pymysql
    
    # 连接数据库,地址：localhost,账号：root,密码：123,数据库：test,
    # 编码UTF-8
    db = pymysql.connect('localhost', 'root', '123', 'test', charset='utf8')
    
    # 创建一个游标对象
    cursor = db.cursor()
    
    # SQL语句
    SQL = '''
          SELECT * FROM STUDENT;
          '''
    try:
        # 查询所有的数据
        cursor.execute(SQL)
        # 查询到的条数
        print('查询到了%s条数据' % (cursor.rowcount))
        # 获取一条数据
        one = cursor.fetchone()
        print(type(one))
        print(one)
        # 获取所有数据
        for item in cursor.fetchall():
            print(item)
    except:
        # 插入失败
        print('查询失败')
    
    #关闭游标
    cursor.close()
    #关闭数据库连接
    db.close()

执行结果

查询到了3条数据
    <class 'tuple'>
    ('2016081111', '张三')
    ('2016081112', '李四')
    ('2016081113', '王五')

### 3、修改数据 ###

# updateData_mysql.py
    import pymysql
    
    # 连接数据库,地址：localhost,账号：root,密码：123,数据库：test
    db = pymysql.connect('localhost', 'root', '123', 'test', charset='utf8')
    
    # 创建一个游标对象
    cursor = db.cursor()
    
    SNAME = '李华'
    # 更新数据SQL语句
    UPDATE_SQL = '''
          UPDATE STUDENT SET SNAME='%s' WHERE SNO='2016081111';
          ''' % (SNAME)
    # 查询数据SQL语句
    SELECT_SQL = '''
                SELECT * FROM STUDENT WHERE SNO='%s'
               ''' % ('2016081111')
    try:
        # 更新数据前
        cursor.execute(SELECT_SQL)
        print('更新前:', cursor.fetchall())
        # 更新数据
        cursor.execute(UPDATE_SQL)
        # 提交到数据库
        db.commit()
        # 更新数据后
        cursor.execute(SELECT_SQL)
        print('更新后:', cursor.fetchall())
    except:
        # 更新失败,回滚
        db.rollback()
        print('更新失败')
    
    #关闭游标
    cursor.close()
    #关闭数据库连接
    db.close()

执行结果：

更新前: (('2016081111', '张三'),)
    更新后: (('2016081111', '李华'),)

### 4、删除数据 ###

# deleteData_mysql.py
    import pymysql
    
    # 连接数据库,地址：localhost,账号：root,密码：123,数据库：test
    db = pymysql.connect('localhost', 'root', '123', 'test', charset='utf8')
    
    # 创建一个游标对象
    cursor = db.cursor()
    
    SNO = '2016081111'
    # 删除数据SQL语句
    DELETE_SQL = '''
          DELETE FROM STUDENT WHERE SNO='%s';
          ''' % (SNO)
    # 查询数据SQL语句
    SELECT_SQL = '''
                SELECT * FROM STUDENT WHERE SNO='%s'
               ''' % (SNO)
    try:
        # 删除数据前
        cursor.execute(SELECT_SQL)
        print('删除前:', cursor.fetchall())
        # 删除数据
        cursor.execute(DELETE_SQL)
        # 提交到数据库
        db.commit()
        # 删除数据后
        cursor.execute(SELECT_SQL)
        print('删除后:', cursor.fetchall())
    except:
        # 删除失败，回滚
        db.rollback()
        print('删除失败')
    
    #关闭游标
    cursor.close()
    #关闭数据库连接
    db.close()

执行结果：

删除前: (('2016081111', '张三'),)
    删除后: ()