[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

modify_changelog.py

@@ -1,16 +1,16 @@
 """
 Changelog Modifier and Version Bumper for Cargo.toml and Changelog.md
 
-This script automates the process of version bumping and changelog updates for Rust projects managed with Cargo. 
-It reads the version from the cargo.toml file, increments it based on the specified component (major, minor, or patch), 
+This script automates the process of version bumping and changelog updates for Rust projects managed with Cargo.
+It reads the version from the cargo.toml file, increments it based on the specified component (major, minor, or patch),
 and updates both the cargo.toml and changelog.md files accordingly.
 
 It can also add PR entries to the UNRELEASED section of the changelog.
 
 Usage:
     Version bumping:
     $ python modify_changelog.py bump_version [major|minor|patch]
-    
+
     Adding PR entry:
     $ python modify_changelog.py update_changelog <number> <title>
 
@@ -20,7 +20,7 @@
         major - Increments the major component of the version, sets minor and patch to 0
         minor - Increments the minor component of the version, sets patch to 0
         patch - Increments the patch component of the version
-    
+
     update_changelog - Add a PR entry to the UNRELEASED section
         number - PR number
         title - PR title
@@ -59,20 +59,19 @@ def find_unreleased_section(lines):
     """Find the line number where UNRELEASED section starts and ends."""
     unreleased_start = None
     content_start = None
-    
+
     for i, line in enumerate(lines):
         if line.strip() == "## UNRELEASED":
             unreleased_start = i
             continue
-        
+
         if unreleased_start is not None and content_start is None:
             # Skip empty lines after ## UNRELEASED
             if line.strip() == "":
                 continue
-            else:
-                content_start = i
-                break
-    
+            content_start = i
+            break
+
     return unreleased_start, content_start
 
 
@@ -87,35 +86,34 @@ def update_changelog_version(file_path: Path, new_version: str) -> None:
 
 def update_changelog_pr(file_path: Path, pr_number: str, pr_title: str, pr_url: str) -> bool:
     """Update the changelog with the new PR entry. If entry exists, update it; otherwise add new entry."""
-    
     # Read the current changelog
-    with open(file_path, 'r', encoding='utf-8') as f:
+    with open(file_path, encoding="utf-8") as f:
         lines = f.readlines()
-    
+
     # Find the UNRELEASED section
     unreleased_start, content_start = find_unreleased_section(lines)
-    
+
     if unreleased_start is None:
         print("ERROR: Could not find '## UNRELEASED' section in changelog")
         return False
-    
+
     if content_start is None:
         print("ERROR: Could not find content start after UNRELEASED section")
         return False
-    
+
     # Create the new entry
     new_entry = f"- {pr_title} [#{pr_number}]({pr_url})\n"
-    
+
     # Check if this PR entry already exists and update it if so
     existing_entry_index = None
     for i, line in enumerate(lines[content_start:], start=content_start):
         if f"[#{pr_number}]" in line:
             existing_entry_index = i
             break
         # Stop checking when we reach the next section
-        if line.startswith("## ") and not line.strip() == "## UNRELEASED":
+        if line.startswith("## ") and line.strip() != "## UNRELEASED":
             break
-    
+
     if existing_entry_index is not None:
         # Update existing entry
         lines[existing_entry_index] = new_entry
@@ -124,11 +122,11 @@ def update_changelog_pr(file_path: Path, pr_number: str, pr_title: str, pr_url:
         # Insert the new entry at the beginning of the unreleased content
         lines.insert(content_start, new_entry)
         print(f"Added changelog entry for PR #{pr_number}: {pr_title}")
-    
+
     # Write the updated changelog
-    with open(file_path, 'w', encoding='utf-8') as f:
+    with open(file_path, "w", encoding="utf-8") as f:
         f.writelines(lines)
-    
+
     return True
 
 
@@ -141,7 +139,7 @@ def handle_bump_version(args):
     if not cargo_path.exists():
         print("ERROR: Cargo.toml not found.")
         sys.exit(1)
-    
+
     if not changelog_path.exists():
         print("ERROR: Changelog file not found.")
         sys.exit(1)
@@ -164,49 +162,47 @@ def handle_update_changelog(args):
     """Handle update changelog subcommand."""
     pr_number = args.number
     pr_title = args.title
-    
+
     # Construct PR URL from repository info and PR number
     # Default to the egglog-python repository
     pr_url = f"https://github.com/egraphs-good/egglog-python/pull/{pr_number}"
-    
-    changelog_path = Path(getattr(args, 'changelog_path', 'docs/changelog.md'))
-    
+
+    changelog_path = Path(getattr(args, "changelog_path", "docs/changelog.md"))
+
     if not changelog_path.exists():
         print(f"ERROR: Changelog file not found: {changelog_path}")
         sys.exit(1)
-    
+
     success = update_changelog_pr(changelog_path, pr_number, pr_title, pr_url)
     if not success:
         sys.exit(1)
 
 
 def main():
-    parser = argparse.ArgumentParser(description='Changelog modifier and version bumper')
-    subparsers = parser.add_subparsers(dest='command', help='Available commands')
-    
+    parser = argparse.ArgumentParser(description="Changelog modifier and version bumper")
+    subparsers = parser.add_subparsers(dest="command", help="Available commands")
+
     # Bump version subcommand
-    bump_parser = subparsers.add_parser('bump_version', help='Bump version and update changelog')
-    bump_parser.add_argument('bump_type', choices=['major', 'minor', 'patch'], 
-                            help='Type of version bump')
-    
+    bump_parser = subparsers.add_parser("bump_version", help="Bump version and update changelog")
+    bump_parser.add_argument("bump_type", choices=["major", "minor", "patch"], help="Type of version bump")
+
     # Update changelog subcommand
-    changelog_parser = subparsers.add_parser('update_changelog', help='Add PR entry to changelog')
-    changelog_parser.add_argument('number', help='Pull request number')
-    changelog_parser.add_argument('title', help='Pull request title')
-    changelog_parser.add_argument('--changelog-path', default='docs/changelog.md', 
-                                 help='Path to changelog file')
-    
+    changelog_parser = subparsers.add_parser("update_changelog", help="Add PR entry to changelog")
+    changelog_parser.add_argument("number", help="Pull request number")
+    changelog_parser.add_argument("title", help="Pull request title")
+    changelog_parser.add_argument("--changelog-path", default="docs/changelog.md", help="Path to changelog file")
+
     args = parser.parse_args()
-    
+
     if not args.command:
         parser.print_help()
         sys.exit(1)
-    
-    if args.command == 'bump_version':
+
+    if args.command == "bump_version":
         handle_bump_version(args)
-    elif args.command == 'update_changelog':
+    elif args.command == "update_changelog":
         handle_update_changelog(args)
 
 
 if __name__ == "__main__":
-    main()
+    main()