Python基础 - 学生管理系统

main_page.py

import model.file_manager as file_manager
import student_manager


def register():
    """注册"""
    while True:
        # 1.输入账号
        user_name = input('请输入账号（3~6位）：')

        # 2.检测账号密码的位数是否符合要求
        if not 3 <= len(user_name) <= 6:
            print('账号有误！请重新输入！')
            continue
        else:
            break

    while True:
        # 3.输入密码
        pass_word = input('请输入密码（6~12位）：')
        # 4.检查密码是否正确
        if not 6 <= len(pass_word) <= 12:
            print('密码有误！请重新输入！')
            continue
        else:
            break

    # 检测账号是否存在
    """
    数据设计：用一个字典保存所有的账号和密码，账号：密码 ————>
             将保存所有的账号和密码对应的字典存到一个文件中  --> 选json
    """
    # 1.获取注册过的用户名
    all_user = file_manager.json_read('all_user.json')

    # 2.判断输入的账号是否存在
    if user_name in all_user:
        print('该用户名已存在！')
        return

    # 3.账号可用,添加到所有的用户中
    all_user[user_name] = pass_word

    # 4. 写入文件
    re = file_manager.json_write('all_user.json', all_user)
    if re:
        print('注册成功！')
    else:
        print('注册失败！')


# 登录
def login():
    user_name = input('请输入账号：')
    pass_word = input('请输入密码：')

    all_user = file_manager.json_read('all_user.json')
    if user_name not in all_user:
        print('登录失败！账号不存在!')
        return
    pass_word_old = all_user[user_name]
    if pass_word == pass_word_old:
        print('登录成功！')
        student_manager.show_manager_page(user_name)
        return
    else:
        print('登录失败！密码错误！')


def show_main_page():
    """
    显示主页
    """
    while True:
        # 显示界面
        print(file_manager.text_read('main_page.txt'))
        # 选择
        value = input('请选择（1-3）：')
        if value == '1':
            login()
        elif value == '2':
            register()
        elif value == '3':
            return
        else:
            print('输入有误！')


if __name__ == '__main__':
    # 1.显示登录注册页面
    show_main_page()

file_manager.py

import json


# 普通文本文件的读操作
def text_read(file_name: str):
    """
    普通文本文件的读操作
    :param file_name:要读的文件的名字
    :return: 读到的内容
    """
    try:
        with open('./files/' + file_name, encoding='utf-8') as f:
            return f.read()
    except FileNotFoundError:
        print('%s文件不存在' % file_name)
        return None


def text_write(file_name: str, content: str):
    """
    将文本写入
    :param file_name: 文件名
    :param content: 写入到文件中的内容
    :return: 是否写入成功
    """
    with open('./files/' + file_name, 'w', encoding='utf-8') as f:
        f.write(content)
        return True


# 读json文件中的内容
def json_read(file_name: str):
    """
    获取json文件中的内容
    :param file_name: 文件名
    :return: python类型的数据
    """
    try:
        with open('./files/' + file_name, encoding='utf-8') as f:
            return json.load(f)
    except FileNotFoundError:
        print('%s文件不存在' % file_name)
        return None


def json_write(file_name: str, content: str):
    """
    将内容写入json文件中
    :param file_name:
    :param content:
    :return: 写入是否成功
    """
    try:
        with open('./files/' + file_name, 'w', encoding='utf-8') as f:
            json.dump(content, f)
            return True
    except:
        # print('写入失败')
        return False

student_manager.py

import model.file_manager as file_manager

# 全局变量用来保存当前登录成功的用户名
current_user = ''

"""
数据结构：
1.账号和账号管理的学生一一对应

2.一个账号要管理多个学生

3.一个学生要存储多个信息。每个学生存储的信息数量一样

方案一：整个管理系系统所有的账号管理的学生放在一起
{
    账号1：[{'name': name, 'age': age, 'tel': tel, 'id': id]}]
    账号2：[{'name': name, 'age': age, 'tel': tel, 'id': id]}]
    账号3：[]
}
将字典写到json中

方案二：一个账号对应一个json文件
一个账号对应一个json文件，每个json文件中就是一个列表
json文件名和账号一一对应
aaa.json:
{
    'count': nums,
    'all_student': [{'name': name, 'age': age, 'tel': tel, 'id': id]}]
}
"""
# 通过全局变量保存常用的key
key_count = 'count'
key_all_students = 'all_students'
key_name = 'name'
key_age = 'age'
key_tel = 'tel'
key_id = 'id'


def get_current_userinfo():
    # 获取当前账号对应的json文件中的内容
    content = file_manager.json_read(current_user + '.json')
    if not content:
        content = {}
    return content


# 添加学生
def add_student():
    user_info = get_current_userinfo()

    while True:
        # 1.学生信息
        name = input('请输入姓名：')
        age = input('请输入年龄：')
        tel = input('请输入电话：')
        # 2.产生学号
        number = user_info.get(key_count, 0)
        number += 1
        stu_id = 'stu' + str(number).rjust(3, '0')
        # 3.创建学生
        stu = {key_name: name, key_age: age, key_tel: tel, key_id: stu_id}
        # 4.添加学生
        all_students = user_info.get(key_all_students, [])
        all_students.append(stu)
        # 5.将新的数据更新到文件中
        user_info[key_count] = number
        user_info[key_all_students] = all_students
        re = file_manager.json_write(current_user + '.json', user_info)
        if re:
            print('添加成功！')
        else:
            print('添加失败！')

        # 6.选择
        print('1.继续')
        print('2.返回')
        value = input('请选择（1-2）：')
        if value == '1':
            continue
        else:
            return


def show_student_info(stu):
    print('学号：%s  姓名：%s  年龄：%s  电话：%s'\
          % (stu[key_id], stu[key_name], stu[key_age], stu[key_tel]))


def find_student_with_name(name, all_students):
    # 按姓名查找学生
    find_students = []   # 保存符合条件的所有学生
    for stu in all_students:
        if stu[key_name] == name:
            find_students.append(stu)
    return find_students


def find_student_with_id(stu_id, all_students):
    for stu in all_students:
        if stu[key_id] == stu_id:
            return stu
    return None


def find_student():
    # 查找学生
    # 1.拿到所有学生
    user_info = get_current_userinfo()
    all_students = user_info.get(key_all_students, None)
    if not all_students:
        print('当前账号没有可管理的学生！')
        return

    print('1.查看所有学生\n2.按姓名查找\n3.按学号查找\n4.返回')
    value = input('请选择（1-4）：')
    if value == '1':
        for stu in all_students:
            show_student_info(stu)
    elif value == '2':
        print('按姓名找')
        name = input('请输入姓名：')
        students = find_student_with_name(name, all_students)
        if students:
            for stu in students:
                show_student_info(stu)
        else:
            print('没有找到%s' % name)

    elif value == '3':
        print('按学号找')
        stu_id = input('请输入学号：')
        stu = find_student_with_id(stu_id, all_students)
        if stu:
            show_student_info(stu)
        else:
            print('没有找到学号 %s 的学生' % stu_id)
    else:
        return


def delete_student():
    """删除学生"""
    # 1.拿到所有学生
    user_info = get_current_userinfo()
    all_students = user_info.get(key_all_students, [])
    if not all_students:
        print('当前账号没有可管理的学生！')
        return
    while True:
        # 2.给出选择
        print('1.按姓名删除\n2.按学生删除\n3.返回')
        value = input('请选择（1-3）：')

        if value == '1':
            name = input('请输入姓名：')
            # 先找到同名的所有学生
            students = find_student_with_name(name, all_students)

            # 判断是否有该学生
            if not students:

                print('该学生不存在！')
                continue

            # 如果找到了学生，先展示找到的所以学生
            index = 0
            for stu in students:
                print(index, end=' ')
                show_student_info(stu)
                index += 1
            print('q 返回上一层')
            # 问想要具体删除哪个学生
            del_index = input('请输入要删除学生的序号：')

            if del_index == 'q':
                continue
            else:
                # 找到要删除的学生对于的字典
                del_stu = students[int(del_index)]
                # 从所有的学生对应的字典中去删除
                all_students.remove(del_stu)

                # 将新的数据更新到json文件中
                re = file_manager.json_write(current_user + '.json', user_info)
                if re:
                    print('删除成功！')
                else:
                    print('删除失败！')

        elif value == '2':
            del_id = input('输入要删除掉的学号：')
            # 根据学号查找学生
            del_stu = find_student_with_id(del_id, all_students)
            if not del_stu:
                print('没有该学生！')
                continue
            show_student_info(del_stu)
            value = input('是否删除（Y/N）：')
            if value == 'Y':
                all_students.remove(del_stu)
            else:
                continue
            # 删除后更新数据
            re = file_manager.json_write(current_user + '.json', user_info)
            if re:
                print('删除成功！')
            else:
                print('删除失败！')
        else:
            return


def alter_student_key_value(stu):
    print('1.修改姓名\n2.修改年龄\n3.修改电话')
    value = input('请选择（1-3）：')
    if value == '1':
        new_name = input('请输入新的姓名：')
        stu[key_name] = new_name
        return stu
    elif value == '2':
        new_age = input('请输入新的年龄：')
        stu[key_age] = new_age
        return stu
    elif value == '3':
        new_tel = input('请输入新的电话：')
        stu[key_tel] = new_tel
        return stu


def alter_student():
    """修改学生信息"""
    user_info = get_current_userinfo()
    all_students = user_info.get(key_all_students, None)
    if not all_students:
        print('当前账号没有可管理的学生！')
        return
    while True:
        print('1.按姓名修改\n2.按学号修改\n3.返回\n')
        value = input('请选择（1-3）：')
        if value == '1':
            name = input('请输入要修改的学生姓名：')

            students = find_student_with_name(name, all_students)

            if not students:
                print('没有姓名为 %s 的学生' % name)
                continue

            index = 0
            for stu in students:
                print(index, end=' ')
                show_student_info(stu)
                index += 1
            print('q 返回上一层')
            alter_index = input('请输入要删除的学生序号')

            if alter_index == 'q':
                continue
            else:
                alter_stu = students[int(alter_index)]
                alter_student_key_value(alter_stu)

                re = file_manager.json_write(current_user + '.json', user_info)
                if re:
                    print('修改成功！')
                else:
                    print('修改失败！')

        elif value == '2':
            alter_id = input('请输入要修改的学生学号：')
            alter_stu = find_student_with_id(alter_id, all_students)
            if not alter_stu:
                print('没有学号 %s 的学生' % alter_id)
                continue
            show_student_info(alter_stu)
            value = input('是否要修改该学生（Y/N）:')
            if value == 'Y':
                alter_student_key_value(alter_stu)
            else:
                continue
            re = file_manager.json_write(current_user + '.json', user_info)
            if re:
                print('修改成功！')
            else:
                print('修改失败！')

        else:
            break


def show_manager_page(user_name):
    """
    显示学生管理页面首页
    :return:
    """
    global current_user
    current_user = user_name

    # 2.显示页面
    while True:
        content = file_manager.text_read('manager_page.txt')
        print(content % user_name)

        # 3. 选择
        value = input('请选择（1-5）：')
        if value == '5':
            return
        elif value == '1':
            add_student()
        elif value == '2':
            find_student()
        elif value == '3':
            alter_student()
        elif value == '4':
            delete_student()
        else:
            print('输入有误！')