Add directory(location) contextmanager

lightweight/__init__.py

@@ -4,4 +4,4 @@
 from .path import SitePath
 from .template import template
 
-__version__ = '1.0.0.dev22'
+__version__ = '1.0.0.dev23'

lightweight/files.py

@@ -1,6 +1,8 @@
+import os
+from contextlib import contextmanager
 from glob import iglob
 from pathlib import Path
-from typing import Iterator, Optional, Union
+from typing import Iterator, Union
 
 
 def paths(glob: Union[str, Path]) -> Iterator[Path]:
@@ -11,3 +13,18 @@ def paths(glob: Union[str, Path]) -> Iterator[Path]:
         return iter([glob])
     return map(Path, iglob(glob, recursive=True))
 
+
+@contextmanager
+def directory(location: str):
+    """Execute the following statements with provided location as "cwd" (current working directory).
+
+        :Example:
+        project_location = os.path.dirname(os.path.realpath(__file__))
+        with directory(project_location):
+            site.include('index.html')
+
+    """
+    cwd = os.getcwd()
+    os.chdir(location)
+    yield
+    os.chdir(cwd)

tests/test_paths.py → tests/test_files.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 
 from lightweight import paths
+from lightweight.files import directory
 
 
 def test_dir():
@@ -24,3 +25,8 @@ def test_recursive_files():
 
 def test_path():
     assert set(paths(Path('resources/test.html'))) == {Path('resources/test.html')}
+
+
+def test_directory():
+    with directory('site'), open('file') as f:
+        assert 'A test file.' == f.read()