Make bump_version.py also bump the version in pyproject.toml (#445)

Author: i80and

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "snooty"
-version = "0.13.12.dev"
+version = "0.13.16.dev"
 description = ""
 authors = ["MongoDB, inc. <[email protected]>"]
 license = "Apache-2.0"

tools/bump_version.py

@@ -8,6 +8,9 @@
 CHANGELOG_PATH = Path("CHANGELOG.md")
 INCOMING_VERISON_PAT = re.compile(r"\d+\.\d+\.\d+")
 PAT = re.compile(r'"(\d+\.\d+\.\S+)"')
+PYPROJECT_VERSION_PAT = re.compile(
+    r"(?<=^version = )\"(\d+\.\d+\.\S+)\"$", re.MULTILINE
+)
 CHANGELOG_UNRELEASED_PAT = re.compile(
     r"\n## \[Unreleased\](?P<unreleased>.*)(?=\n## )", re.DOTALL
 )
@@ -91,6 +94,19 @@ def replace(match: Match[str]) -> str:
         f.truncate(0)
         f.write(data)
 
+    with Path("pyproject.toml").open("r+") as f:
+        data = f.read()
+        data, n_replaced = PYPROJECT_VERSION_PAT.subn(f'"{version_to_bump_to}"', data)
+        if n_replaced != 1:
+            print(
+                "Error bumping version: expected 1 version string in pyproject.toml",
+                file=sys.stderr,
+            )
+            sys.exit(1)
+        f.seek(0)
+        f.truncate(0)
+        f.write(data)
+
     if version_to_bump_to is not None and ".dev" not in version_to_bump_to:
         try:
             new_changelog = release_changelog(