From 26c85bb9ed2b2073eccafc263ea99abc64d982b2 Mon Sep 17 00:00:00 2001
From: Sivagiri Visakan <siva.g.visakan@gmail.com>
Date: Sun, 27 Jan 2019 16:00:45 +0530
Subject: [PATCH] zulip/examples: Add a script to get complete history of a
 narrow.

Fixes #476.
---
 zulip/zulip/examples/get-history | 59 ++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 zulip/zulip/examples/get-history

diff --git a/zulip/zulip/examples/get-history b/zulip/zulip/examples/get-history
new file mode 100644
index 0000000..f177e97
--- /dev/null
+++ b/zulip/zulip/examples/get-history
@@ -0,0 +1,59 @@
+#!/usr/bin/env python
+import argparse
+import json
+from typing import Dict, List, Any
+
+import zulip
+
+usage = """\nGets the complete history of a particular stream or a particular topic in Zulip
+and store them in JSON format.
+
+    get-history --stream <stream_name> [--topic <topic_name> --filename <file_to_store_messages>]
+
+Example: get-history --stream announce --topic important"""
+
+parser = zulip.add_default_arguments(argparse.ArgumentParser(usage=usage))
+parser.add_argument('--stream', required=True, help="The stream name to get the history")
+parser.add_argument('--topic', help="The topic name to get the history")
+parser.add_argument('--filename', default='history.json', help="The file name to store the fetched  \
+                    history.\n Default 'history.json'")
+options = parser.parse_args()
+
+client = zulip.init_from_options(options)
+
+narrow = [{'operator': 'stream', 'operand': options.stream}]
+if options.topic:
+    narrow.append({'operator': 'topic', 'operand': options.topic})
+
+request = {
+    # Initially we have the anchor as 0, so that it starts fetching
+    # from the oldest message in the narrow
+    'anchor': 0,
+    'num_before': 0,
+    'num_after': 1000,
+    'narrow': narrow,
+    'client_gravatar': False,
+    'apply_markdown': False
+}
+
+all_messages = []  # type: List[Dict[str, Any]]
+found_newest = False
+
+while not found_newest:
+    result = client.get_messages(request)
+    try:
+        found_newest = result["found_newest"]
+        if result['messages']:
+            # Setting the anchor to the next immediate message after the last fetched message.
+            request['anchor'] = result['messages'][-1]['id'] + 1
+
+            all_messages.extend(result["messages"])
+    except KeyError:
+        # Might occur when the request is not returned with a success status
+        print('Error occured: Payload was:')
+        print(result)
+        quit()
+
+with open(options.filename, "w+") as f:
+    print('Writing %d messages...' % len(all_messages))
+    f.write(json.dumps(all_messages))