2.2 python数据库-mysql

以mysql为例，数据库的操作包括：连接、断开、新增库表、新增数据、查询数据、删除数据和修改数据

1. 连接与关闭

python 复制代码

#!/usr/bin/python3
 
import pymysql

db = pymysql.connect(host='localhost',
                     user='testuser',
                     password='test123',
                     database='TESTDB')

# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()
 
# 使用 execute()  方法执行 SQL 查询 
cursor.execute("SELECT VERSION()")
 
# 使用 fetchone() 方法获取单条数据.
data = cursor.fetchone()
 
print ("Database version : %s " % data)
 
# 关闭数据库连接
db.close()

特别注意，使用完后记得调close关闭连接，否则mysql数据库将维持这个连接直到超时断开。

当然，可以结合装饰器或者with、finnaly等关键字，实现兜底。这里不做展开了。

2. 创建库表

python 复制代码

# 使用预处理语句创建表
sql = """CREATE TABLE EMPLOYEE (
         FIRST_NAME  CHAR(20) NOT NULL,
         LAST_NAME  CHAR(20),
         AGE INT,  
         SEX CHAR(1),
         INCOME FLOAT )"""
 
cursor.execute(sql)

3. 查询数据

Python查询Mysql使用 fetchone() 方法获取单条数据, 使用fetchall() 方法获取多条数据。

fetchone(): 该方法获取下一个查询结果集。结果集是一个对象
**fetchall():**接收全部的返回结果行.
rowcount: 这是一个只读属性，并返回执行execute()方法后影响的行数。

python 复制代码

# SQL 查询语句
sql = "SELECT * FROM EMPLOYEE \
       WHERE INCOME > %s" % (1000)
try:
   # 执行SQL语句
   cursor.execute(sql)
   # 获取所有记录列表
   results = cursor.fetchall()
   for row in results:
      fname = row[0]
      lname = row[1]
      age = row[2]
      sex = row[3]
      income = row[4]
       # 打印结果
      print ("fname=%s,lname=%s,age=%s,sex=%s,income=%s" % \
             (fname, lname, age, sex, income ))
except:
   print ("Error: unable to fetch data")

4. 新增数据

python 复制代码

# SQL 插入语句
sql = """INSERT INTO EMPLOYEE(FIRST_NAME,
         LAST_NAME, AGE, SEX, INCOME)
         VALUES ('Mac', 'Mohan', 20, 'M', 2000)"""
try:
   # 执行sql语句
   cursor.execute(sql)
   # 提交到数据库执行
   db.commit()
except:
   # 如果发生错误则回滚
   db.rollback()

注意：commit方法是用于提交db事务的，主要是ddl操作会自动提交，dml操作不会

这里推荐大家看Python操作Mysql_pymysql commit-CSDN博客这篇文章

5. 修改数据

python 复制代码

# SQL 更新语句
sql = "UPDATE EMPLOYEE SET AGE = AGE + 1 WHERE SEX = '%c'" % ('M')
try:
   # 执行SQL语句
   cursor.execute(sql)
   # 提交到数据库执行
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()

6. 删除数据

python 复制代码

# SQL 删除语句
sql = "DELETE FROM EMPLOYEE WHERE AGE > %s" % (20)
try:
   # 执行SQL语句
   cursor.execute(sql)
   # 提交修改
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()

关于mysql的一些基本知识和语法，本文不做展开；另外，大部分时候，我们用到的框架例如django等，都会继承orm，会有一套操作方式，至少目前，用到pymysql操作的，都还是脚本编写时。如果是服务化或常驻化的场景，不推荐使用。

后续章节中会分享django orm相关内容。

参考

Python3 MySQL 数据库连接 -- PyMySQL 驱动 | 菜鸟教程

Python操作Mysql_pymysql commit-CSDN博客