Update create_release_tag.py

Author: wangzelin007

.github/workflows/create_release_tag.py

@@ -6,6 +6,7 @@
 import zipfile
 from typing import List, Dict, Tuple, Optional
 import sys
+import json
 
 TARGET_FILE = "src/index.json"
 
@@ -22,7 +23,8 @@ def get_api_urls() -> Tuple[str, str]:
     return f"{base_url}/releases", f"{base_url}/git/tags"
 
 
-def get_file_changes() -> List[str]:
+def get_file_diff() -> List[str]:
+    """Get added lines from git diff"""
     diff_output = subprocess.check_output(
         ["git", "diff", "HEAD^", "HEAD", "--", TARGET_FILE],
         text=True
@@ -37,70 +39,37 @@ def get_file_changes() -> List[str]:
     return added_lines
 
 
-def parse_filename(added_lines: List[str]) -> Optional[Dict]:
-    """Parse filename and metadata from added lines"""
-    filename = None
-    metadata = {}
-    download_url = None
-    sha256_digest = None
-    in_metadata = False
-
+def parse_filename(added_lines: List[str]) -> Optional[str]:
+    """Parse filename from added lines"""
     for line in added_lines:
-        line = line.strip()
-
         if '"filename":' in line:
             try:
                 filename = line.split(":")[1].strip().strip('",')
+                return filename
             except IndexError:
                 print(f"Error parsing line: {line}")
+    return None
 
-        elif '"downloadUrl":' in line:
-            try:
-                download_url = line.split(":")[1].strip().strip('",')
-            except IndexError:
-                print(f"Error parsing downloadUrl: {line}")
-
-        elif '"sha256Digest":' in line:
-            try:
-                sha256_digest = line.split(":")[1].strip().strip('",')
-            except IndexError:
-                print(f"Error parsing sha256Digest: {line}")
-
-        elif '"metadata":' in line:
-            in_metadata = True
-            continue
-
-        elif in_metadata and line.startswith('"'):
-            # Parse metadata key-value pairs
-            try:
-                key = line.split(":")[0].strip().strip('"')
-                value = line.split(":")[1].strip().strip('",')
-                metadata[key] = value
-            except IndexError:
-                print(f"Error parsing metadata line: {line}")
 
-        elif line.startswith("}"):
-            in_metadata = False
+def get_extension_info(filename: str) -> Optional[Dict]:
+    """Get extension information from index.json"""
+    try:
+        with open(TARGET_FILE, 'r') as f:
+            index_data = json.load(f)
 
-    if not filename:
-        print("No filename found in added lines")
-        return None
+        # Search for the extension entry with matching filename
+        for ext_name, versions in index_data.get("extensions", {}).items():
+            for version in versions:
+                if version.get("filename") == filename:
+                    return version
 
-    if not download_url:
-        print("No downloadUrl found in added lines")
+        print(f"Extension {filename} not found in index.json")
         return None
 
-    if not sha256_digest:
-        print("No sha256Digest found in added lines")
+    except Exception as e:
+        print(f"Error reading index.json: {e}")
         return None
 
-    return {
-        "filename": filename,
-        "metadata": metadata,
-        "downloadUrl": download_url,
-        "sha256Digest": sha256_digest
-    }
-
 
 def generate_tag_and_title(filename: str) -> Tuple[str, str, str]:
     """Generate tag name and release title from filename"""
@@ -115,13 +84,15 @@ def generate_tag_and_title(filename: str) -> Tuple[str, str, str]:
     release_title = f"{name} {version}"
     return tag_name, release_title, version
 
+
 def check_tag_exists(url: str, tag_name: str, headers: Dict[str, str]) -> bool:
     response = requests.get(
         f"{url}/{tag_name}",
         headers=headers
     )
     return response.status_code == 200
 
+
 def create_release(url: str, release_data: Dict[str, str], headers: Dict[str, str], wheel_url: str = None) -> None:
     try:
         # Create release
@@ -155,7 +126,7 @@ def create_release(url: str, release_data: Dict[str, str], headers: Dict[str, st
                 data=wheel_response.content
             )
             upload_response.raise_for_status()
-            print(f"Successfully uploaded wheel file")
+            print("Successfully uploaded wheel file")
 
     except requests.exceptions.RequestException as e:
         print(f"\nError creating release for {release_data['tag_name']}")
@@ -237,31 +208,37 @@ def main():
         sys.exit(1)
 
     release_url, tag_url = get_api_urls()
-
     headers = {
         "Authorization": f"token {github_token}",
         "Accept": "application/vnd.github.v3+json"
     }
 
     try:
-        added_lines = get_file_changes()
+        # Get added lines from git diff
+        added_lines = get_file_diff()
         if not added_lines:
             print("No changes found in index.json")
             return
 
-        file_info = parse_filename(added_lines)
-        if not file_info:
+        # Parse filename from added lines
+        filename = parse_filename(added_lines)
+        if not filename:
             print("No filename found in changes")
             return
 
+        # Get extension info from index.json
+        extension_info = get_extension_info(filename)
+        if not extension_info:
+            print("Could not get extension information from index.json")
+            return
+
         try:
-            filename = file_info["filename"]
             tag_name, release_title, version = generate_tag_and_title(filename)
 
             # Get history notes from wheel package
             print("Getting history notes from wheel package...")
-            history_note = get_history_note(file_info["downloadUrl"], version)
-
+            history_note = get_history_note(extension_info["downloadUrl"], version)
+            
             # If no notes found in wheel, try source code
             if "No history notes found" in history_note:
                 print("No history notes found in wheel, trying source code...")
@@ -274,7 +251,7 @@ def main():
                     history_note = f"Release {extension_name} {version}"
 
             # Generate release body
-            release_body = generate_release_body(history_note, file_info["sha256Digest"], filename)
+            release_body = generate_release_body(history_note, extension_info["sha256Digest"], filename)
 
             commit_sha = subprocess.check_output(
                 ["git", "rev-parse", "HEAD"],
@@ -298,7 +275,7 @@ def main():
             print(f"Target commit: {commit_sha}")
             print(f"Body preview: {release_body[:200]}...")
 
-            create_release(release_url, release_data, headers, file_info["downloadUrl"])
+            create_release(release_url, release_data, headers, extension_info["downloadUrl"])
 
         except ValueError as e:
             print(f"Error generating tag for filename {filename}: {e}")