Unwind pre-tomllib workaround (#7121)

Author: achamayou

.github/workflows/release.yml

@@ -51,13 +51,13 @@ jobs:
       - name: "Check Release Notes"
         run: |
           set -ex
-          python scripts/extract-release-notes.py --target-git-version
+          scripts/extract-release-notes.py --target-git-version
         shell: bash
       - name: "Produce Release Notes"
         run: |
           set -ex
           set -o pipefail
-          python ./scripts/extract-release-notes.py --target-git-version --describe-path-changes "./samples/constitution" | tee rel-notes.md
+          ./scripts/extract-release-notes.py --target-git-version --describe-path-changes "./samples/constitution" | tee rel-notes.md
       - name: "Upload Release Notes"
         uses: actions/upload-artifact@v4
         with:

scripts/ci-checks.sh

@@ -68,9 +68,9 @@ endgroup
 
 group "Release notes"
 if [ $FIX -ne 0 ]; then
-  python3 "$SCRIPT_DIR"/extract-release-notes.py -f
+  "$SCRIPT_DIR"/extract-release-notes.py -f
 else
-  python3 "$SCRIPT_DIR"/extract-release-notes.py
+  "$SCRIPT_DIR"/extract-release-notes.py
 fi
 endgroup
 

scripts/extract-release-notes.py

@@ -1,10 +1,12 @@
+#!/usr/bin/env python3
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the Apache 2.0 License.
 
 import argparse
 import re
 import sys
 import subprocess
+import tomllib
 
 
 def main():
@@ -53,15 +55,11 @@ def main():
     release_notes = {}
     links_found = []
 
-    # Check that pyproject.toml is up to date
-    # Once we have upgraded to Python 3.11, we can use tomllib to parse pyproject.toml
-    pyproject_version = None
-    with open("python/pyproject.toml") as pyproject:
-        for line in pyproject:
-            if line.startswith("version"):
-                _, version = line.split("=")
-                pyproject_version = version.strip().strip('"')
-    assert pyproject_version is not None, "Could not find version in pyproject.toml"
+    version_in_pyproject = None
+    with open("python/pyproject.toml", "rb") as pyproject:
+        config = tomllib.load(pyproject)
+        version_in_pyproject = config.get("project", {}).get("version")
+    assert version_in_pyproject is not None, "Could not find version in pyproject.toml"
 
     # Parse file, bucketing lines into each version's release notes
     current_release_notes = None
@@ -72,8 +70,8 @@ def main():
                 current_release_notes = []
                 if not release_notes:
                     assert (
-                        log_version == pyproject_version
-                    ), f"First version in CHANGELOG ({log_version}) must match version in pyproject.toml ({pyproject_version})"
+                        log_version == version_in_pyproject
+                    ), f"First version in CHANGELOG ({log_version}) must match version in pyproject.toml ({version_in_pyproject})"
                 release_notes[log_version] = current_release_notes
             elif match := link_definition.match(line):
                 link_version = match.group(1)