Update calendar_app.py

calendar_app.py

@@ -113,6 +113,42 @@ class Calendar:
                     print(f"{print_counter}. {dt} {event_description}")
                     print_counter+=1
 
+    
+    def check_valid_date(date):
+        """Input sanitation for valid date format"""
+        date_parts = date.split("-")
+        count_date_parts = len(date_parts)
+
+        # check 3 date parts are found: YYYY, MM, DD
+        if count_date_parts<3:
+            print(f"You entered {date} containing only {count_date_parts} date parts. Please enter date in YYYY-MM-DD format")
+            return "invalid_date"
+
+        # check separator of date parts
+        elif "." in date:
+            print(f"You entered {date} split on '.'. Please enter date in YYYY-MM-DD format split by '-'")
+            return "invalid_date"
+        elif "/" in date:
+            print(f"You entered {date} split on '/'. Please enter date in YYYY-MM-DD format split by '-'")
+            return "invalid_date"
+
+        # check length of date parts
+        elif len(date_parts[0])!=4:
+            print(f"You entered {date} with year value {date_parts[0]}. Please enter date in YYYY-MM-DD format, starting with a 4-digit year")
+            return "invalid_date"
+        elif len(date_parts[1])!=2:
+            print(f"You entered {date} with month value {date_parts[1]}. Please enter date in YYYY-MM-DD format, with a 2-digit month in the middle")
+            return "invalid_date"
+        elif len(date_parts[2])!=2:
+            print(f"You entered {date} with day value {date_parts[2]}. Please enter date in YYYY-MM-DD format, with a 2-digit day at the end")
+            return "invalid_date"
+
+        # in this last scenario, we have a valid date
+        else:
+            return "valid_date"
+            
+            
+
 
 
 
@@ -133,13 +169,16 @@ if __name__ == "__main__":
         # user create event reminder
         if choice == "1":
             date = input("Enter date (YYYY-MM-DD): ")
-            reminder = input("Enter description for event reminder: ")
-            calendar.add_reminder(date, reminder)
+            if check_valid_date(date) == "valid_date":
+                reminder = input("Enter description for event reminder: ")
+                calendar.add_reminder(date, reminder)
         elif choice == "2":
             date_start = input("Enter start date (YYYY-MM-DD): ")
             date_end = input("Enter end date (YYYY-MM-DD): ")
-            reminder = input("Enter description for event reminder: ")
-            calendar.add_daterange(date_start, date_end, reminder)
+            if check_valid_date(date_start) == "valid_date":
+                if check_valid_date(date_end) == "valid_date":
+                    reminder = input("Enter description for event reminder: ")
+                    calendar.add_daterange(date_start, date_end, reminder)
 
         # list ALL or future reminders
         elif choice == "3":
@@ -150,7 +189,8 @@ if __name__ == "__main__":
         # list reminders for specific date or month
         elif choice == "5":
             date = input("Enter date (YYYY-MM-DD): ")
-            calendar.show_reminders(date)
+            if check_valid_date(date) == "valid_date":
+                calendar.show_reminders(date)
         elif choice == "6":
             user_mth = input("Enter month number (MM): ")
             calendar.show_month(user_mth)