allow scripts to be run locally

Author: sleepyStick

.evergreen/remove-unimplemented-tests.sh

@@ -43,4 +43,4 @@ rm $PYMONGO/test/crud/unified/updateMany-rawdata.json
 rm $PYMONGO/test/crud/unified/updateOne-rawdata.json
 rm $PYMONGO/test/index_management/index-rawdata.json
 
-echo "done removing unimplemented tests"
+echo "Done removing unimplemented tests\n"

.evergreen/scripts/resync-all-specs.py

@@ -4,11 +4,14 @@
 import os
 import pathlib
 import subprocess
+from argparse import Namespace
 from subprocess import CalledProcessError
+from typing import Optional
 
 
 def resync_specs(directory: pathlib.Path, errored: dict[str, str]) -> None:
     """Actually sync the specs"""
+    print("Beginning to sync specs")  # noqa: T201
     for spec in os.scandir(directory):
         if not spec.is_dir():
             continue
@@ -24,11 +27,13 @@ def resync_specs(directory: pathlib.Path, errored: dict[str, str]) -> None:
             )
         except CalledProcessError as exc:
             errored[spec.name] = exc.stderr
+    print("Done syncing specs")  # noqa: T201
 
 
 def apply_patches():
+    print("Beginning to apply patches")  # noqa: T201
     subprocess.run(["bash", "./.evergreen/remove-unimplemented-tests.sh"], check=True)  # noqa: S603, S607
-    subprocess.run(["git apply -R --allow-empty ./.evergreen/patch/*"], shell=True, check=True)  # noqa: S602, S607
+    subprocess.run(["git apply -R --allow-empty ./.evergreen/spec-patch/*"], shell=True, check=True)  # noqa: S602, S607
 
 
 def check_new_spec_directories(directory: pathlib.Path) -> list[str]:
@@ -61,7 +66,7 @@ def check_new_spec_directories(directory: pathlib.Path) -> list[str]:
     return list(spec_set - test_set)
 
 
-def write_summary(errored: dict[str, str], new: list[str]) -> None:
+def write_summary(errored: dict[str, str], new: list[str], filename: Optional[str]) -> None:
     """Generate the PR description"""
     pr_body = ""
     process = subprocess.run(
@@ -86,24 +91,29 @@ def write_summary(errored: dict[str, str], new: list[str]) -> None:
         pr_body += "\n -".join(new)
         pr_body += "\n"
     if pr_body != "":
-        with open("spec_sync.txt", "w") as f:
-            # replacements made for proper json
-            f.write(pr_body.replace("\n", "\\n").replace("\t", "\\t"))
+        if filename is None:
+            print(f"\n{pr_body}")  # noqa: T201
+        else:
+            with open(filename, "w") as f:
+                # replacements made for proper json
+                f.write(pr_body.replace("\n", "\\n").replace("\t", "\\t"))
 
 
-def main():
+def main(args: Namespace):
     directory = pathlib.Path("./test")
     errored: dict[str, str] = {}
     resync_specs(directory, errored)
     apply_patches()
     new = check_new_spec_directories(directory)
-    write_summary(errored, new)
+    write_summary(errored, new, args.filename)
 
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
         description="Python Script to resync all specs and generate summary for PR."
     )
-    parser.add_argument("filename", help="Name of file for the summary to be written into.")
+    parser.add_argument(
+        "--filename", help="Name of file for the summary to be written into.", default=None
+    )
     args = parser.parse_args()
-    main()
+    main(args)

.evergreen/scripts/resync-all-specs.sh

@@ -3,9 +3,9 @@
 
 # SETUP
 SRC_URL="https://github.com/mongodb/specifications.git"
-# needs to be set for resunc-specs.sh
-SPEC_SRC="$(realpath -s "../specifications")"
-SCRIPT="$(realpath -s "./.evergreen/resync-specs.sh")"
+# needs to be set for resync-specs.sh
+SPEC_SRC="$(realpath "../specifications")"
+SCRIPT="$(realpath "./.evergreen/resync-specs.sh")"
 
 # Clone the spec repo if the directory does not exist
 if [[ ! -d $SPEC_SRC ]]; then
@@ -28,11 +28,15 @@ fi
 PR_DESC="spec_sync.txt"
 
 # run python script that actually does all the resyncing
-/opt/devtools/bin/python3.11 ./.evergreen/scripts/resync-all-specs.py "$PR_DESC"
-
-
-if [[ -f $PR_DESC ]]; then
-  # changes were made -> call scrypt to create PR for us
-  .evergreen/scripts/create-pr.sh "$PR_DESC"
-  rm "$PR_DESC"
+if ! [ -n "${CI:-}" ]
+then
+  # we're running locally
+  python3 ./.evergreen/scripts/resync-all-specs.py
+else
+  /opt/devtools/bin/python3.11 ./.evergreen/scripts/resync-all-specs.py "$PR_DESC"
+  if [[ -f $PR_DESC ]]; then
+    # changes were made -> call scrypt to create PR for us
+    .evergreen/scripts/create-spec-pr.sh "$PR_DESC"
+    rm "$PR_DESC"
+  fi
 fi