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

Beautify

Browse Source
This commit is contained in:
KnugiHK
2023-06-10 19:58:14 +08:00
parent c419dd5d39
commit a42ec5d762
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_exported.py
View File

@@ -8,7 +8,8 @@ def messages(path, data, assume_first_as_me=False):
"""Extracts messages from the exported file""" """Extracts messages from the exported file"""
with open(path, "r", encoding="utf8") as file: with open(path, "r", encoding="utf8") as file:
you = "" you = ""
data["chat"] = ChatStore() data["ExportedChat"] = ChatStore()
chat = data["ExportedChat"]
total_row_number = len(file.readlines()) total_row_number = len(file.readlines())
file.seek(0) file.seek(0)
for index, line in enumerate(file): for index, line in enumerate(file):
@@ -22,7 +23,7 @@ def messages(path, data, assume_first_as_me=False):
message = line.split(time)[1].split(name + ":")[1].strip() message = line.split(time)[1].split(name + ":")[1].strip()
name = name[3:] name = name[3:]
if you == "": if you == "":
if data["chat"].name is None: if chat.name is None:
if not assume_first_as_me: if not assume_first_as_me:
while True: while True:
ans = input(f"Is '{name}' you? (Y/N)").lower() ans = input(f"Is '{name}' you? (Y/N)").lower()
@@ -30,16 +31,16 @@ def messages(path, data, assume_first_as_me=False):
you = name you = name
break break
elif ans == "n": elif ans == "n":
data["chat"].name = name chat.name = name
break break
else: else:
you = name you = name
else: else:
if name != data["chat"].name: if name != chat.name:
you = name you = name
if data["chat"].name is None and you != "": elif chat.name is None:
if name != you: if name != you:
data["chat"].name = name chat.name = name
msg = Message( msg = Message(
you == name, you == name,
datetime.strptime(time, "%d/%m/%Y, %H:%M").timestamp(), datetime.strptime(time, "%d/%m/%Y, %H:%M").timestamp(),
@@ -71,12 +72,12 @@ def messages(path, data, assume_first_as_me=False):
msg.data = message.replace("\r\n", "<br>") msg.data = message.replace("\r\n", "<br>")
if "\n" in message: if "\n" in message:
msg.data = message.replace("\n", "<br>") msg.data = message.replace("\n", "<br>")
data["chat"].add_message(index, msg) chat.add_message(index, msg)
else: else:
lookback = index - 1 lookback = index - 1
while lookback not in data["chat"].messages: while lookback not in chat.messages:
lookback -= 1 lookback -= 1
msg = data["chat"].messages[lookback] msg = chat.messages[lookback]
if msg.media: if msg.media:
msg.caption = line.strip() msg.caption = line.strip()
else: else: