mirror of
https://github.com/gryf/slack-backup.git
synced 2025-12-17 11:30:25 +01:00
Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 9ddd470b54 | |||
| feb773956c | |||
| 3f95986981 |
@@ -1,122 +1,10 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
"""
|
"""
|
||||||
Create backup for certain date for specified channel in slack
|
Execute commands for slack-backup
|
||||||
"""
|
"""
|
||||||
import argparse
|
from slack_backup import command
|
||||||
import logging
|
|
||||||
|
|
||||||
from slack_backup import client
|
|
||||||
from slack_backup import config
|
|
||||||
|
|
||||||
|
|
||||||
def setup_logger(args):
|
|
||||||
"""Setup logger format and level"""
|
|
||||||
|
|
||||||
level = logging.WARNING
|
|
||||||
|
|
||||||
if args.quiet:
|
|
||||||
level = logging.ERROR
|
|
||||||
if args.quiet > 1:
|
|
||||||
level = logging.CRITICAL
|
|
||||||
|
|
||||||
if args.verbose:
|
|
||||||
level = logging.INFO
|
|
||||||
if args.verbose > 1:
|
|
||||||
level = logging.DEBUG
|
|
||||||
|
|
||||||
logging.basicConfig(level=level,
|
|
||||||
format="%(asctime)s %(levelname)s: %(message)s")
|
|
||||||
|
|
||||||
def generate_raport(args):
|
|
||||||
"""Generate logs"""
|
|
||||||
slack = client.Client(args)
|
|
||||||
slack.generate_history()
|
|
||||||
|
|
||||||
def fetch_data(args):
|
|
||||||
"""Fetch and store data"""
|
|
||||||
slack = client.Client(args)
|
|
||||||
slack.update()
|
|
||||||
|
|
||||||
|
|
||||||
def main():
|
|
||||||
"""Main function"""
|
|
||||||
parser = argparse.ArgumentParser()
|
|
||||||
|
|
||||||
subparser = parser.add_subparsers(dest='parser')
|
|
||||||
subparser.required = True
|
|
||||||
|
|
||||||
fetch = subparser.add_parser('fetch', help='Update local db with Slack'
|
|
||||||
' data')
|
|
||||||
fetch.add_argument('-t', '--token', default=None, help='Slack token - '
|
|
||||||
'a string, which can be generated/obtained via '
|
|
||||||
'https://api.slack.com/docs/oauth-test-tokens page.')
|
|
||||||
fetch.add_argument('-u', '--user', default=None, help='Username for your '
|
|
||||||
'Slack account')
|
|
||||||
fetch.add_argument('-p', '--password', default=None, help='Password for '
|
|
||||||
'your Slack account.')
|
|
||||||
fetch.add_argument('-e', '--team', default=None, help='Team name, which '
|
|
||||||
'is part of slack url, for example: if url is '
|
|
||||||
'"https://team.slack.com" than "team" is a name of '
|
|
||||||
'the team.')
|
|
||||||
fetch.add_argument('-v', '--verbose', help='Be verbose. Adding more "v" '
|
|
||||||
'will increase verbosity', action="count",
|
|
||||||
default=None)
|
|
||||||
fetch.add_argument('-q', '--quiet', help='Be quiet. Adding more "q" will'
|
|
||||||
' decrease verbosity', action="count", default=None)
|
|
||||||
fetch.add_argument('-c', '--channels', default=None, nargs='+',
|
|
||||||
help='List of channels to perform actions on. '
|
|
||||||
'Default is all channels.')
|
|
||||||
|
|
||||||
fetch.add_argument('-d', '--database', default=None,
|
|
||||||
help='Path to the database file.')
|
|
||||||
|
|
||||||
fetch.add_argument('-i', '--config', default=None,
|
|
||||||
help='Use specific config file.')
|
|
||||||
|
|
||||||
fetch.set_defaults(func=fetch_data)
|
|
||||||
|
|
||||||
generate = subparser.add_parser('generate', help='Generate logs out of '
|
|
||||||
'data in provided database')
|
|
||||||
generate.add_argument('-o', '--output', default=None, help="Output "
|
|
||||||
"directory for store logs. All logs are organised "
|
|
||||||
"per channel. By default it's `logs' directory")
|
|
||||||
generate.add_argument('-f', '--format', default=None,
|
|
||||||
choices=('text', 'none'),
|
|
||||||
help='Output format. Default is none; only database '
|
|
||||||
'is updated by latest messages for all/selected '
|
|
||||||
'channels.')
|
|
||||||
generate.add_argument('-t', '--theme', default=None,
|
|
||||||
choices=('plain', 'unicode'),
|
|
||||||
help='Choose theme for text output. It doesn\'t '
|
|
||||||
'affect other output formats.')
|
|
||||||
|
|
||||||
generate.add_argument('-v', '--verbose', help='Be verbose. Adding more '
|
|
||||||
'"v" will increase verbosity', action="count",
|
|
||||||
default=None)
|
|
||||||
generate.add_argument('-q', '--quiet', help='Be quiet. Adding more "q" '
|
|
||||||
'will decrease verbosity', action="count",
|
|
||||||
default=None)
|
|
||||||
generate.add_argument('-c', '--channels', default=[], nargs='+',
|
|
||||||
help='List of channels to perform actions on. '
|
|
||||||
'Default is all channels.')
|
|
||||||
|
|
||||||
generate.add_argument('-d', '--database', default=None,
|
|
||||||
help='Path to the database file.')
|
|
||||||
|
|
||||||
generate.add_argument('-i', '--config', default=None,
|
|
||||||
help='Use specific config file.')
|
|
||||||
|
|
||||||
generate.set_defaults(func=generate_raport)
|
|
||||||
|
|
||||||
args = parser.parse_args()
|
|
||||||
cfg = config.Config()
|
|
||||||
msg = cfg.update(args)
|
|
||||||
setup_logger(args)
|
|
||||||
logging.info(msg)
|
|
||||||
|
|
||||||
args.func(args)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
command.main()
|
||||||
|
|||||||
2
setup.py
2
setup.py
@@ -10,7 +10,7 @@ except ImportError:
|
|||||||
|
|
||||||
setup(name="slack-backup",
|
setup(name="slack-backup",
|
||||||
packages=["slack_backup"],
|
packages=["slack_backup"],
|
||||||
version="0.4.1",
|
version="0.4.4",
|
||||||
description="Make copy of slack converstaions",
|
description="Make copy of slack converstaions",
|
||||||
author="Roman Dobosz",
|
author="Roman Dobosz",
|
||||||
author_email="gryf73@gmail.com",
|
author_email="gryf73@gmail.com",
|
||||||
|
|||||||
118
slack_backup/command.py
Normal file
118
slack_backup/command.py
Normal file
@@ -0,0 +1,118 @@
|
|||||||
|
"""
|
||||||
|
Create backup for certain date for specified channel in slack
|
||||||
|
"""
|
||||||
|
import argparse
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from slack_backup import client
|
||||||
|
from slack_backup import config
|
||||||
|
|
||||||
|
|
||||||
|
def setup_logger(args):
|
||||||
|
"""Setup logger format and level"""
|
||||||
|
|
||||||
|
level = logging.WARNING
|
||||||
|
|
||||||
|
if args.quiet:
|
||||||
|
level = logging.ERROR
|
||||||
|
if args.quiet > 1:
|
||||||
|
level = logging.CRITICAL
|
||||||
|
|
||||||
|
if args.verbose:
|
||||||
|
level = logging.INFO
|
||||||
|
if args.verbose > 1:
|
||||||
|
level = logging.DEBUG
|
||||||
|
|
||||||
|
logging.basicConfig(level=level,
|
||||||
|
format="%(asctime)s %(levelname)s: %(message)s")
|
||||||
|
|
||||||
|
|
||||||
|
def generate_raport(args):
|
||||||
|
"""Generate logs"""
|
||||||
|
slack = client.Client(args)
|
||||||
|
slack.generate_history()
|
||||||
|
|
||||||
|
|
||||||
|
def fetch_data(args):
|
||||||
|
"""Fetch and store data"""
|
||||||
|
slack = client.Client(args)
|
||||||
|
slack.update()
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
"""Main function"""
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
|
||||||
|
subparser = parser.add_subparsers(dest='parser')
|
||||||
|
subparser.required = True
|
||||||
|
|
||||||
|
fetch = subparser.add_parser('fetch', help='Update local db with Slack'
|
||||||
|
' data')
|
||||||
|
fetch.add_argument('-t', '--token', default=None, help='Slack token - '
|
||||||
|
'a string, which can be generated/obtained via '
|
||||||
|
'https://api.slack.com/docs/oauth-test-tokens page.')
|
||||||
|
fetch.add_argument('-u', '--user', default=None, help='Username for your '
|
||||||
|
'Slack account')
|
||||||
|
fetch.add_argument('-p', '--password', default=None, help='Password for '
|
||||||
|
'your Slack account.')
|
||||||
|
fetch.add_argument('-e', '--team', default=None, help='Team name, which '
|
||||||
|
'is part of slack url, for example: if url is '
|
||||||
|
'"https://team.slack.com" than "team" is a name of '
|
||||||
|
'the team.')
|
||||||
|
fetch.add_argument('-v', '--verbose', help='Be verbose. Adding more "v" '
|
||||||
|
'will increase verbosity', action="count",
|
||||||
|
default=None)
|
||||||
|
fetch.add_argument('-q', '--quiet', help='Be quiet. Adding more "q" will'
|
||||||
|
' decrease verbosity', action="count", default=None)
|
||||||
|
fetch.add_argument('-c', '--channels', default=None, nargs='+',
|
||||||
|
help='List of channels to perform actions on. '
|
||||||
|
'Default is all channels.')
|
||||||
|
|
||||||
|
fetch.add_argument('-d', '--database', default=None,
|
||||||
|
help='Path to the database file.')
|
||||||
|
|
||||||
|
fetch.add_argument('-i', '--config', default=None,
|
||||||
|
help='Use specific config file.')
|
||||||
|
|
||||||
|
fetch.set_defaults(func=fetch_data)
|
||||||
|
|
||||||
|
generate = subparser.add_parser('generate', help='Generate logs out of '
|
||||||
|
'data in provided database')
|
||||||
|
generate.add_argument('-o', '--output', default=None, help="Output "
|
||||||
|
"directory for store logs. All logs are organised "
|
||||||
|
"per channel. By default it's `logs' directory")
|
||||||
|
generate.add_argument('-f', '--format', default=None,
|
||||||
|
choices=('text', 'none'),
|
||||||
|
help='Output format. Default is none; only database '
|
||||||
|
'is updated by latest messages for all/selected '
|
||||||
|
'channels.')
|
||||||
|
generate.add_argument('-t', '--theme', default=None,
|
||||||
|
choices=('plain', 'unicode'),
|
||||||
|
help='Choose theme for text output. It doesn\'t '
|
||||||
|
'affect other output formats.')
|
||||||
|
|
||||||
|
generate.add_argument('-v', '--verbose', help='Be verbose. Adding more '
|
||||||
|
'"v" will increase verbosity', action="count",
|
||||||
|
default=None)
|
||||||
|
generate.add_argument('-q', '--quiet', help='Be quiet. Adding more "q" '
|
||||||
|
'will decrease verbosity', action="count",
|
||||||
|
default=None)
|
||||||
|
generate.add_argument('-c', '--channels', default=[], nargs='+',
|
||||||
|
help='List of channels to perform actions on. '
|
||||||
|
'Default is all channels.')
|
||||||
|
|
||||||
|
generate.add_argument('-d', '--database', default=None,
|
||||||
|
help='Path to the database file.')
|
||||||
|
|
||||||
|
generate.add_argument('-i', '--config', default=None,
|
||||||
|
help='Use specific config file.')
|
||||||
|
|
||||||
|
generate.set_defaults(func=generate_raport)
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
cfg = config.Config()
|
||||||
|
msg = cfg.update(args)
|
||||||
|
setup_logger(args)
|
||||||
|
logging.info(msg)
|
||||||
|
|
||||||
|
args.func(args)
|
||||||
@@ -58,7 +58,7 @@ class Config(object):
|
|||||||
path = args.config
|
path = args.config
|
||||||
|
|
||||||
locations = [path,
|
locations = [path,
|
||||||
'./slack-backup.conf',
|
'./slack-backup.ini',
|
||||||
os.path.expandvars('$XDG_CONFIG_HOME/slack-backup.ini'),
|
os.path.expandvars('$XDG_CONFIG_HOME/slack-backup.ini'),
|
||||||
os.path.expandvars('$HOME/.config/slack-backup.ini')]
|
os.path.expandvars('$HOME/.config/slack-backup.ini')]
|
||||||
|
|
||||||
@@ -83,19 +83,25 @@ class Config(object):
|
|||||||
val = self.cp.get(section, option, fallback='[]')
|
val = self.cp.get(section, option, fallback='[]')
|
||||||
val = json.loads(val)
|
val = json.loads(val)
|
||||||
else:
|
else:
|
||||||
val = self.cp.get(section, option, fallback='')
|
val = self.cp.get(section, option, fallback=None)
|
||||||
|
|
||||||
self._options[option] = val
|
self._options[option] = val
|
||||||
|
|
||||||
def update_args(self, args):
|
def update_args(self, args):
|
||||||
|
if 'parser' not in args:
|
||||||
|
# it doesn't make sense to update args, since no action was
|
||||||
|
# choosen
|
||||||
|
return
|
||||||
|
|
||||||
# special case, re-set information for verbose/quiet options
|
# special case, re-set information for verbose/quiet options
|
||||||
if args.verbose is not None and self._options['quiet'] is not None:
|
if 'verbose' in args and args.verbose is not None:
|
||||||
self._options['quiet'] = 0
|
|
||||||
self._options['verbose'] = args.verbose
|
self._options['verbose'] = args.verbose
|
||||||
if args.quiet is not None and self._options['verbose'] is not None:
|
if self._options['quiet'] is not None:
|
||||||
self._options['verbose'] = 0
|
self._options['quiet'] = 0
|
||||||
|
if 'quiet' in args and args.quiet is not None:
|
||||||
self._options['quiet'] = args.quiet
|
self._options['quiet'] = args.quiet
|
||||||
|
if self._options['verbose'] is not None:
|
||||||
|
self._options['verbose'] = 0
|
||||||
|
|
||||||
for sec_id in (args.parser, 'common'):
|
for sec_id in (args.parser, 'common'):
|
||||||
for option in self.sections[sec_id]:
|
for option in self.sections[sec_id]:
|
||||||
|
|||||||
@@ -55,8 +55,8 @@ class Download(object):
|
|||||||
'file': self._files}
|
'file': self._files}
|
||||||
|
|
||||||
if filetype == 'file' and not self._authorized:
|
if filetype == 'file' and not self._authorized:
|
||||||
logging.info("There was no (valid) credentials passed, therefore "
|
logging.warning("There was no (valid) credentials passed, "
|
||||||
"file `%s' cannot be downloaded", url)
|
"therefore file `%s' cannot be downloaded", url)
|
||||||
return
|
return
|
||||||
|
|
||||||
splitted = url.split('/')
|
splitted = url.split('/')
|
||||||
|
|||||||
@@ -51,13 +51,6 @@ class TestConfig(unittest.TestCase):
|
|||||||
args.config = None
|
args.config = None
|
||||||
args.parser = 'fetch'
|
args.parser = 'fetch'
|
||||||
args.verbose = 2
|
args.verbose = 2
|
||||||
args.quiet = None
|
|
||||||
args.channels = None
|
|
||||||
args.database = None
|
|
||||||
args.user = None
|
|
||||||
args.password = None
|
|
||||||
args.team = None
|
|
||||||
args.token = None
|
|
||||||
|
|
||||||
conf = config.Config()
|
conf = config.Config()
|
||||||
conf.update(args)
|
conf.update(args)
|
||||||
@@ -66,7 +59,7 @@ class TestConfig(unittest.TestCase):
|
|||||||
'verbose': 2,
|
'verbose': 2,
|
||||||
'quiet': 0,
|
'quiet': 0,
|
||||||
'channels': [],
|
'channels': [],
|
||||||
'database': '',
|
'database': None,
|
||||||
'user': None,
|
'user': None,
|
||||||
'password': None,
|
'password': None,
|
||||||
'team': None,
|
'team': None,
|
||||||
|
|||||||
Reference in New Issue
Block a user