From 0ef5e7246bf659ac5da618570979f3846ba325ba Mon Sep 17 00:00:00 2001 From: Debjit Date: Tue, 9 Dec 2025 02:06:54 +0530 Subject: [PATCH] Commiting Date issue fix --- main.py | 5 ++++- tests/test_utils.py | 3 +-- utils.py | 8 ++++++-- 3 files changed, 11 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 0ba2987..4af7549 100644 --- a/main.py +++ b/main.py @@ -26,6 +26,9 @@ def main(): return deadline = format_date(args.deadline) + if deadline == "error": + print(f"Error: '{args.deadline}' is not a valid date. Please use YYYY-MM-DD format and ensure the date actually exists.") + return assignment = manager.add_assignment(args.value, deadline, args.subject) print(f"Added assignment: {assignment['title']}") @@ -57,4 +60,4 @@ def main(): if __name__ == '__main__': - main() + main() \ No newline at end of file diff --git a/tests/test_utils.py b/tests/test_utils.py index 4223a8e..8d665ca 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -13,7 +13,6 @@ class TestUtils(unittest.TestCase): - def test_format_date_valid(self): """Test format_date with valid date string""" result = format_date("2024-12-25") @@ -44,4 +43,4 @@ def test_get_priority_level(self): if __name__ == '__main__': - unittest.main() + unittest.main() \ No newline at end of file diff --git a/utils.py b/utils.py index 936dbe0..8197631 100644 --- a/utils.py +++ b/utils.py @@ -9,7 +9,11 @@ def format_date(date_string): """Convert date string to datetime object""" # TODO: Add error handling for invalid date formats - return datetime.strptime(date_string, "%Y-%m-%d") + try: + myDate = datetime.strptime(date_string, "%Y-%m-%d") + return myDate + except ValueError: + return "error" def calculate_days_remaining(deadline): @@ -50,4 +54,4 @@ def get_priority_level(days_remaining): elif days_remaining <= 7: return "MEDIUM" else: - return "LOW" + return "LOW" \ No newline at end of file