pymysql 库的使用

pymysql库的使用

1.安装

pip install pymysql -i https://mirrors.aliyun.com/pypi/simple

MySQL 的安装可以按照操作系统来操作。

2.初始化

初始化连接 MySQL 数据库
host：MySQL 安装地址，如果是本地就是 localhost 或者 127.0.0.1。
user：用户名默认值：root。
password：在安装的过程中输入的密码。
port：端口号，默认就是 3306。
database：输入创建的数据库

import pymysql

db = pymysql.connect(host=host, user=user, password=password, port=port, database=database)

3.创建游标

完成上述操作已经就是连接上数据库了，想要操作数据还需要建立游标

cursor = db.cursor()  # 创建游标用来执行SQL 语句

4.插入数据

插入、更新和删除操作都是对数据库进行更改的操作，而更改操作都必须是一个事务，所以标准写法需要加上异常处理：commit或是rollback

sql = "insert into students (id, user_name, age) values ('1', '张三', 20)"
try:
	cursor.execute(sql)
	db.commit()
except:
	db.rollback()
db.close()

4.1.动态插入SQL

# 创建对应表结构的字典
data = {
'id': 2,
'user_name': '李四',
'age': 30,
}
table = 'students'
keys = ', '.join(data.keys())
values = ', '.join(['%s'] * len(data))  # 此处返回的是[%s, %s, %s]
sql = f'insert into {table} ({keys}) values ({values})'
try:
    if cursor.execute(sql, tuple(data.values())):
        print('成功')
        db.commit()
except:
    print('错误')
    db.rollback()

5.更新数据

sql = 'update students set age = %s where user_name = %s'
try:
	cursor.execute(sql, (25, '张三'))
	db.commit()
except:
	db.rollback()

6.删除数据

sql = 'delete from students where id = 2;'
try:
	cursor.execute(sql)
	db.commit()
except:
	db.rollback()

7.查询数据

rowcount：可以获取，这条查询语句有多少行。
fetchone：获取第一条数据，内部有一个偏移指针，调用了 fetchone 之后指针移动到第二条了，所以最后的 fetchall 不是三个。
fetchall：一次性获取查询语句的所有结果。

sql = 'select * from students'
try:
    cursor.execute(sql)
    print('数量：', cursor.rowcount)
    result = cursor.fetchone()
    print('第一个条：', result)
    results = cursor.fetchall()
    print(type(results), results)
except:
    print('Error')

可以通过下面的方法逐条获取

cursor.execute(sql)
row = cursor.fetchone()
while row:
	print('result：', row)
	row = cursor.fetchone()

关注公众号可以领取python 学习相关资料

公众号：爬虫探索者或者pctansuo

完整代码

import pymysql

db = pymysql.connect(host='localhost', user='root', password='ambition', port=3306, database='spiders')

cursor = db.cursor()  # 创建游标用来执行SQL 语句

sql = "insert into students (id, user_name, age) values ('1', '张三', 20)"
try:
    cursor.execute(sql)
    db.commit()
except:
    db.rollback()

data = {
    'id': '2',
    'user_name': '李四',
    'age': 30,
}
table = 'students'
keys = ', '.join(data.keys())
values = ', '.join(['%s'] * len(data))  # 此处返回的是[%s, %s, %s]
sql = f'insert into {table} ({keys}) values ({values})'
try:
    if cursor.execute(sql, tuple(data.values())):
        print('成功')
        db.commit()
except:
    print('错误')
    db.rollback()

sql = 'update students set age = %s where user_name = %s'
try:
    cursor.execute(sql, (25, '张三'))
    db.commit()
except:
    db.rollback()

sql = 'delete from students where id = 2;'
try:
    cursor.execute(sql)
    db.commit()
except:
    db.rollback()

sql = 'select * from students'
try:
    cursor.execute(sql)
    print('数量：', cursor.rowcount)
    result = cursor.fetchone()
    print('第一个条：', result)
    results = cursor.fetchall()
    print(type(results), results)
except:
    print('Error')

cursor.execute(sql)
row = cursor.fetchone()
while row:
    print('result：', row)
    row = cursor.fetchone()
db.close()  # 关闭数据库连接