aboutsummaryrefslogtreecommitdiff
path: root/main.py
diff options
context:
space:
mode:
Diffstat (limited to 'main.py')
-rw-r--r--main.py41
1 files changed, 41 insertions, 0 deletions
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..4daad96
--- /dev/null
+++ b/main.py
@@ -0,0 +1,41 @@
+"""Slack stuff."""
+import json
+import os
+
+from slack_sdk import WebClient
+
+RESULTS_FILE = "results.json"
+
+
+def get_slack_client():
+ """Initialize the slack client."""
+ token = os.environ["SLACK_TOKEN"]
+ return WebClient(token=token)
+
+
+def write_results(data):
+ """Write the results to a file."""
+ with open(RESULTS_FILE, "w") as f:
+ json.dump(data, f)
+
+
+def load_results():
+ """Load the recent search results."""
+ with open(RESULTS_FILE, "r") as f:
+ return json.load(f)
+
+
+def fetch_results():
+ """Query slack for the latest results."""
+ c = get_slack_client()
+ r = c.search_messages(query="@cody", sort="timestamp")
+
+ # Write the results if
+ if r.data["ok"]:
+ write_results(r.data)
+ print("Search complete.")
+ else:
+ print("Failed to search.")
+
+
+def write_to_database