summaryrefslogtreecommitdiff
path: root/src/pyssg/pyssg.py
blob: 6d0df3ad888e0dff1c0ec567409c83dd7113295f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import os
from argparse import ArgumentParser, Namespace

from .file_discovery import get_md_files
from .file_structure import create_structure


def get_options() -> Namespace:
    parser = ArgumentParser(prog='pyssg',
                            description='''Static Site Generator that reads
                            Markdown files and creates HTML files.''')
    parser.add_argument('-d', '--directory',
                        default='.',
                        type=str,
                        help='''root directory for all site files,
                        defaults to "." (cwd), uses relative or absolute
                        resolution''')
    parser.add_argument('-i', '--init',
                        action='store_true',
                        help='''initialize the directory structure where -d
                        specifies''')

    return parser.parse_args()


def main():
    opts = vars(get_options())
    directory = opts['directory']

    if opts['init']:
        create_structure(directory)

    os.chdir(directory)
    root_dir = os.getcwd()