Add printer script for cleaner formatting

2 files changed, 35 insertions, 7 deletions

diff --git a/main.py b/main.py
index f5fd103..6e29c47 100644
--- a/main.py
+++ b/main.py
@@ -14,14 +14,20 @@ def get_slack_client():
     return WebClient(token=token)
 
 
-def fetch_results():
-    """Query slack for the latest results."""
+def get_messages():
+    """Get messages."""
     c = get_slack_client()
     r = c.search_messages(query="@cody", sort="timestamp")
-
-    # Write the results if
     if r.data["ok"]:
-        messages = r.data["messages"]["matches"]
+        return r.data["messages"]["matches"]
+    else:
+        return []
+
+
+def fetch_results():
+    """Query slack for the latest results."""
+    messages = get_messages()
+    if messages:
         write_to_database(messages)
         print("Archival Complete.")
     else:
@@ -34,7 +40,13 @@ def write_to_database(messages):
     for msg in messages:
         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:
+        if (
+            session.query(Message)
+            .filter(Message.username == username)
+            .filter(Message.timestamp == timestamp)
+            .count()
+            != 0
+        ):
             continue
         print("New notification")
         session.add(
@@ -49,4 +61,5 @@ def write_to_database(messages):
     session.commit()
 
 
-fetch_results()
+if __name__ == "__main__":
+    fetch_results()
diff --git a/printer.py b/printer.py
new file mode 100644
index 0000000..6e13f2c
--- /dev/null
+++ b/printer.py
@@ -0,0 +1,15 @@
+"""Print the results."""
+from datetime import timedelta
+
+from mappings import Message, get_session
+
+session = get_session()
+msgs = session.query(Message).order_by(Message.timestamp.desc()).all()
+
+my_timezone = timedelta(hours=7)
+
+for msg in msgs:
+    sent_at = msg.timestamp - my_timezone
+    sent_at_str = sent_at.strftime("[%m-%d %H:%M]")
+    text = msg.text[:70].replace("\n", "")
+    print(f"{sent_at_str} {msg.username:^10} {msg.type:^8} {text}")