mirror of
https://github.com/gryf/slack-backup.git
synced 2025-12-17 11:30:25 +01:00
Compare commits
4 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 9ddd470b54 | |||
| feb773956c | |||
| 3f95986981 | |||
| 6d5f3746a2 |
@@ -1,122 +1,10 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Create backup for certain date for specified channel in slack
|
||||
Execute commands for slack-backup
|
||||
"""
|
||||
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)
|
||||
from slack_backup import command
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
command.main()
|
||||
|
||||
2
setup.py
2
setup.py
@@ -10,7 +10,7 @@ except ImportError:
|
||||
|
||||
setup(name="slack-backup",
|
||||
packages=["slack_backup"],
|
||||
version="0.4",
|
||||
version="0.4.4",
|
||||
description="Make copy of slack converstaions",
|
||||
author="Roman Dobosz",
|
||||
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)
|
||||
@@ -53,8 +53,12 @@ class Config(object):
|
||||
|
||||
def load_config(self, args):
|
||||
|
||||
locations = [args.config,
|
||||
'./slack-backup.conf',
|
||||
path = ''
|
||||
if hasattr(args, 'config') and args.config:
|
||||
path = args.config
|
||||
|
||||
locations = [path,
|
||||
'./slack-backup.ini',
|
||||
os.path.expandvars('$XDG_CONFIG_HOME/slack-backup.ini'),
|
||||
os.path.expandvars('$HOME/.config/slack-backup.ini')]
|
||||
|
||||
@@ -79,19 +83,25 @@ class Config(object):
|
||||
val = self.cp.get(section, option, fallback='[]')
|
||||
val = json.loads(val)
|
||||
else:
|
||||
val = self.cp.get(section, option, fallback='')
|
||||
val = self.cp.get(section, option, fallback=None)
|
||||
|
||||
self._options[option] = val
|
||||
|
||||
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
|
||||
if args.verbose is not None and self._options['quiet'] is not None:
|
||||
self._options['quiet'] = 0
|
||||
if 'verbose' in args and args.verbose is not None:
|
||||
self._options['verbose'] = args.verbose
|
||||
if args.quiet is not None and self._options['verbose'] is not None:
|
||||
self._options['verbose'] = 0
|
||||
if self._options['quiet'] is not None:
|
||||
self._options['quiet'] = 0
|
||||
if 'quiet' in args and args.quiet is not None:
|
||||
self._options['quiet'] = args.quiet
|
||||
if self._options['verbose'] is not None:
|
||||
self._options['verbose'] = 0
|
||||
|
||||
for sec_id in (args.parser, 'common'):
|
||||
for option in self.sections[sec_id]:
|
||||
|
||||
@@ -55,8 +55,8 @@ class Download(object):
|
||||
'file': self._files}
|
||||
|
||||
if filetype == 'file' and not self._authorized:
|
||||
logging.info("There was no (valid) credentials passed, therefore "
|
||||
"file `%s' cannot be downloaded", url)
|
||||
logging.warning("There was no (valid) credentials passed, "
|
||||
"therefore file `%s' cannot be downloaded", url)
|
||||
return
|
||||
|
||||
splitted = url.split('/')
|
||||
|
||||
@@ -51,13 +51,6 @@ class TestConfig(unittest.TestCase):
|
||||
args.config = None
|
||||
args.parser = 'fetch'
|
||||
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.update(args)
|
||||
@@ -66,7 +59,7 @@ class TestConfig(unittest.TestCase):
|
||||
'verbose': 2,
|
||||
'quiet': 0,
|
||||
'channels': [],
|
||||
'database': '',
|
||||
'database': None,
|
||||
'user': None,
|
||||
'password': None,
|
||||
'team': None,
|
||||
|
||||
Reference in New Issue
Block a user