Add files via upload

Author: Linuxndroid

config.cfg

@@ -0,0 +1,17 @@
+[input]
+# Path to the file msgstore.db
+msgstore_path=C:\Users\india\Desktop\WhatsApp-Viewer-main\WhatsApp-Viewer-main\msgstore.db
+# Use external contacts database wa.db?
+use_wa_db = True
+# Path to the file wa.db
+wa_path=C:\Users\india\Desktop\WhatsApp-Viewer-main\WhatsApp-Viewer-main\wa.db
+
+[output]
+# Create an HTML export?
+export_html=True
+# Path to the HTML file to export
+html_output_path=C:\Users\india\Desktop\WhatsApp-Viewer-main\WhatsApp-Viewer-main\index.html
+# Create an export to txt files?
+export_txt=False
+# Path to the directory for exporting txt files
+txt_output_directory_path=/root/

exporter.py

@@ -0,0 +1,77 @@
+import html
+from string import Template
+
+from models import Message
+
+
+def chats_to_txt(chats: list, directory_path: str):
+    for chat in chats:
+        messages = "\n".join([str(message) for message in chat.messages])
+        with open(f"{directory_path}/{chat.key_remote_jid}.txt", "w", encoding="utf-8") as file:
+            file.write(chat.title + "\n" + messages)
+
+
+def chats_to_html(chats: list, filepath: str):
+    chat_contents = ""
+    chats_list = ""
+    for chat in chats:
+        # Hide empty chats
+        if len(chat.messages) == 0:
+            continue
+
+        # Add the chat contents to the HTML
+        t = Template("""<div class="chat" data-chatid="$chat_id">$messages</div>""")
+        chat_contents += t.substitute(
+            chat_id=_esc(chat.key_remote_jid),
+            messages="".join([_message_to_html(m) for m in chat.messages])
+        )
+
+        # Add a row to the list of chats
+        t = Template("""<div class="chat-partner"><a href="#$id" title="$phone_number">$title<div>$preview</div></a></div>""")
+        last_message = chat.messages[-1].get_content()
+        preview = last_message[0:55] if last_message is not None else ""
+        chats_list += t.substitute(
+            id=_esc(chat.key_remote_jid),
+            phone_number=_esc(chat.phone_number),
+            title=_esc(chat.title),
+            preview=_esc(preview)
+        )
+
+    _save_to_html_file(chat_contents, chats_list, filepath)
+
+
+def _message_to_html(m: Message) -> str:
+    direction_class = " sent" if m.key_from_me else ""
+    sender = _esc(m.get_sender_name())
+    content = _esc(m.get_content())
+    time = _esc(m.received_timestamp_str)
+
+    if m.remote_resource:
+        t = Template("""<div class="message$direction_class"><div class="sender">$sender</div>$content<div class="time">$time</div></div>""")
+        return t.substitute(direction_class=direction_class, sender=sender, content=content, time=time)
+    else:
+        t = Template("""<div class="message$direction_class">$content<div class="time">$time</div></div>""")
+        return t.substitute(direction_class=direction_class, content=content, time=time)
+
+
+def _esc(content) -> str:
+    return html.escape(str(content))
+
+
+def _load_file_content(filepath: str) -> str:
+    with open(filepath, "r") as file:
+        return file.read()
+
+
+def _save_to_html_file(chat_contents: str, chats_list: str, filepath: str):
+    # Load the HTML template from file and populate it with data
+    t = Template(_load_file_content("resources/template.html"))
+    html_output = t.substitute(
+        js_code=_load_file_content("resources/main.js"),
+        css_code=_load_file_content("resources/styles.css"),
+        chats_list=chats_list,
+        chat_contents=chat_contents
+    )
+
+    with open(filepath, "w", encoding="utf-8") as file:
+        file.write(html_output)