calvin.erfmann/WhatsApp-Chat-Exporter
1
0
Fork 0
mirror of https://github.com/KnugiHK/WhatsApp-Chat-Exporter.git synced 2026-04-22 22:14:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add an option to disable html output

Browse Source
This commit is contained in:
KnugiHK
2023-03-25 18:26:03 +08:00
parent ee5f8b82be
commit cdfaf69f7a
1 changed files with 23 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
Whatsapp_Chat_Exporter/__main__.py
View File

@@ -6,7 +6,7 @@ from Whatsapp_Chat_Exporter import extract
from Whatsapp_Chat_Exporter import extract_iphone from Whatsapp_Chat_Exporter import extract_iphone
from Whatsapp_Chat_Exporter import extract_iphone_media from Whatsapp_Chat_Exporter import extract_iphone_media
from Whatsapp_Chat_Exporter.data_model import ChatStore from Whatsapp_Chat_Exporter.data_model import ChatStore
from Whatsapp_Chat_Exporter.extract import Crypt from Whatsapp_Chat_Exporter.utility import Crypt
from argparse import ArgumentParser from argparse import ArgumentParser
import os import os
import sqlite3 import sqlite3
@@ -69,7 +69,8 @@ def main():
dest='json', dest='json',
nargs='?', nargs='?',
default=None, default=None,
type=str, const="result.json", type=str,
const="result.json",
help="Save the result to a single JSON file (default if present: result.json)") help="Save the result to a single JSON file (default if present: result.json)")
parser.add_argument( parser.add_argument(
'-d', '-d',
@@ -129,6 +130,13 @@ def main():
default=None, default=None,
help="Maximum size of a single output file in bytes, 0 for auto (not yet implemented)" help="Maximum size of a single output file in bytes, 0 for auto (not yet implemented)"
) )
parser.add_argument(
"--no-html",
dest="no_html",
default=False,
action='store_true',
help="Do not output html files"
)
args = parser.parse_args() args = parser.parse_args()
if args.android and args.iphone: if args.android and args.iphone:
@@ -137,6 +145,10 @@ def main():
if not args.android and not args.iphone: if not args.android and not args.iphone:
print("You must define the device type.") print("You must define the device type.")
exit(1) exit(1)
if args.no_html and not args.json:
print("You must either specify a JSON output file or enable HTML output.")
exit(1)
data = {} data = {}
if args.android: if args.android:
@@ -217,14 +229,15 @@ def main():
messages(db, data) messages(db, data)
media(db, data, args.media) media(db, data, args.media)
vcard(db, data) vcard(db, data)
create_html( if not args.no_html:
data, create_html(
args.output, data,
args.template, args.output,
args.embedded, args.template,
args.offline, args.embedded,
args.size args.offline,
) args.size
)
else: else:
print( print(
"The message database does not exist. You may specify the path " "The message database does not exist. You may specify the path "