mirror of
https://github.com/gryf/slack-backup.git
synced 2025-12-17 19:40:21 +01:00
104 lines
2.8 KiB
Python
104 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Reporters module.
|
|
|
|
There are several classes for specific format reporting, and also some of the
|
|
slack conversation/convention parsers.
|
|
"""
|
|
import os
|
|
import logging
|
|
|
|
from slack_backup import objects as o
|
|
from slack_backup import utils
|
|
|
|
|
|
class Reporter(object):
|
|
"""Base reporter class"""
|
|
ext = ''
|
|
|
|
def __init__(self, args, query):
|
|
self.out = args.output
|
|
self.q = query
|
|
|
|
self.channels = self._get_channels(args.channels)
|
|
|
|
def _get_channels(self, selected_channels):
|
|
"""
|
|
Retrieve channels from db and return those which names matched from
|
|
selected_channels list
|
|
"""
|
|
all_channels = self.q(o.Channel).all()
|
|
if not selected_channels:
|
|
return all_channels
|
|
|
|
result = []
|
|
for channel in all_channels:
|
|
if channel.name in selected_channels:
|
|
result.append(channel)
|
|
|
|
return result
|
|
|
|
def generate(self):
|
|
"""Generate raport it's a dummmy one - for use with none reporter"""
|
|
return
|
|
|
|
def get_log_path(self, name):
|
|
"""Return relative log file name """
|
|
return os.path.join(self.out, name + self.ext)
|
|
|
|
def write_msg(self, message, log):
|
|
"""Write message to file"""
|
|
raise NotImplementedError()
|
|
|
|
|
|
class TextReporter(Reporter):
|
|
"""Text aka IRC reporter"""
|
|
ext = '.log'
|
|
|
|
def __init__(self, args, query):
|
|
super().__init__(args, query)
|
|
utils.makedirs(self.out)
|
|
self._line = ""
|
|
|
|
def _prepare_line(self, channel):
|
|
users = [m.user for m in channel.messages]
|
|
users = set([u.name for u in users])
|
|
|
|
return
|
|
|
|
def generate(self):
|
|
"""Generate raport"""
|
|
for channel in self.channels:
|
|
log_path = self.get_log_path(channel.name)
|
|
self._prepare_line(channel)
|
|
for message in self.q(o.Message).\
|
|
filter(o.Message.channel == channel).\
|
|
order_by(o.Message.ts).all():
|
|
self.write_msg(message, log_path)
|
|
|
|
def write_msg(self, message, log):
|
|
"""Write message to file"""
|
|
with open(log, "a") as fobj:
|
|
fobj.write(self._format_message(message))
|
|
|
|
def _format_message(self, msg):
|
|
"""
|
|
Check what kind of message we are dealing with and do appropriate
|
|
formatting
|
|
"""
|
|
return msg.text
|
|
return (msg.datetime().strftime("%Y-%m-%d %H:%M:%S"),
|
|
msg.user.name,
|
|
msg.text)
|
|
|
|
|
|
def get_reporter(args, query):
|
|
"""Return object of right reporter class"""
|
|
reporters = {'text': TextReporter}
|
|
|
|
klass = reporters.get(args.format, Reporter)
|
|
if klass.__name__ == 'Reporter':
|
|
logging.warning('None, or wrong (%s) formatter selected, falling to'
|
|
' None Reporter', args.format)
|
|
return klass(args, query)
|