python 操作 MySQL （pymysql）

综述

pymysql.png

查询

from pymysql import *


def main():
    # 创建 connection 对象，连接数据库
    conn = connect(host='localhost', port=3306, database='your_db', user='root', password='your_psw', charset='utf8')
    # 获取 cursor 对象
    cs1 = conn.cursor()
    
    # 调用 execute 方法，来执行 sql 语句
    # execute 方法，默认返回值为 生效行数
    count = cs1.execute('select id,name from goods where id>=4')
    # 调用 cursor 的 fetchone() 逐个查询结果、fetchmany(3) 查询3行，fetchall() 查询所有
    for i in range(count):
        result = cs1.fetchone()
        print(result)
    
    # 关闭 cursor 对象，关闭 connection 连接
    cs1.close()
    conn.close()


if __name__ == '__name__':
    main()

增删改

因为增删改会修改数据库，所以，除了要写不一样的 SQL 语句，还要多加一个提交的步骤。

from pymysql import *


def main():
    # 创建 connection 对象，连接数据库
    conn = connect(host='localhost', port=3306, database='your_db', user='root', password='your_psw', charset='utf8')
    # 获取 cursor 对象
    cs1 = conn.cursor()

    # 增加
    count = cs1.execute('insert into goods_cates (name) values("硬盘")')
    # 更新
    count = cs1.execute('update goods_cates set name="机械硬盘" where name="硬盘"')
    # 删除
    count = cs1.execute('delete from good_cates where id=6')

    # 提交之前的操作
    # 如果之前已经执行过多次的 execute，那么就都进行提交
    conn.commit()
    # 撤销之前的操作，回滚
    # conn.rollback()

    # 关闭 cursor 对象，关闭 connection 连接
    cs1.close()
    conn.close()


if __name__ == '__main__':
    main()