edit user IO

2019-07-30 07:40:22 -07:00
parent 97c70c6457
commit f5259dfe8e
1 changed files with 11 additions and 2 deletions
@@ -12,5 +12,14 @@ def userinput():
            subject = input("Which subject do you want to add? \n")
            return "input", subject
        elif todo == "category":
-            subject = input("Where do you want to add the category? \n")
-            category = input("Which category do you want ot add? \n")
+            subject = input("To which subject do you want to add the category? \n")
+            category = input("Which category do you want to add? \n")
+            rating = float(input("Rating? \n"))
+            return "input", subject, category, rating
+        elif todo == "grade":
+            subject = input("To which subject do you want to add the grade? \n")
+            category = input("To which category do you want to add it? \n")
+            grade = input("Grade? \n")
+            gradename = input("What's the name of the grade? \n")
+            return "input", subject, category, grade, gradename
+