initial template creation, barebones arg parser

1 files changed, 34 insertions, 0 deletions

diff --git a/src/pyssg/pyssg.py b/src/pyssg/pyssg.py
new file mode 100644
index 0000000..6d0df3a
--- /dev/null
+++ b/src/pyssg/pyssg.py
@@ -0,0 +1,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()