strict type checking

2 files changed, 9 insertions, 2 deletions

diff --git a/pyproject.toml b/pyproject.toml
index 461b620..1d168dd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -27,3 +27,6 @@ exclude = ["**/.gitignore"]
 
 [tool.isort]
 profile = "black"
+
+[tool.pyright]
+strict = ["**/*.py"]
diff --git a/ssg.py b/ssg.py
index da94b03..3b2ee94 100644
--- a/ssg.py
+++ b/ssg.py
@@ -16,7 +16,9 @@ import tomllib
 from jinja2 import Environment, FileSystemLoader
 
 
-def parse_args(p=ArgumentParser(), argv=sys.argv):
+def parse_args(
+    p: ArgumentParser = ArgumentParser(), argv: list[str] = sys.argv
+) -> Namespace:
     p.add_argument(
         "-o", "--output", default=Path("output"), help="Output directory", type=Path
     )
@@ -40,7 +42,9 @@ class Generator:
         self.env = Environment(
             loader=FileSystemLoader("."), autoescape=jinja2.select_autoescape()
         )
-        self.env.globals["prefix"] = args.prefix
+        self.env.globals[  # pyright: ignore [reportUnknownMemberType]
+            "prefix"
+        ] = args.prefix
 
     def write(self, dest: str, content: str | bytes) -> None:
         dest_path: Path = self.output / Path(dest)