calvin.erfmann/WhatsApp-Chat-Exporter
1
0
Fork 0
mirror of https://github.com/KnugiHK/WhatsApp-Chat-Exporter.git synced 2026-03-16 11:02:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Extract whole WhatsApp directory instead of wanted files only

Browse Source
This commit is contained in:
KnugiHK
2023-06-13 16:29:27 +08:00
parent 200dea218f
commit 8294f06587
1 changed files with 29 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
Whatsapp_Chat_Exporter/extract_iphone_media.py
View File

@@ -84,34 +84,43 @@ def extract_media(base_dir):
extract_encrypted(base_dir, password) extract_encrypted(base_dir, password)
else: else:
wts_db = os.path.join(base_dir, "7c/7c7fba66680ef796b916b067077cc246adacf01d") wts_db = os.path.join(base_dir, "7c/7c7fba66680ef796b916b067077cc246adacf01d")
contact_db = os.path.join(base_dir, "b8/b8548dc30aa1030df0ce18ef08b882cf7ab5212f")
if not os.path.isfile(wts_db): if not os.path.isfile(wts_db):
print("WhatsApp database not found.") print("WhatsApp database not found.")
exit() exit()
else: else:
shutil.copyfile(wts_db, "7c7fba66680ef796b916b067077cc246adacf01d") shutil.copyfile(wts_db, "7c7fba66680ef796b916b067077cc246adacf01d")
with sqlite3.connect(f"{base_dir}/Manifest.db") as manifest: if not os.path.isfile(contact_db):
print("Contact database not found.")
exit()
else:
shutil.copyfile(contact_db, "b8548dc30aa1030df0ce18ef08b882cf7ab5212f")
_wts_id = "AppDomainGroup-group.net.whatsapp.WhatsApp.shared"
with sqlite3.connect(os.path.join(base_dir, "Manifest.db")) as manifest:
manifest.row_factory = sqlite3.Row manifest.row_factory = sqlite3.Row
c = manifest.cursor() c = manifest.cursor()
c.execute("""SELECT count() c.execute(
FROM Files f"""SELECT count()
WHERE relativePath FROM Files
LIKE 'Message/Media/%'""") WHERE domain = '{_wts_id}'"""
)
total_row_number = c.fetchone()[0] total_row_number = c.fetchone()[0]
print(f"Extracting media...(0/{total_row_number})", end="\r") print(f"Extracting WhatsApp files...(0/{total_row_number})", end="\r")
c.execute("""SELECT fileID, c.execute(f"""SELECT fileID,
relativePath, relativePath,
flags flags,
ROW_NUMBER() OVER(ORDER BY relativePath) AS _index
FROM Files FROM Files
WHERE relativePath WHERE domain = '{_wts_id}'
LIKE 'Message/Media/%'""") ORDER BY relativePath""")
if not os.path.isdir("WhatsApp"):
os.mkdir("WhatsApp")
row = c.fetchone() 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: while row is not None:
destination = row["relativePath"] if row["relativePath"] == "":
row = c.fetchone()
continue
destination = os.path.join("WhatsApp/", row["relativePath"])
hashes = row["fileID"] hashes = row["fileID"]
folder = hashes[:2] folder = hashes[:2]
flags = row["flags"] flags = row["flags"]
@@ -121,12 +130,11 @@ def extract_media(base_dir):
except FileExistsError: except FileExistsError:
pass pass
elif flags == 1: elif flags == 1:
shutil.copyfile(f"{base_dir}/{folder}/{hashes}", destination) shutil.copyfile(os.path.join(base_dir, folder, hashes), destination)
i += 1 if row["_index"] % 100 == 0:
if i % 100 == 0: print(f"Extracting WhatsApp files...({row['_index']}/{total_row_number})", end="\r")
print(f"Extracting media...({i}/{total_row_number})", end="\r")
row = c.fetchone() row = c.fetchone()
print(f"Extracting media...({total_row_number}/{total_row_number})", end="\n") print(f"Extracting WhatsApp files...({total_row_number}/{total_row_number})", end="\n")
if __name__ == "__main__": if __name__ == "__main__":