Changes for Ruff and Timezone awareness

Author: Yeeyash

bug_triage.py

@@ -1,31 +1,30 @@
 """
 Automated Bug Triage Tool
-This script scans markdown bug reports, classifies them by severity based on 
+This script scans markdown bug reports, classifies them by severity based on
 keywords, and generates a summarized triage report in Markdown format.
 """
 
 import glob
 import os
-from datetime import datetime
-from typing import Dict, List
+from datetime import datetime, UTC
 from pathlib import Path
 
 # Configuration: Adjust these paths based on your local environment
 BASE_DIR = Path(__file__).parent
 BUG_PATH = str(BASE_DIR / "production" / "qa" / "bugs" / "*.md")
 OUTPUT_PATH = str(BASE_DIR / "production" / "qa")
 
+
 def classify_severity(content: str) -> str:
     """
     Classifies bug severity based on specific keywords found in the content.
     Returns S1 (Critical) through S4 (Minor).
     >>> classify_severity("The application had a fatal crash on startup.")
     'S1'
     >>> classify_severity("The UI is a bit slow today.")
-    'S3' 
+    'S3'
     """
     content = content.lower()
-
     if any(k in content for k in ["crash", "data loss", "cannot start", "fatal"]):
         return "S1"
     if any(k in content for k in ["broken", "not working", "fail"]):
@@ -43,25 +42,22 @@ def classify_priority(severity: str) -> str:
     >>> classify_priority("S4")
     'P4'
     """
-    priority_map = {
-        "S1": "P1",
-        "S2": "P2",
-        "S3": "P3"
-    }
+    priority_map = {"S1": "P1", "S2": "P2", "S3": "P3"}
     return priority_map.get(severity, "P4")
 
 
-def read_bugs() -> List[Dict]:
+def read_bugs() -> list[dict]:
     """
     Reads all markdown files in the BUG_PATH and extracts metadata.
+    >>> read_bugs()
+    []
     """
     files = glob.glob(BUG_PATH)
     bugs = []
 
-    # Sorting files ensures consistent BUG-ID assignment across runs
     for i, file_path in enumerate(sorted(files)):
         try:
-            with open(file_path, "r", encoding="utf-8") as f:
+            with open(file_path, encoding="utf-8") as f:
                 content = f.read()
 
             severity = classify_severity(content)
@@ -72,28 +68,31 @@ def read_bugs() -> List[Dict]:
                 "file": file_path,
                 "severity": severity,
                 "priority": priority,
-                # Extract first line as summary, capped at 80 chars
                 "summary": content.strip().split("\n")[0][:80]
             })
-        except IOError as e:
+        except OSError as e:
             print(f"⚠️ Could not read file {file_path}: {e}")
 
     return bugs
 
 
-def generate_report(bugs: List[Dict]) -> None:
+def generate_report(bugs: list[dict]) -> None:
     """
     Groups bugs by priority and writes a summarized Markdown report.
+    >>> generate_report([])
+    ❌ No bugs to report.
     """
-    date = datetime.now().strftime("%Y-%m-%d")
+    if not bugs:
+        print("❌ No bugs to report.")
+        return
+
+    date = datetime.now(UTC).strftime("%Y-%m-%d")
 
-    # Ensure the output directory exists
     if not os.path.exists(OUTPUT_PATH):
         os.makedirs(OUTPUT_PATH)
 
     output_file = os.path.join(OUTPUT_PATH, f"bug-triage-{date}.md")
 
-    # Filter bugs into priority buckets
     p1 = [b for b in bugs if b["priority"] == "P1"]
     p2 = [b for b in bugs if b["priority"] == "P2"]
     p3 = [b for b in bugs if b["priority"] == "P3"]
@@ -133,10 +132,7 @@ def generate_report(bugs: List[Dict]) -> None:
 
 
 if __name__ == "__main__":
+    import doctest
+    doctest.testmod()
     extracted_bugs = read_bugs()
-
-    if not extracted_bugs:
-        print(f"❌ No bug files found in: {BUG_PATH}")
-        print("Tip: Ensure the directory exists and contains .md files.")
-    else:
-        generate_report(extracted_bugs)
+    generate_report(extracted_bugs)