summaryrefslogtreecommitdiffstats
path: root/ssg.py
diff options
context:
space:
mode:
Diffstat (limited to 'ssg.py')
-rw-r--r--ssg.py8
1 files changed, 6 insertions, 2 deletions
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)