Implement category determination using a dedicated Python script

Author: SableRaf

.github/workflows/issue_to_pr.yml

@@ -35,26 +35,18 @@ jobs:
       - name: Install dependencies
         run: pip install -r requirements.txt
 
+      - name: Get issue labels
+        id: getLabels
+        uses: snnaplab/get-labels-action@v1
+        with:
+          issue-number: ${{ github.event.issue.number }}
+
       - name: Determine category from labels
         id: determineCategory
         run: |
-          labels_json="${{ toJson(github.event.issue.labels) }}"
+          labels_json="${{ steps.getLabels.outputs.labels }}"
           echo "Labels: $labels_json"
-          category=$(
-            if [[ $(echo "$labels_json" | grep -q 'examples') ]]; then
-              echo 'examples'
-            elif [[ $(echo "$labels_json" | grep -q 'mode') ]]; then
-              echo 'mode'
-            elif [[ $(echo "$labels_json" | grep -q 'tool') ]]; then
-              echo 'tool'
-            elif [[ $(echo "$labels_json" | grep -q 'library') ]]; then
-              echo 'library'
-            fi
-          )
-          if [ -z "$category" ]; then
-            echo "Category is empty. Please ensure the issue has a valid label."
-            exit 1
-          fi
+          category=$(python scripts/determine_category.py "$labels_json")
           echo "category=$category" >> $GITHUB_OUTPUT
           echo "Category found: $category"
 

scripts/determine_category.py

@@ -0,0 +1,32 @@
+"""
+Reads issue labels from a JSON string and determines the category of the issue.
+"""
+
+import json
+import sys
+
+valid_categories = ["examples", "mode", "tool", "library"]
+
+def determine_category(labels_json):
+    labels = json.loads(labels_json)
+    found_categories = [label for label in labels if label in valid_categories]
+
+    if len(found_categories) == 1:
+        return found_categories[0]
+    elif len(found_categories) > 1:
+        raise ValueError(f"Multiple valid categories found: ({', '.join(found_categories)}). Please ensure only one valid label is applied.")
+    else:
+        return None
+
+if __name__ == "__main__":
+    labels_json = sys.argv[1]
+    try:
+        category = determine_category(labels_json)
+        if category:
+            print(f"category={category}")
+        else:
+            print(f"Category is empty. Please ensure the issue has a valid label (options: {', '.join(valid_categories)}).")
+            sys.exit(1)
+    except ValueError as e:
+        print(str(e))
+        sys.exit(1)