Python简单学生信息管理系统（数据库存储）

sql.py

import  pymysql
#获取数据库的连接方法
def get_connection():
    connection=pymysql.connect(host="localhost",user="root",password="1234567",database="test",port=3306)
    return connection

#初始化的时候从数据库中查询学生信息
def query():
    conn=get_connection()
    cur=conn.cursor(cursor=pymysql.cursors.DictCursor)
    sql="select * from stu"
    try:
        cur.execute(sql)
        students=cur.fetchall()
        if type(students)==list:
            return  students
        else:
            return []
    except Exception as ex:
        raise ex
    finally:
        conn.close()

#初始化函数，从数据库中查询到的赋值给name_list
def name_init():
    try:
        global name_list
        name_list=query()
        print("成功获取数据库中的数据")
    except Exception as ex:
        raise ex

#添加学生 ——直接添加到数据库里面
def  stu_insert(stu):
    conn=get_connection()
    cur=conn.cursor(cursor=pymysql.cursors.DictCursor)
    sql="insert into stu(id,name,score) values('%s','%s','%s')"
    id=stu['id']
    name=stu['name']
    score=stu['score']
    try:
        cur.execute(sql % (id,name,score))
        conn.commit()
    except Exception as ex:
        conn.rollback()
        raise ex
    finally:
        conn.close()

#删除学生
def stu_delete(sid):
    conn=get_connection()
    cur=conn.cursor(pymysql.cursors.DictCursor)
    sql="delete from stu where id='%s'"
    try:
        cur.execute(sql%sid)
        conn.commit()
    except Exception as ex:
        conn.rollback()
        raise  ex
    finally:
        conn.close()


#修改学生信息
def stu_update(stu):
    conn=get_connection()
    cur=conn.cursor(cursor=pymysql.cursors.DictCursor)
    sid=stu['id']
    name=stu['name']
    score=stu['score']
    sql="update stu set name ='%s',score='%s' where id='%s'"
    try:
        cur.execute(sql%(name,score,sid))
        conn.commit()
    except Exception as ex:
        conn.rollback()
        raise ex
    finally:
        conn.close()

tools.py

import sql
#记录学生的信息
list=sql.query()

#显示菜单
def menu():
    print("*" * 20)
    print("1、添加学生")
    print("2、查询学生")
    print("3、显示全部")
    print("0、退出系统")
    print("*" * 20)

#检查学号是否重复或者有误
def check_id(new_id):
    flag=True
    while flag:
        if new_id.isdigit():
            for i in  range(len(list)):
                if list[i]['id']==new_id:
                    new_id=check_id(input("学号重复，请重新输入:"))
            flag=False
        else:
            new_id=input("输入有误，请重新输入:")
    return new_id



#添加学生信息
def add():
    try:
        dict={}
        new_id=check_id(input("请输入学号:"))
        dict['id']=new_id
        new_name=input("请输入姓名:")
        dict['name']=new_name
        new_score=input("请输入分数:")
        dict['score']=new_score
        list.append(dict)
        sql.stu_insert(dict)
        print("【%s】信息添加成功" %new_name)
    except Exception as ex:
        raise ex


#查询学生信息
def select():
    list=sql.query()
    select_id=input("请输入你要查询的学生学号:")
    for dict in  list:
        if select_id==dict["id"]:
            print("*" * 30)
            print("学号:%s\t姓名:%s\t分数:%s" % (dict["id"], dict["name"], dict["score"]))
            print("*"*30)
            # 对学生信息进行删除和修改操作
            deal(dict)
            break

    else:
        print("*" * 20)
        print("没有找到学号为%s的学生"%select_id)
        print("*" * 20)

#显示学生信息
def show():
    list = sql.query()
    #判断学生信息是否为空
    if (len(list)==0):
        print("没有添加学生信息")
        # return 下方的代码不会被执行
        return
    # 遍历学生信息列表，依次输出列表信息
    for i in ["学号","姓名","分数"]:
    #如果不换行，可以在后面加一个 end=" "
        print(i,end="\t")
    print("")
    print("-"*20)
    for dict in list:
        print("%s\t\t%s\t\t%s" % (dict["id"], dict["name"], dict["score"]))

#处理查找到的学生信息
def deal(find_dict):
    """
    :param find_dict: 查找到的学生信息
    """
    print("[1]修改---[2]删除---[0]返回")

    action=input("请输入:")


    if  action=='0':
        pass
    #修改学生信息
    elif action=='1':
        while True:
            choice=input("请输入要修改的学生的(1.姓名 2.分数 3.全部修改):")
            if choice=='3':
                # new_id=input("请输入新的学号:")
                # find_dict['id']=new_id
                new_name=input("请输入新的姓名:")
                find_dict['name']=new_name
                new_score=input("请输入新的分数:")
                find_dict['score']=new_score
                sql.stu_update(find_dict)
                break
            elif choice=='1':
                new_name=input("请输入新的姓名:")
                find_dict['name']=new_name
                sql.stu_update(find_dict)
                break
            elif choice=='2':
                new_score=input("请输入新的分数:")
                find_dict['score']=new_score
                sql.stu_update(find_dict)
                break
            else:
                print("输入错误")
    #删除学生信息
    elif action=='2':
        sql.stu_delete(find_dict['id'])
        print("删除成功")

    else:
        print("没有匹配的操作")

main.py

import tools
tools.menu()

while True:

    action = input("请输入操作码:")

    if action=='0':
        print("欢迎再次访问")
        break
    elif action=='1':
        tools.add()
    elif action=='2':
        tools.select()
    elif action=='3':
        tools.show()
    else:
        print("请输入正确的号码")