init snaparser scriptmain

1 files changed, 70 insertions, 0 deletions

diff --git a/snaparser.py b/snaparser.py
new file mode 100644
index 0000000..eead386
--- /dev/null
+++ b/snaparser.py
@@ -0,0 +1,70 @@
+import json
+
+path = ''
+
+chat_history = json.load(open(path + 'chat_history.json'))
+
+received_saved = chat_history['Received Saved Chat History']
+
+sent_saved = chat_history['Sent Saved Chat History']
+
+conversations = {}
+
+def clean_message_in_chat(chat):
+    message = chat['Text']
+    cleaned = message.replace("'", "'")
+    chat['Text'] = cleaned
+    return chat
+
+for received in received_saved:
+    if (received['Media Type'] != 'MEDIA'):
+        chat = received['From']
+        cleaned_chat = clean_message_in_chat(received)
+        if chat in conversations:
+            conversations[chat]['messages'] += [cleaned_chat]
+        else:
+            conversations[chat] = {
+                'messages': [cleaned_chat]
+            }
+
+for sent in sent_saved:
+    if (sent['Media Type'] != 'MEDIA'):
+        chat = sent['To']
+        cleaned_chat = clean_message_in_chat(sent)
+        if chat in conversations:
+            conversations[chat]['messages'] += [cleaned_chat]
+        else:
+            conversations[chat] = {
+                'messages': [cleaned_chat]
+            }
+
+for chat, content in conversations.items():
+    messages = content['messages']
+    msgs_sorted_by_date = sorted(messages, key = lambda x:x['Created'])
+    content['messages'] = msgs_sorted_by_date
+
+for idx, chat in enumerate(conversations):
+    print(f"{idx}: {chat} (Messages: {len(conversations[chat]['messages'])})")
+
+selected_chat_idx = input("Select a chat: ")
+selected_chat = list(conversations.values())[int(selected_chat_idx)]
+
+messages = selected_chat['messages']
+for message in messages:
+    print("\n")
+    message_content = message['Text']
+    time = message['Created']
+    if 'From' in message:
+        recipient = message['From']
+        print(message_content)
+        print(f"From: {recipient}")
+        print(time)
+    else:
+        recipient = message['To']
+        if (len(message_content) >= 150):
+            for submessage in list(message_content[0+i:150+i] for i in range(0, len(message_content), 150)):
+                print(submessage.rjust(150))
+        else:
+            print(message_content.rjust(150))
+        print(f"To: {recipient}".rjust(150))
+        print(time.rjust(150))