Python连接数据库（学生管理系统案例，可以实现增删改查）

一、准备数据库

首先，我们需要准备好数据库

这是我的数据库表设计的截图，大家可以借鉴一下，根据个人需求自行设计数据库

上面这些就是表内的数据了

create table if not exists student
(
    id       int auto_increment
        primary key,
    name     varchar(200) null,
    class    varchar(200) null,
    birthday date         null
);

这是建表语句，大家有需要可以直接复制到控制台创建

二、书写Python代码

接下来我们就开始书写Python代码了，

如果大家是第一次用Python连接数据库，那么大家需要用pip下载一个名为pymysql的包

pip install pymysql

先给大家展示一下我的Python的文件结构，其实这个借鉴了点Java的DAO模式

app.py：作为启动项，里面只有一行作为启动用的代码

dao.py：作为和数据库交互的文件，有关增删改查的具体步骤放在这里面

view.py：用来展示操作者能看到的视图界面

2.1 书写dao.py

我是先从dao.py写起的，以下为dao.py的代码

第一步是准备工作，我们需要导包，定义数据库的地址、用户名、密码以及端口号

import pymysql
import prettytable as pt

dbhost = 'localhost'
dbname = 'root'
dbpassword = '1234'
dbport = 3306

第二步，书写增删改查的方法

1. 查询所有的学生信息select_student()

def select_all():
    # tb是一个输出表格的工具，它可以使输出的内容以表格形式输出，整洁美观
    tb = pt.PrettyTable()
    # 设定输出表格的列名，这里建议和数据库中的列名对齐就可以
    tb.field_names = ["id", "name", "class", "birthday"]
    # 定义SQL语句
    sql = 'select * from studentManager.student'
    # 使用pymysql连接数据库
    db = pymysql.connect(host=dbhost, user=dbname, passwd=dbpassword, port=dbport)
    # 创建一个光标，用于执行SQL语句
    cursor = db.cursor()
    cursor.execute(sql)
    # 查询所有结果
    results = cursor.fetchall()
    # 这里简单遍历一下，将数据库中的每一条内容整齐的添加到上面提到的tb表格中
    for result in results:
        tb.add_row(result)
    # 记得关闭数据库连接和光标连接
    db.close()
    cursor.close()
    # 将查询的结果返回
    return tb

2. 添加学生信息add_student()

def add_student(id, name, cls, birthday):
    # 定义SQL语句
    sql = "insert into studentManager.student(id, name, class, birthday) values (%s, %s, %s, %s)"
    # 设置参数
    param = (id, name, cls, birthday)
    # 使用pymysql连接数据库
    db = pymysql.connect(host=dbhost, user=dbname, passwd=dbpassword, port=dbport)
    cursor = db.cursor()
    cursor.execute(sql, param)
    # 对数据库提交更改
    db.commit()
    db.close()

3. 修改学生信息modify_student()

def modify_student(id, name, cls, birthday):
    # 定义SQL语句
    sql = "update studentManager.student set name=%s, class=%s, birthday=%s where id=%s;"
    # 设置参数
    param = (name, cls, birthday, id)
    # 使用pymysql连接数据库
    db = pymysql.connect(host=dbhost, user=dbname, passwd=dbpassword, port=dbport)
    cursor = db.cursor()
    cursor.execute(sql, param)
    db.commit()
    db.close()

4. 删除学生信息delete_student()

def delete_student(id):
    # 定义SQL语句
    sql = "delete from studentManager.student where id = %s;"
    # 设置参数
    param = (id)
    # 使用pymysql连接数据库
    db = pymysql.connect(host=dbhost, user=dbname, passwd=dbpassword, port=dbport)
    cursor = db.cursor()
    cursor.execute(sql, param)
    db.commit()
    db.close()

这样，dao.py的方法就写好了

2.2 书写view.py

然后书写view.py的代码，主要是用来展示

# 导入刚刚编写完成的dao.py
import dao

def view():
    flag = True
    while flag:
        print("欢迎使用学生管理系统")
        print("1.查询所有学生 2.添加学生 3.编辑学生 4.删除学生 5.退出系统")
        command = int(input("请输入指令"))
        if command == 1:
            print(dao.select_all())
        elif command == 2:
            id = (input("请输入所添加学生的学号"))
            name = input("请输入所添加学生的姓名")
            cls = input("请输入所添加学生的班级")
            birthday = input("请输入所添加学生的出生日期")
            dao.add_student(id, name, cls, birthday)
            print("添加完成")
        elif command == 3:
            id = input("请输入需要修改的学生ID")
            name = input("请输入修改后的学生姓名")
            cls = input("请输入修改后的学生班级")
            birthday = input("请输入修改后的学生出生日期")
            dao.modify_student(id, name, cls, birthday)
            print("编辑完成")
        elif command == 4:
            id = input("请输入需要删除的学生ID")
            dao.delete_student(id)
            print("删除完成")
        elif command == 5:
            print("谢谢使用")
            break

这里面就是简单的while循环语句，大家也根据自己的逻辑书写

2.3 书写app.py

最后就是最简单的一个，app.py，这个作为程序的入口，想要运行程序时直接运行app.py即可

# 导入刚刚书写完的视图
import view

if __name__ == '__main__':
    view.view()

运行效果图

最后就是大家如果在书写过程中遇到一些bug或难题，又或者以后有一些需求，可以关注下面的公众号并留言