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

Rewrite a bit

Browse Source
This commit is contained in:
KnugiHK
2023-06-11 01:22:21 +08:00
parent ed658d78dc
commit 2fa5c4268e
1 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
Whatsapp_Chat_Exporter/extract_iphone_media.py
View File

@@ -25,7 +25,7 @@ def extract_encrypted(base_dir, password):
LIKE 'Message/Media/%'""" LIKE 'Message/Media/%'"""
) )
total_row_number = data[0][0] total_row_number = data[0][0]
print(f"Gathering media...(0/{total_row_number})", end="\r") print(f"Extracting media...(0/{total_row_number})", end="\r")
data = backup.execute_sql("""SELECT fileID, data = backup.execute_sql("""SELECT fileID,
relativePath, relativePath,
flags, flags,
@@ -56,8 +56,8 @@ def extract_encrypted(base_dir, password):
f.write(decrypted) f.write(decrypted)
i += 1 i += 1
if i % 100 == 0: if i % 100 == 0:
print(f"Gathering media...({i}/{total_row_number})", end="\r") print(f"Extracting media...({i}/{total_row_number})", end="\r")
print(f"Gathering media...({total_row_number}/{total_row_number})", end="\r") print(f"Extracting media...({total_row_number}/{total_row_number})", end="\n")
def is_encrypted(base_dir): def is_encrypted(base_dir):
@@ -90,13 +90,14 @@ def extract_media(base_dir):
else: else:
shutil.copyfile(wts_db, "7c7fba66680ef796b916b067077cc246adacf01d") shutil.copyfile(wts_db, "7c7fba66680ef796b916b067077cc246adacf01d")
with sqlite3.connect(f"{base_dir}/Manifest.db") as manifest: with sqlite3.connect(f"{base_dir}/Manifest.db") as manifest:
manifest.row_factory = sqlite3.Row
c = manifest.cursor() c = manifest.cursor()
c.execute("""SELECT count() c.execute("""SELECT count()
FROM Files FROM Files
WHERE relativePath WHERE relativePath
LIKE 'Message/Media/%'""") LIKE 'Message/Media/%'""")
total_row_number = c.fetchone()[0] total_row_number = c.fetchone()[0]
print(f"Gathering media...(0/{total_row_number})", end="\r") print(f"Extracting media...(0/{total_row_number})", end="\r")
c.execute("""SELECT fileID, c.execute("""SELECT fileID,
relativePath, relativePath,
flags flags
@@ -110,10 +111,10 @@ def extract_media(base_dir):
os.mkdir("Message/Media") os.mkdir("Message/Media")
i = 0 i = 0
while row is not None: while row is not None:
destination = row[1] destination = row["relativePath"]
hashes = row[0] hashes = row["fileID"]
folder = hashes[:2] folder = hashes[:2]
flags = row[2] flags = row["flags"]
if flags == 2: if flags == 2:
try: try:
os.mkdir(destination) os.mkdir(destination)
@@ -123,9 +124,9 @@ def extract_media(base_dir):
shutil.copyfile(f"{base_dir}/{folder}/{hashes}", destination) shutil.copyfile(f"{base_dir}/{folder}/{hashes}", destination)
i += 1 i += 1
if i % 100 == 0: if i % 100 == 0:
print(f"Gathering media...({i}/{total_row_number})", end="\r") print(f"Extracting media...({i}/{total_row_number})", end="\r")
row = c.fetchone() row = c.fetchone()
print(f"Gathering media...({total_row_number}/{total_row_number})", end="\r") print(f"Extracting media...({total_row_number}/{total_row_number})", end="\n")
if __name__ == "__main__": if __name__ == "__main__":