108 lines
2.7 KiB
Python
108 lines
2.7 KiB
Python
#!/usr/bin/python
|
|
# Copyright 2015 (c) Anna Schumaker.
|
|
|
|
import slack
|
|
import json
|
|
import os
|
|
import sys
|
|
from datetime import datetime
|
|
from urllib import request
|
|
|
|
|
|
# TODO: Move the token file outside of code directory
|
|
TOKEN = ""
|
|
with open("token") as f:
|
|
TOKEN = f.read().strip()
|
|
|
|
|
|
def call_method(method, args = dict()):
|
|
arglist = []
|
|
for (key, value) in args.items():
|
|
arglist += ["%s=%s" % (key, value)]
|
|
argstr = "&&".join(arglist)
|
|
|
|
server = "https://slack.com/api/"
|
|
with request.urlopen("%s/%s?%s" % (server, method, argstr)) as f:
|
|
return json.loads(f.read().decode())
|
|
return {"ok" : False}
|
|
|
|
|
|
def call_method_auth(method, args = dict()):
|
|
args["token"] = TOKEN
|
|
return call_method(method, args)
|
|
|
|
|
|
|
|
#
|
|
# Test connection before doing anything else
|
|
#
|
|
if slack.api.test() == False:
|
|
print("Connection error!")
|
|
sys.exit(1)
|
|
auth = call_method_auth("auth.test")
|
|
if auth["ok"] == False:
|
|
sys.exit(1)
|
|
team = auth["team"]
|
|
|
|
|
|
|
|
#
|
|
# Find list of channels, we'll get info later
|
|
#
|
|
|
|
def list_channel_ids():
|
|
call = call_method_auth("channels.list")
|
|
if call["ok"] == True:
|
|
for channel in call["channels"]:
|
|
yield channel["id"]
|
|
|
|
def find_channel_info(id):
|
|
call = call_method_auth("channels.info", { "channel" : id })
|
|
if call["ok"] == True:
|
|
return call["channel"]
|
|
|
|
def read_channel(id, ts):
|
|
args = { "channel" : id, "oldest" : ts }
|
|
call = call_method_auth("channels.history", args)
|
|
if call["ok"] == True:
|
|
messages = call["messages"]
|
|
messages.reverse()
|
|
for message in messages:
|
|
yield message
|
|
|
|
def _write_message(fout, user, message):
|
|
dt = datetime.fromtimestamp(float(message["ts"]))
|
|
text = message["text"].encode("utf-8").decode("latin-1")
|
|
fout.write("%s %s: %s\n" % (str(dt), user, text))
|
|
|
|
def write_message(fout, message):
|
|
call = call_method_auth("users.info", {"user" : message["user"]})
|
|
if call["ok"] == False:
|
|
name = message["user"]
|
|
else:
|
|
first = call["user"]["profile"].get("first_name", "")
|
|
last = call["user"]["profile"].get("last_name", "")
|
|
name = "%s %s" % (first, last)
|
|
if first == "" and last == "":
|
|
name = call["user"]["name"]
|
|
_write_message(fout, name, message)
|
|
|
|
|
|
tmp_dir = "/tmp/%s" % team
|
|
if not os.path.exists(tmp_dir):
|
|
os.mkdir(tmp_dir)
|
|
|
|
for id in list_channel_ids():
|
|
channel = find_channel_info(id)
|
|
if not channel["is_member"]:
|
|
continue
|
|
if channel["unread_count"] == 0:
|
|
continue
|
|
|
|
fout = open("%s/%s" % (tmp_dir, channel["name"]), 'w')
|
|
fout.write(channel["purpose"]["value"].encode("utf-8").decode("latin-1"))
|
|
fout.write("\n")
|
|
for message in read_channel(id, channel["last_read"]):
|
|
write_message(fout, message)
|
|
fout.close()
|