school_management/main.py

from pprint import pprint

from student import Student, SaveStudent, UpdateStudent, DeleteStudent
from teacher import Teacher, SaveTeacher
from helper import FileData, ClassName, Subject
import settings


def add_student():
    fname = input("First Name: ")
    lname = input("Last Name: ")
    contact = input("Phone Number(Must contains 10 digits): ")
    rollnum = input("Roll Number: ")
    class_code = input(
        f"Class: Select from {[cl.value for cl in ClassName.get_classnames_list()]}: ")
    cl = ClassName(class_code)
    student = Student(fname, lname, contact, rollnum, cl)
    print("=============================================================")
    print("Following details added to the students file(student.json).")
    student.print_instance_detail()
    SaveStudent(student)


def update_student(update_id):
    stu_tbu = UpdateStudent(update_id)
    stu = stu_tbu.get_student()
    fname = input(f"First Name(current: {stu['fname']}): ")
    lname = input(f"Last Name(current: {stu['lname']}): ")
    contact = input(f"Phone Number(current: {stu['contact']}): ")
    rollnum = input(f"Roll Number(current: {stu['rollnum']}): ")
    class_code = input(f"{ClassName.get_classnames_list()}: ")
    cl = ClassName(class_code).name
    print("cl is: ", cl)

    stu['id'] = update_id
    stu['fname'] = fname
    stu['lname'] = lname
    stu['email'] = f"{fname}_{lname}@schoolname.com"
    stu['contact'] = contact
    stu['rollnum'] = rollnum
    stu['class'] = cl

    stu_tbu.update(stu)


def delete_student(delete_id):
    DeleteStudent(delete_id)


def display_students():
    _, data = FileData(settings.STUDENT_FILENAME).load()
    print("===========================================")
    pprint(data)


def add_teacher():
    fname = input("First Name: ")
    lname = input("Last Name: ")
    contact = input("Phone Number: ")
    empid = input("Employee ID: ")
    subject_code = int(input(f"{Subject.get_subjects_list()}: "))
    subject = Subject(subject_code)
    teacher = Teacher(fname, lname, contact, empid, subject)
    print("=============================================================")
    print("Following details added to the teachers file.")
    teacher.print_instance_detail()
    SaveTeacher(teacher)


print("1: Manage Student")
print("2: Manage Teacher")

manage = input("Select manage: ")
if manage == "1":
    print("===========================================")
    print("1: Add New Student")
    print("2: Edit Student")
    print("3: Delete Student")
    print("4: Display All Student")

    user_action = input("Select student action: ")
    if user_action == '1':
        add_student()

    elif user_action == "2":
        update_id = int(input("Enter the Id of student to be update: "))
        update_student(update_id)

    elif user_action == "3":
        delete_id = int(input("Enter the Id of student to be deleted: "))
        delete_student(delete_id)

    elif user_action == "4":
        display_students()

elif manage == "2":
    add_teacher()