fix the logging

1 files changed, 15 insertions, 11 deletions

diff --git a/main.py b/main.py
index e0ebcb9..f5fd103 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,6 @@
 """Slack stuff."""
-from datetime import datetime
 import os
+from datetime import datetime
 
 from mappings import Message, get_session
 from slack_sdk import WebClient
@@ -21,7 +21,7 @@ def fetch_results():
 
     # Write the results if
     if r.data["ok"]:
-        messages = r.data['messages']['matches']
+        messages = r.data["messages"]["matches"]
         write_to_database(messages)
         print("Archival Complete.")
     else:
@@ -32,16 +32,20 @@ def write_to_database(messages):
     """Write messages to database."""
     session = get_session()
     for msg in messages:
-        iid = msg["iid"]
-        if session.query(Message).filter(Message.iid == iid).count() != 0:
+        username = msg["username"]
+        timestamp = datetime.fromtimestamp(int(msg["ts"].split(".")[0]))
+        if session.query(Message).filter(Message.username == username).filter(Message.timestamp == timestamp).count() != 0:
             continue
-        session.add(Message(
-            iid=iid,
-            type=msg["type"],
-            username=msg["username"],
-            text=msg["text"],
-            timestamp=datetime.fromtimestamp(int(msg["ts"].split('.')[0]))
-        ))
+        print("New notification")
+        session.add(
+            Message(
+                iid=msg["iid"],
+                type=msg["type"],
+                username=username,
+                text=msg["text"],
+                timestamp=timestamp,
+            )
+        )
     session.commit()