mirror of
https://github.com/venthur/blag.git
synced 2025-11-25 20:52:43 +00:00
added templatedir argument
This commit is contained in:
16
blag/blag.py
16
blag/blag.py
@@ -60,7 +60,11 @@ def parse_args(args):
|
|||||||
default='build',
|
default='build',
|
||||||
help='Ouptut directory (default: build)',
|
help='Ouptut directory (default: build)',
|
||||||
)
|
)
|
||||||
|
build_parser.add_argument(
|
||||||
|
'-t', '--template-dir',
|
||||||
|
default='templates',
|
||||||
|
help='Template directory (default: templates)',
|
||||||
|
)
|
||||||
return parser.parse_args()
|
return parser.parse_args()
|
||||||
|
|
||||||
|
|
||||||
@@ -85,14 +89,18 @@ def build(args):
|
|||||||
path = os.path.relpath(f'{root}/{dirname}', start=args.input_dir)
|
path = os.path.relpath(f'{root}/{dirname}', start=args.input_dir)
|
||||||
os.makedirs(f'{args.output_dir}/{path}', exist_ok=True)
|
os.makedirs(f'{args.output_dir}/{path}', exist_ok=True)
|
||||||
|
|
||||||
convert_to_html(convertibles, args.input_dir, args.output_dir)
|
convert_to_html(
|
||||||
|
convertibles, args.input_dir,
|
||||||
|
args.output_dir,
|
||||||
|
args.template_dir
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def convert_to_html(convertibles, input_dir, output_dir):
|
def convert_to_html(convertibles, input_dir, output_dir, template_dir):
|
||||||
|
|
||||||
env = Environment(
|
env = Environment(
|
||||||
loader=ChoiceLoader([
|
loader=ChoiceLoader([
|
||||||
FileSystemLoader(['templates']),
|
FileSystemLoader([template_dir]),
|
||||||
PackageLoader('blag', 'templates'),
|
PackageLoader('blag', 'templates'),
|
||||||
])
|
])
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user