Provide helper chdir method on the context object.

Fixes #6

Signed-off-by: Pedro Algarvio <[email protected]>

Author: s0undt3ch

changelog/6.improvement.rst

@@ -0,0 +1 @@
+Provide helper `chdir` method on the context object.

src/ptscripts/parser.py

@@ -10,6 +10,8 @@
 import pathlib
 import sys
 import typing
+from collections.abc import Iterator
+from contextlib import contextmanager
 from functools import partial
 from subprocess import CompletedProcess
 from types import FunctionType
@@ -147,6 +149,21 @@ def run(
             capture=capture,
         )
 
+    @contextmanager
+    def chdir(self, path: pathlib.Path) -> Iterator[pathlib.Path]:
+        """
+        Change the current working directory to the provided path.
+        """
+        cwd = pathlib.Path.cwd()
+        try:
+            os.chdir(path)
+            yield path
+        finally:
+            if not cwd.exists():
+                self.error(f"Unable to change back to path {cwd}")
+            else:
+                os.chdir(cwd)
+
 
 class Parser:
     """