exercise csv 2/4

steven.chen3 · steven.chen3 · commit bd76b604d18b · 2022-10-26T02:03:31.000-07:00
diff --git a/student_manager/command/command_processor.py b/student_manager/command/command_processor.py
@@ -11,15 +11,34 @@ def process_list(student_data: dict):
 
 
 def process_add(student_data: dict):
-    pass
+    name = input("Please input name: ")
+    chinese_score = input("Please input chinese score: ")
+    english_score = input("Please input english score: ")
+    math_score = input("Please input math score: ")
+
+    record = {
+        "name": name,
+        "chinese": int(chinese_score),
+        "english": int(english_score),
+        "math": int(math_score)
+    }
+
+    student_data[name] = record
+    print(f"Added student scores for '{name}' successfully")
 
 
 def process_edit(student_data: dict):
     pass
 
 
 def process_delete(student_data: dict):
-    pass
+    name = input("Please input name: ")
+
+    if name in student_data.keys():
+        student_data.pop(name)
+        print(f"Delete '{name}' successfully")
+    else:
+        print(f"Student '{name}' does not exist")
 
 
 def process_average(student_data: dict):
diff --git a/student_manager/store/csv_store.py b/student_manager/store/csv_store.py
@@ -1,9 +1,10 @@
 import csv
 import os.path
 
+file = "students.csv"
+
 
 def load_data() -> dict:
-    file = "students.csv"
 
     result = {}
 
@@ -21,4 +22,12 @@ def load_data() -> dict:
 
 
 def save_data(data: dict):
-    pass
+    file_instance = open(file, 'w', encoding="UTF8")
+
+    if len(data) > 0:
+        rows = list(data.values())
+        csv_writer = csv.DictWriter(file_instance, rows[0].keys())
+        csv_writer.writeheader()
+        csv_writer.writerows(rows)
+
+    file_instance.close()
