In a modify task you take code that already works and adapt it so that it does something else.

The editor starts you off with the program from the investigate task. It adds students to the end of StudentInfo.txt until the user answers n:

repeat = True

while repeat == True:
    print("Enter student name")
    student_name = input()
    student_name = student_name + "\n"

    data_file = open("StudentInfo.txt", "a")
    data_file.write(student_name)
    data_file.close()

    print("Enter another student? y/n")
    choice = input()

    if choice == "n":
        repeat = False
        print("Goodbye!")

Your task

Change the program so that it can store teachers as well as students, each in a file of their own.

• There is a second file for teacher data, called TeacherInfo.txt.
• At the start of every repetition, the program asks whether the next person is a teacher or a student. The user answers teacher or student, and the program uses the answer to open the correct file.
• For a teacher, it asks for the teacher’s name and their teacher code (for example ACO), and appends them to TeacherInfo.txt.
• For a student, it asks for the student’s name and their grade, and appends them to StudentInfo.txt.
• Every value goes on a line of its own, so each person adds two lines to their file: the name first, then the code or the grade.
• Both files are opened in append mode. They already hold people entered earlier, and those must still be there when the program has finished.

The user will only ever answer teacher or student to the first question, so you do not need to handle any other answer.

The output

Each repetition starts with this question:

Enter teacher or student

For a teacher, the next two questions are:

Enter teacher name
Enter teacher code

For a student, they are:

Enter student name
Enter student grade

Now that the program handles teachers too, the question at the end of every repetition asks about a person rather than a student:

Enter another person? y/n

As before, only n ends the loop, and the program then prints Goodbye!.

The files

StudentInfo.txt and TeacherInfo.txt might start like this:

Leo
6

Mr Price
JPR

Example

Input:

student
Ben
5
y
teacher
Mr Hughes
HUG
n

Output:

Enter teacher or student
Enter student name
Enter student grade
Enter another person? y/n
Enter teacher or student
Enter teacher name
Enter teacher code
Enter another person? y/n
Goodbye!

StudentInfo.txt afterwards:

Leo
6
Ben
5

TeacherInfo.txt afterwards:

Mr Price
JPR
Mr Hughes
HUG