CLDR-18745 JIRA_classify_ticket_with_rules.py (unicode-org#4938)

Author: preetsojitra2712

tools/scripts/llm/JIRA_classify_ticket_with_rules.py

@@ -0,0 +1,117 @@
+import json
+import os
+import re # Import the regular expressions module
+from jira import JIRA
+
+def classify_ticket_with_rules_advanced(title, description):
+    """
+    Classifies a Jira ticket using an advanced rule-set with
+    regex and a scoring system.
+    """
+    text = (title + " " + (description or "")).lower()
+
+    # --- Initialize default classification and scoring ---
+    classification = {
+        "is_cldr_related": True,
+        "ticket_type": "task",
+        "component": "other",
+        "needs_engineering_work": False,
+        "needs_language_specialist": False,
+        "is_potential_duplicate": False,
+        "potential_duplicate_of": None,
+        "summary": "Classification based on advanced rules."
+    }
+    scores = {"priority": 0, "sentiment": 0}
+
+    # --- Advanced Rule 1: Use Regex to find duplicate ticket keys ---
+    # This pattern looks for "CLDR-" followed by one or more digits.
+    duplicate_match = re.search(r"cldr-\d+", text)
+    if duplicate_match:
+        classification["is_potential_duplicate"] = True
+        # .group(0) gets the matched text, .upper() makes it standard
+        classification["potential_duplicate_of"] = duplicate_match.group(0).upper()
+
+    # --- Advanced Rule 2: Scoring System for Priority ---
+    # Assign points based on keywords.
+    priority_keywords = {
+        "critical": ["critical", "crash", "urgent", "blocker", "unusable"],
+        "high": ["important", "high priority", "major", "severe"],
+        "low": ["trivial", "minor", "cosmetic", "suggestion"]
+    }
+    for word in priority_keywords["critical"]:
+        if word in text: scores["priority"] += 20
+    for word in priority_keywords["high"]:
+        if word in text: scores["priority"] += 10
+    for word in priority_keywords["low"]:
+        if word in text: scores["priority"] -= 5
+        
+    # --- Advanced Rule 3: Simple Sentiment Scoring ---
+    sentiment_keywords = {
+        "positive": ["great", "love", "helpful", "like", "suggestion"],
+        "negative": ["frustrating", "confusing", "broken", "hate", "fails"]
+    }
+    for word in sentiment_keywords["positive"]:
+        if word in text: scores["sentiment"] += 1
+    for word in sentiment_keywords["negative"]:
+        if word in text: scores["sentiment"] -= 1
+
+    # --- Rule 4: Basic Keyword Matching (can still be used) ---
+    if scores["sentiment"] < 0 or any(w in text for w in ["error", "bug"]):
+        classification["ticket_type"] = "bug"
+    elif any(w in text for w in ["add", "create", "implement", "feature"]):
+        classification["ticket_type"] = "feature"
+        
+    # --- Make final decisions based on scores ---
+    if scores["priority"] > 15:
+        classification["priority"] = "critical"
+    elif scores["priority"] > 5:
+        classification["priority"] = "high"
+    elif scores["priority"] < 0:
+        classification["priority"] = "low"
+    else:
+        classification["priority"] = "medium"
+        
+    # --- Set routing based on ticket type ---
+    if classification["ticket_type"] == "bug":
+        classification["routing_group"] = "QA Team"
+    else:
+        classification["routing_group"] = "CLDR Core Team"
+
+    # --- Final structure ---
+    return {
+        "classification_status": "valid_ticket",
+        "details": classification
+    }
+
+
+# (Configuration and Jira Connection logic)
+JIRA_SERVER = os.getenv('JIRA_SERVER', 'https://unicode-org.atlassian.net')
+JIRA_USER_EMAIL = os.getenv('JIRA_USER_EMAIL', '[email protected]')
+JIRA_API_TOKEN = os.getenv('JIRA_API_TOKEN', 'your_api_token_here')
+
+
+if not all([JIRA_SERVER, JIRA_USER_EMAIL, JIRA_API_TOKEN]):
+    print(" Error: One or more environment variables are not set.")
+else:
+    print(f"\n--- Connecting to Jira server at: {JIRA_SERVER} ---")
+    try:
+        jira_connection = JIRA(server=JIRA_SERVER, basic_auth=(JIRA_USER_EMAIL, JIRA_API_TOKEN))
+        print(" Connection successful!")
+
+        print(f"\n--- Fetching ticket: {TICKET_KEY_TO_FETCH} ---")
+        issue = jira_connection.issue(TICKET_KEY_TO_FETCH)
+        ticket_summary = issue.fields.summary
+        ticket_description = issue.fields.description
+        print(f"Title: {ticket_summary}")
+        print(f"Description: {ticket_description if ticket_description else 'No description found.'}")
+
+        print("\n" + "="*50)
+        print("--- Dynamic Advanced Rule-Based Classification ---")
+        print("="*50)
+        
+        advanced_classification = classify_ticket_with_rules_advanced(ticket_summary, ticket_description)
+        
+        print(json.dumps(advanced_classification, indent=2))
+
+    except Exception as e:
+        print(f" An error occurred: {e}")