Support custom template

This commit is contained in:
KnugiHK
2021-07-10 21:46:45 +08:00
parent 35ef4031fc
commit 790f4ec5e0
3 changed files with 25 additions and 9 deletions
+7 -1
View File
@@ -67,6 +67,12 @@ def main():
default=None,
help="Path to key file"
)
parser.add_option(
"-t",
"--template",
dest="template",
default=None,
help="Path to custom HTML template")
(options, args) = parser.parse_args()
if options.android and options.iphone:
@@ -138,7 +144,7 @@ def main():
messages(db, data)
media(db, data, options.media)
vcard(db, data)
create_html(data, options.output)
create_html(data, options.output, options.template)
if not os.path.isdir(f"{options.output}/{options.media}"):
shutil.move(options.media, f"{options.output}/")