calvin.erfmann/WhatsApp-Chat-Exporter
1
0
Fork 0
mirror of https://github.com/KnugiHK/WhatsApp-Chat-Exporter.git synced 2026-05-30 21:09:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle case that the database file does not exist and clearer exit code

Browse Source
This commit is contained in:
KnugiHK
2021-12-30 11:59:30 +08:00
parent 9140c07feb
commit 1faf111e64
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
Whatsapp_Chat_Exporter/__main__.py
View File

@@ -6,6 +6,7 @@ import os
import sqlite3 import sqlite3
import shutil import shutil
import json import json
from sys import exit
def main(): def main():
@@ -79,10 +80,10 @@ def main():
if options.android and options.iphone: if options.android and options.iphone:
print("You must define only one device type.") print("You must define only one device type.")
exit() exit(1)
if not options.android and not options.iphone: if not options.android and not options.iphone:
print("You must define the device type.") print("You must define the device type.")
exit() exit(1)
data = {} data = {}
if options.android: if options.android:
@@ -98,7 +99,7 @@ def main():
if options.key is not None: if options.key is not None:
if options.backup is None: if options.backup is None:
print("You must specify the backup file with -b") print("You must specify the backup file with -b")
return False exit(1)
print("Decryption key specified, decrypting WhatsApp backup...") print("Decryption key specified, decrypting WhatsApp backup...")
key = open(options.key, "rb").read() key = open(options.key, "rb").read()
db = open(options.backup, "rb").read() db = open(options.backup, "rb").read()
@@ -106,7 +107,7 @@ def main():
if not extract.decrypt_backup(db, key, msg_db, is_crypt14): if not extract.decrypt_backup(db, key, msg_db, is_crypt14):
print("Dependencies of decrypt_backup are not " print("Dependencies of decrypt_backup are not "
"present. For details, see README.md") "present. For details, see README.md")
return False exit(3)
if options.wa is None: if options.wa is None:
contact_db = "wa.db" contact_db = "wa.db"
else: else:
@@ -148,12 +149,19 @@ def main():
media(db, data, options.media) media(db, data, options.media)
vcard(db, data) vcard(db, data)
create_html(data, options.output, options.template) create_html(data, options.output, options.template)
else:
print(
"The message database does not exist. You may specify the path "
"to database file with option -d or check your provided path.",
end="\r"
)
exit(2)
if not os.path.isdir(f"{options.output}/{options.media}"): if os.path.isdir(options.media) and not os.path.isdir(f"{options.output}/{options.media}"):
try: try:
shutil.move(options.media, f"{options.output}/") shutil.move(options.media, f"{options.output}/")
except PermissionError: except PermissionError:
print(f"Cannot remove original WhatsApp directory. Perhaps the directory is opened?") print("Cannot remove original WhatsApp directory. Perhaps the directory is opened?")
if options.json: if options.json:
with open("result.json", "w") as f: with open("result.json", "w") as f: