文章目录
一程序功能(Function)
1. 功能概述
- 利用Python完成一个关于MySQL数据库的学生信息的增删改查程序,主要涉及知识点:游标的运用、SQL语句以及之前学习的基础知识。
2. 思维导图
想法已有开始实践
二.开始实践
战前准备
你需要先导入sqlite3 模块与pymysql数据库如下图:
import pymysql
# 定义数据库链接参数
host = '127.0.0.1'
port = 3306
db = 'student'
user = 'root'
password = 'password'
# 获取数据库链接
conn = pymysql.connect(host=host,
user=user,
password=password,
port=port,
db=db, )
如果没有安装sqlite3
可通过Windows+R–>输入cmd–>执行pip install sqlite3命令,如下图:
因为博主已经下载过,再次就不重复安装。
定义学生管理系统菜单
在我们开始做之前我们应该在大脑里构思,这个数据库应该具有怎样的操作,思维很重要!
def menu():
print('============学生信息管理系统=====================')
print('============功能菜单=====================')
print('\t\t1.录入学生信息')
print('\t\t2.查找学生信息')
print('\t\t3.删除学生信息')
print('\t\t4.修改学生信息')
print('\t\t5.排序')
print('\t\t6.统计学生人数')
print('\t\t7.显示学生所有信息')
print('\t\t0.退出')
print('==============================================')
定义主函数
接下来 我们 定义主要函数:增加、查找、删除、修改、排序、统计、显示所有。
# 定义主函数
def main():
while True:
menu()
num = int(input('请选择你要做的操作:'))
if num in (0, 1, 2, 3, 4, 5, 6, 7):
if num == 0:
answer = input('你确定退出系统吗?y/n\t') # 判断为0
if answer == 'y' or answer == 'Y':
print('谢谢您的使用!!!')
break
else:
continue
elif num == 1:
insert()
elif num == 2:
search()
elif num == 3:
delete()
elif num == 4:
modify()
elif num == 5:
sort()
elif num == 6:
total()
elif num == 7:
show()
1.录入学生信息
本小节重点
- 数据库链接的重中之重是必须创建游标对象,不能忘记这个就相当于Access里记录指针,切记别忘记写了
- 注意SQL语句的写法(如果有对自己写的有疑问,可以复制到MySQL中检验代码是否错误)如下图:
注此处图是借用后面的所以别太在意(哈哈)
def insert():
# 创建游标对象
cursor = conn.cursor(pymysql.cursors.DictCursor)
# cursor = conn.cursor(pymysql.cursors.SSCursor)
while True:
name = input("请输入学生姓名:")
if not name:
print('请输入有效的名字!')
continue
try:
# 执行SQL查询
gender = input("请输入学生姓别:")
age = input("请输入学生年龄:")
sql = "insert into student (name,gender,age) values ('" + name + "','" + gender + "'," + age + ")"
count = cursor.execute(sql)
if count > 0:
# 提交数据库修改
conn.commit()
# 提示用户成功了
print('记录插入成功~')
answer = input('是否继续添加?y/n\n')
if answer != 'y':
break
else:
continue
except pymysql.err.IntegrityError:
print('主键不允许重复')
2.查找学生信息
本小节重点
- 创建游标对象
- 查询的方式:是按ID,还是按着姓名
def search():
# 创建游标对象
cursor = conn.cursor(pymysql.cursors.SSCursor)
while True:
mode = input('按ID查找请输入1,按姓名查找输入2:')
if mode == '1':
id = input('请输入学生ID')
sql = f"select * from student where id={id}"
cursor.execute(sql)
print(cursor.fetchall())
answer = input('是否继续添加?y/n\n')
if answer != 'y':
break
else:
continue
elif mode == '2':
name = input('请输入学生姓名:')
sql = f"select * from student where name='{name}'"
cursor.execute(sql)
print(cursor.fetchall())
answer = input('是否继续添加?y/n\n')
if answer != 'y':
break
else:
continue
else:
print('您的输入有误,请重新输入')
continue # 返回while True
3.删除学生信息
基本内容差不多,可以抽空自己练习
def delete():
# 创建游标对象
cursor = conn.cursor(pymysql.cursors.SSCursor)
while True:
student_id = input('请输入要删除学生的ID:')
if student_id != '':
sql = f"select * from student where id={student_id}"
cursor.execute(sql)
answer = input('是否继续删除?y/n\n')
if answer != 'y':
show()
break
else:
continue
else:
print('输入有误!请重新输入')
continue
4.修改学生信息
本小节重点
- 明确修改的是什么,
- formart的另一种简便书写格式
def modify():
show()
# 创建游标对象
cursor = conn.cursor(pymysql.cursors.SSCursor)
while True:
student_id = input('请输入你要修改的学生学号(如 1,2):')
if student_id != '':
name = input('请输入准备修改的姓名:')
gender = input('请输入准备修改的性别:')
age = input('请输入准备修改的年龄:')
sql = f"update student set name='{name}',gender='{gender}',age={age} where id={student_id}"
d=cursor.execute(sql)
if d:
print("修改成功!")
else:
print("修改失败!")
answer = input('是否继续修改?y/n\n')
if answer != 'y':
show()
break
else:
continue
else:
print('输入有误!请重新输入')
continue
5.排序
本小节重点
- 本节代码易写错注意检查SQL语句
- 遍历的方式(复习)当然还有其他方法,改日在写
def sort():
show()
# 创建游标对象
cursor = conn.cursor(pymysql.cursors.SSCursor)
asc_or_desc = input('请选择(0.升序,1.降序):')
sor = ""
if asc_or_desc == '0':
sor = "asc"
elif asc_or_desc == '1':
sor = "desc"
else:
print('您的输入有误,请重新输入')
sort()
# 有空在补充
# mode = input('请选择排序方式(1.按英语成绩排序,2.按Python成绩排序3.按Java排序 0.按着主程序排序)')
mode = input('请选择排序方式(1.按学号排序,2.按年龄排序)')
if mode == '1':
sql = f"select * from student order by id " + sor
cursor.execute(sql)
all_records = cursor.fetchall()
for record in all_records:
print(record[0], record[1], record[2], record[3])
elif mode == '2':
sql = f"select * from student order by age " + sor
cursor.execute(sql)
all_records = cursor.fetchall()
for record in all_records:
print(record[0], record[1], record[2], record[3])
6.统计总人数
本节到是没什么,挺简单的
def total():
# 创建游标对象
cursor = conn.cursor(pymysql.cursors.SSCursor)
sql = f"select gender,count(gender) as '人数' from student group by gender"
cursor.execute(sql)
all_records = cursor.fetchall()
for record in all_records:
print(record[0], record[1])
7.显示学生所有信息
本小节重点
- 注意占位符的位置,抽空练的时候可以考虑换一种方法来写
def show():
# 创建游标对象
cursor = conn.cursor(pymysql.cursors.SSCursor)
sql = f"select * from student"
cursor.execute(sql)
all_records = cursor.fetchall()
format_title = '{:^6}\t{:^12}\t{:^8}\t{:^10}'
print(format_title.format('ID', '姓名', '性别', '年龄'))
format_data = '{:^6}\t{:^12}\t{:^8}\t{:^10}'
for record in all_records:
print(format_data.format(record[0], record[1], record[2], record[3]))
8.调用函数
if __name__ == '__main__':
main()