From 1e1d5cb38cade198382fec5ef1e587fbfcf9488f Mon Sep 17 00:00:00 2001
From: gryf <gryf73@gmail.com>
Date: Fri, 18 Nov 2016 19:11:12 +0100
Subject: [PATCH] Added executable script and client class

---
 scripts/slack-backup.py  | 27 ++++++++++++++++++
 slack_backup/__init__.py |  0
 slack_backup/client.py   | 61 ++++++++++++++++++++++++++++++++++++++++
 3 files changed, 88 insertions(+)
 create mode 100755 scripts/slack-backup.py
 create mode 100644 slack_backup/__init__.py
 create mode 100644 slack_backup/client.py

diff --git a/scripts/slack-backup.py b/scripts/slack-backup.py
new file mode 100755
index 0000000..3f36c31
--- /dev/null
+++ b/scripts/slack-backup.py
@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Create backup for certain date for specified channel in slack
+"""
+import argparse
+import pprint
+
+from slack_backup import client
+
+
+def main():
+    """Main function"""
+    parser = argparse.ArgumentParser()
+    parser.add_argument("token", help="Slack token - a string, which can be"
+                        " generated/obtained via "
+                        "https://api.slack.com/docs/oauth-test-tokens page")
+    # parser.add_argument("channel", help="Slack channel name")
+    args = parser.parse_args()
+
+    slack = client.Client(args.token)
+
+    pprint.pprint(slack.get_hisotry(selected_channels=['elysium']))
+
+
+if __name__ == "__main__":
+    main()
diff --git a/slack_backup/__init__.py b/slack_backup/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/slack_backup/client.py b/slack_backup/client.py
new file mode 100644
index 0000000..68e8960
--- /dev/null
+++ b/slack_backup/client.py
@@ -0,0 +1,61 @@
+#!/usr/bin/env python3
+"""
+Create backup for certain date for specified channel in slack
+"""
+import argparse
+import logging
+
+import slackclient
+
+from slack_backup import objects
+
+
+class Client(object):
+    def __init__(self, token):
+        self.slack = slackclient.SlackClient(token)
+
+    def get_hisotry(self, selected_channels=None, from_date=0):
+        channels = self._get_channel_list()
+        if selected_channels:
+            selected_channels = [c for c in channels
+                                 if c.name in selected_channels]
+        else:
+            selected_channels = channels
+
+        users = self._get_user_list()
+
+        for channel in selected_channels:
+            history = []
+            latest = 'now'
+
+            while True:
+                messages, latest = self._get_channel_history(channel, latest)
+                # TODO: merge messages witihn a channel
+                if not messages:
+                    break
+
+    def _get_channel_history(self, channel, latest='now'):
+        result = self.slack.api_call("channels.history", channel=channel._id,
+                                     count=1000, latest=latest)
+
+        if not result.get("ok"):
+            logging.error(result['error'])
+            return None, None
+
+    def _get_channel_list(self):
+        result = self.slack.api_call("channels.list")
+
+        if not result.get("ok"):
+            logging.error(result['error'])
+            return None
+
+        return [objects.Channel(chan) for chan in result['channels']]
+
+    def _get_user_list(self):
+        result = self.slack.api_call("users.list", presence=0)
+
+        if not result.get("ok"):
+            logging.error(result['error'])
+            return None
+
+        return [objects.User(user) for user in result['members']]