mirror of
https://github.com/KnugiHK/WhatsApp-Chat-Exporter.git
synced 2026-01-28 21:30:43 +00:00
123 lines
4.5 KiB
Python
123 lines
4.5 KiB
Python
#!/usr/bin/python3
|
|
|
|
import shutil
|
|
import sqlite3
|
|
import os
|
|
import getpass
|
|
try:
|
|
from iphone_backup_decrypt import EncryptedBackup, RelativePath
|
|
except:
|
|
support_encrypted = False
|
|
else:
|
|
support_encrypted = True
|
|
|
|
def extract_encrypted(base_dir, password):
|
|
backup = EncryptedBackup(backup_directory=base_dir, passphrase=password)
|
|
print("Decrypting WhatsApp database...")
|
|
backup.extract_file(relative_path=RelativePath.WHATSAPP_MESSAGES, output_filename="7c7fba66680ef796b916b067077cc246adacf01d")
|
|
backup.extract_file(relative_path=RelativePath.WHATSAPP_CONTACTS, output_filename="ContactsV2.sqlite")
|
|
data = backup.execute_sql("""SELECT count()
|
|
FROM Files
|
|
WHERE relativePath
|
|
LIKE 'Message/Media/%'"""
|
|
)
|
|
total_row_number = data[0][0]
|
|
print(f"Gathering media...(0/{total_row_number})", end="\r")
|
|
data = backup.execute_sql("""SELECT fileID,
|
|
relativePath,
|
|
flags,
|
|
file
|
|
FROM Files
|
|
WHERE relativePath
|
|
LIKE 'Message/Media/%'"""
|
|
)
|
|
if not os.path.isdir("Message"):
|
|
os.mkdir("Message")
|
|
if not os.path.isdir("Message/Media"):
|
|
os.mkdir("Message/Media")
|
|
i = 0
|
|
for row in data:
|
|
destination = row[1]
|
|
hashes = row[0]
|
|
folder = hashes[:2]
|
|
flags = row[2]
|
|
file = row[3]
|
|
if flags == 2:
|
|
try:
|
|
os.mkdir(destination)
|
|
except FileExistsError:
|
|
pass
|
|
elif flags == 1:
|
|
decrypted = backup.decrypt_inner_file(file_id=hashes, file_bplist=file)
|
|
with open(destination, "wb") as f:
|
|
f.write(decrypted)
|
|
i += 1
|
|
if i % 100 == 0:
|
|
print(f"Gathering media...({i}/{total_row_number})", end="\r")
|
|
print(f"Gathering media...({total_row_number}/{total_row_number})", end="\r")
|
|
|
|
def is_encrypted(base_dir):
|
|
with sqlite3.connect(f"{base_dir}/Manifest.db") as f:
|
|
c = f.cursor()
|
|
try:
|
|
c.execute("""SELECT count()
|
|
FROM Files
|
|
""")
|
|
except sqlite3.DatabaseError:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def extract_media(base_dir):
|
|
if is_encrypted(base_dir):
|
|
if not support_encrypted:
|
|
print("You don't have the dependencies to handle encrypted backup.")
|
|
print("Read more about how to deal with encrypted backup:")
|
|
print("https://github.com/KnugiHK/Whatsapp-Chat-Exporter/blob/main/README.md#encrypted-iphone-backup")
|
|
return False
|
|
password = getpass.getpass("Enter the password:")
|
|
extract_encrypted(base_dir, password)
|
|
else:
|
|
with sqlite3.connect(f"{base_dir}/Manifest.db") as manifest:
|
|
c = manifest.cursor()
|
|
c.execute("""SELECT count()
|
|
FROM Files
|
|
WHERE relativePath
|
|
LIKE 'Message/Media/%'""")
|
|
total_row_number = c.fetchone()[0]
|
|
print(f"Gathering media...(0/{total_row_number})", end="\r")
|
|
c.execute("""SELECT fileID,
|
|
relativePath,
|
|
flags
|
|
FROM Files
|
|
WHERE relativePath
|
|
LIKE 'Message/Media/%'""")
|
|
row = c.fetchone()
|
|
if not os.path.isdir("Message"):
|
|
os.mkdir("Message")
|
|
if not os.path.isdir("Message/Media"):
|
|
os.mkdir("Message/Media")
|
|
i = 0
|
|
while row is not None:
|
|
destination = row[1]
|
|
hashes = row[0]
|
|
folder = hashes[:2]
|
|
flags = row[2]
|
|
if flags == 2:
|
|
os.mkdir(destination)
|
|
elif flags == 1:
|
|
shutil.copyfile(f"{base_dir}/{folder}/{hashes}", destination)
|
|
i += 1
|
|
if i % 100 == 0:
|
|
print(f"Gathering media...({i}/{total_row_number})", end="\r")
|
|
row = c.fetchone()
|
|
print(f"Gathering media...({total_row_number}/{total_row_number})", end="\r")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from optparse import OptionParser
|
|
parser = OptionParser()
|
|
(_, args) = parser.parse_args()
|
|
base_dir = args[0]
|
|
extract_media(base_dir)
|