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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
import os
import sys
import json
from importlib.resources import path as rpath
from typing import Union
from logging import Logger, getLogger, DEBUG
from argparse import ArgumentParser
from pyssg.arg_parser import get_parser
from pyssg.utils import create_dir, copy_file, get_expanded_path
from pyssg.cfg.configuration import get_parsed_config, VERSION
from pyssg.db.database import Database
from pyssg.builder import Builder
log: Logger = getLogger(__name__)
def main() -> None:
arg_parser: ArgumentParser = get_parser()
args: dict[str, Union[str, bool]] = vars(arg_parser.parse_args())
# TODO: move this logic to the logger
# too messy to place at utils.py, don't want to be
# passing the arg parser around
def _log_perror(message: str) -> None:
arg_parser.print_usage()
# even if it's an error, print it as info
# as it is not critical, only config related
log.info(f'pyssg: error: {message}, --help for more')
sys.exit(1)
# -1 as first argument is program path
num_args = len(sys.argv) - 1
if num_args == 2 and args['config']:
_log_perror('only config argument passed')
elif not num_args > 0 or (num_args == 1 and args['debug']):
_log_perror('no arguments passed')
elif num_args == 3 and (args['debug'] and args['config']):
_log_perror('no arguments passed other than "debug" and "config"')
if args['version']:
log.info('pyssg v%s', VERSION)
sys.exit(0)
# TODO: move this logic to the logger
if args['debug']:
# need to modify the root logger specifically,
# as it is the one that holds the config
# (__name__ happens to resolve to pyssg in __init__)
root_logger: Logger = getLogger('pyssg')
root_logger.setLevel(DEBUG)
for handler in root_logger.handlers:
handler.setLevel(DEBUG)
log.debug('changed logging level to DEBUG')
if args['init']:
idir: str = os.path.normpath(get_expanded_path(str(args['init'])))
log.info('initializing directory structure and copying templates')
create_dir(idir)
with rpath('pyssg.plt', 'default.yaml') as p:
copy_file(str(p), os.path.join(idir, 'config.yaml'))
create_dir(os.path.join(idir, 'src'))
create_dir(os.path.join(idir, 'dst'))
create_dir(os.path.join(idir, 'plt'))
files: list[str] = ['index.html',
'page.html',
'tag.html',
'rss.xml',
'sitemap.xml',
'entry.md']
log.debug('list of files to copy over: %s', files)
for f in files:
plt_file: str = os.path.join(os.path.join(idir, 'plt'), f)
with rpath('pyssg.plt', f) as p:
copy_file(str(p), plt_file)
log.info('finished initialization')
sys.exit(0)
config_path: str = get_expanded_path(str((args['config']))) \
if args['config'] else 'config.yaml'
if not os.path.exists(config_path):
_log_perror(f'config file "{config_path}" doesn\'t exist')
log.debug('reading config file')
config: list[dict] = get_parsed_config(config_path)
# print(json.dumps(config, sort_keys=True, indent=2))
if args['build']:
log.info('building the html files')
db: Database = Database(config[0]['path']['db'])
log.debug('building all dir_paths found in conf')
builder: Builder = Builder(config[0], db, config[1])
builder.build()
db.write()
log.info('finished building the html files')
sys.exit(0)
|
