GroupProject/student.py at master · maxclingroth/GroupProject · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
# ----------------------------------------------------------------
# Author: Brady Camp
# Date: November 28, 2021
#
# This module supports changes in the registered courses
# for students in the class registration system.  It allows
# students to add courses, drop courses and list courses they are
# registered for.
# -----------------------------------------------------------------

def list_courses(id, c_roster):
    # ------------------------------------------------------------
    # This function displays and counts courses a student has
    # registered for.  It has two parameters: id is the ID of the
    # student; c_roster is the list of class rosters. This function
    # has no return value.
    # -------------------------------------------------------------
    # Creates a list and iterates over the dictionary, storing classes taken in the list
    list = []
    for i in c_roster:
        if id in c_roster[i]:
            list.append(i)
    length = len(list)
    i = 0
    print("You are currently enrolled in the following courses:")
    while i < length:
        print(list[i])
        i += 1
    print('Total number:', len(list))


def add_course(id, c_roster, c_max_size):
    # ------------------------------------------------------------
    # This function adds a student to a course.  It has three
    # parameters: id is the ID of the student to be added; c_roster is the
    # list of class rosters; c_max_size is the list of maximum class sizes.
    # This function asks user to enter the course he/she wants to add.
    # If the course is not offered, display error message and stop.
    # If student has already registered for this course, display
    # error message and stop.
    # If the course is full, display error message and stop.
    # If everything is okay, add student ID to the course’s
    # roster and display a message if there is no problem.  This
    # function has no return value.
    # -------------------------------------------------------------
    course = input("What course would you like to add? ")

    # Tests if the course entered is a key in the c_roster dictionary
    if course not in c_roster:
        print("The course you entered does not exist.")

    # Tests if the id is in the list associated with the course key
    elif id in c_roster[course]:
        print("You are already registered in this course.")

    # Tests if the length of the list for the course is equal to the max size of the course
    elif len(c_roster[course]) == c_max_size[course]:
        print("The course you entered has reached its maximum capacity.")

    else:
        c_roster[course].append(id)
        print("You have been added to that course.")


def drop_course(id, c_roster):
    # ------------------------------------------------------------
    # This function drops a student from a course.  It has two
    # parameters: id is the ID of the student to be dropped;
    # c_roster is the list of class rosters. This function asks
    # the user to enter the course he/she wants to drop.  If the course
    # is not offered, display error message and stop.  If the student
    # is not enrolled in that course, display error message and stop.
    # Remove student ID from the course’s roster and display a message
    # if there is no problem.  This function has no return value.
    # -------------------------------------------------------------
    course = input("What course would you like to drop? ")

    # Tests if the course entered is a key in the c_roster dictionary
    if course not in c_roster:
        print("The course you entered does not exist.")

    # Tests if the id is not in the list associated with the course key
    elif id not in c_roster[course]:
        print("You were not enrolled in this course.")

    else:
        c_roster[course].remove(id)
        print("You have been dropped from that course.")