Python3中操作MySQL数据库

在Python3中使用mysql数据库需要安装pymysql库。

pip install pymysql

操作MySQL

导包

import pymysql

第一步：打开数据库连接

db = pymysql.connect(host="数据库地址", 
                     user="用户名", 
                     password="密码", 
                     port=3306,# 端口  
                     database="数据库名", 
                     charset='utf8')

第二步：创建游标

cursor = db.cursor()

第三步：操作数据库

1、创建表

# 如果数据表已经存在使用execute()方法删除表。
cursor.execute("DROP TABLE IF EXISTS EMPLOYEE")

# 创建数据表SQL语句
sql = """CREATE TABLE EMPLOYEE (
         FIRST_NAME  CHAR(20) NOT NULL,
         LAST_NAME  CHAR(20),
         AGE INT,  
         SEX CHAR(1),
         INCOME FLOAT )"""

cursor.execute(sql)

2、添加数据

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

3、查询数据

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

2、fetchone(): 该收全部的返回结果行.

3、rowcount: 这是方法获取下一个查询结果集。结果集是一个对象

4、fetchall():接一个只读属性，并返回执行execute()方法后影响的行数。

# SQL 查询语句
sql = "SELECT * FROM EMPLOYEE WHERE INCOME > {}".format(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={},lname={},age={},sex={},income={}".format(fname, lname, age, sex, income))

except:
    print("Error: unable to fecth data")

4、修改数据

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

5、删除数据

# SQL 刪除语句
sql = "DELETE FROM EMPLOYEE WHERE AGE > {}".format(20)
try:
    cursor.execute(sql)
    # 提交到数据库执行
    db.commit()
except:
    # 发生错误时回滚
    db.rollback()

第四步：关闭游标，数据库连接

cursor.close()
db.close()

示例

import pymysql

db = pymysql.connect(host="127.0.0.1",
                     user="root",
                     password="root",
                     port=3306,# 端口
                     database="demo",
                     charset='utf8')

cursor = db.cursor()

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

cursor.close()
db.close()

python .\insertData.py