diff options
Diffstat (limited to 'src/pyssg/pyssg.py')
-rw-r--r-- | src/pyssg/pyssg.py | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/src/pyssg/pyssg.py b/src/pyssg/pyssg.py index 7378705..5b736ca 100644 --- a/src/pyssg/pyssg.py +++ b/src/pyssg/pyssg.py @@ -1,8 +1,8 @@ import os from argparse import ArgumentParser, Namespace -from .templates import create_templates -from .parser import generate_static_site +from .template import Template +from .generator import generate_static_site def get_options() -> Namespace: @@ -31,13 +31,20 @@ def get_options() -> Namespace: return parser.parse_args() -def main(): +def main() -> None: opts = vars(get_options()) src = opts['src'] dst = opts['dst'] if opts['init']: - create_templates(src, dst) + try: + os.mkdir(src) + os.makedirs(os.path.join(dst, 'tag')) + except FileExistsError: + pass + + template = Template() + template.write_templates(src) return if opts['build']: |