一、准备数据库
首先,我们需要准备好数据库
这是我的数据库表设计的截图,大家可以借鉴一下,根据个人需求自行设计数据库
上面这些就是表内的数据了
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或难题,又或者以后有一些需求,可以关注下面的公众号并留言