Pass channel_list to save_replies

This commit is contained in:
Seb Seager 2022-02-15 20:44:21 -05:00
parent b26f4d454a
commit bbb34418fc

View file

@ -444,14 +444,13 @@ if __name__ == "__main__":
else: else:
f.write(data) f.write(data)
def save_replies(channel_hist, channel_id, users): def save_replies(channel_hist, channel_id, channel_list, users):
ch_replies = channel_replies( reply_timestamps = [x["ts"] for x in channel_hist if "reply_count" in x]
[x["ts"] for x in channel_hist if "reply_count" in x], channel_id ch_replies = channel_replies(reply_timestamps, channel_id)
)
if a.json: if a.json:
data_replies = ch_replies data_replies = ch_replies
else: else:
ch_name, ch_type = name_from_ch_id(ch_id, ch_list) ch_name, ch_type = name_from_ch_id(channel_id, channel_list)
header_str = "Threads in %s: %s\n%s Messages" % ( header_str = "Threads in %s: %s\n%s Messages" % (
ch_type, ch_type,
ch_name, ch_name,
@ -461,37 +460,35 @@ if __name__ == "__main__":
data_replies = "%s\n%s\n\n%s" % (header_str, sep_str, data_replies) data_replies = "%s\n%s\n\n%s" % (header_str, sep_str, data_replies)
save(data_replies, "channel-replies_%s" % channel_id) save(data_replies, "channel-replies_%s" % channel_id)
def save_channel(ch_id, ch_list, users): def save_channel(channel_id, channel_list, users):
ch_hist = channel_history(ch_id, oldest=a.fr, latest=a.to) ch_hist = channel_history(channel_id, oldest=a.fr, latest=a.to)
if a.json: if a.json:
data_ch = ch_hist data_ch = ch_hist
else: else:
data_ch = parse_channel_history(ch_hist, users) data_ch = parse_channel_history(ch_hist, users)
ch_name, ch_type = name_from_ch_id(ch_id, ch_list) ch_name, ch_type = name_from_ch_id(channel_id, channel_list)
header_str = "%s Name: %s" % (ch_type, ch_name) header_str = "%s Name: %s" % (ch_type, ch_name)
data_ch = ( data_ch = (
"Channel ID: %s\n%s\n%s Messages\n%s\n\n" "Channel ID: %s\n%s\n%s Messages\n%s\n\n"
% (ch_id, header_str, len(ch_hist), sep_str) % (channel_id, header_str, len(ch_hist), sep_str)
+ data_ch + data_ch
) )
save(data_ch, "channel_%s" % ch_id) save(data_ch, "channel_%s" % channel_id)
if a.r: if a.r:
save_replies(ch_hist, ch_id, users) save_replies(ch_hist, channel_id, channel_list, users)
ch_list = channel_list()
user_list = user_list()
if a.lc: if a.lc:
data = ( data = ch_list if a.json else parse_channel_list(ch_list, user_list)
channel_list()
if a.json
else parse_channel_list(channel_list(), user_list())
)
save(data, "channel_list") save(data, "channel_list")
if a.lu: if a.lu:
data = user_list() if a.json else parse_user_list(user_list()) data = user_list if a.json else parse_user_list(user_list)
save(data, "user_list") save(data, "user_list")
if a.c: if a.c:
ch_id = a.ch ch_id = a.ch
ch_list = channel_list() users = user_list
users = user_list()
if ch_id: if ch_id:
save_channel(ch_id, ch_list, users) save_channel(ch_id, ch_list, users)
else: else:
@ -500,4 +497,4 @@ if __name__ == "__main__":
# elif, since we want to avoid asking for channel_history twice # elif, since we want to avoid asking for channel_history twice
elif a.r: elif a.r:
for ch_id in [x["id"] for x in channel_list()]: for ch_id in [x["id"] for x in channel_list()]:
save_replies(channel_history(ch_id), ch_id, user_list()) save_replies(channel_history(ch_id), ch_id, ch_list, user_list)