Add tool for injecting large numbers of messages into site for testing

(imported from commit 44643cc95cd0e66d91fcc1ea576379f5de76ac67)

tools/inject-messages/inject-messages

@@ -0,0 +1,105 @@
+#!/usr/bin/python
+import sys
+import os
+import optparse
+import re
+
+usage = """inject-messages [options]
+
+Sends a bunch of messages to the specified site.
+"""
+parser = optparse.OptionParser(usage=usage)
+parser.add_option("--site",
+                  dest="site",
+                  default="http://localhost:9991",
+                  action="store")
+(options, args) = parser.parse_args()
+
+sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
+import api.common
+
+characters = {'othello': {'api-key': '4e5d97591bec64bf57d2698ffbb563e3', 'intro': ''},
+              'iago':    {'api-key': 'd43b53c27a8106195b46781abc67901a', 'intro': ''}}
+
+users = {}
+
+def gen_next_user():
+    global users
+    user_list = users.keys()
+    user_list.sort()
+    while True:
+        for user in user_list:
+            yield user
+next_user = gen_next_user()
+
+def connect_all(users):
+    for user in users:
+        users[user]['client'] = api.common.HumbugAPI(email=user + '@humbughq.com',
+                                                     api_key=users[user]['api-key'],
+                                                     site=options.site,
+                                                     verbose=True)
+
+connect_all(characters)
+connect_all(users)
+
+def truncate(string, length):
+    if len(string) <= length:
+        return string
+    return string[:length - 3] + "..."
+
+last_character = None
+def send_message(character, text, subject):
+    global characters
+    global last_character
+    message = text
+    if character != last_character:
+        message = characters[character]['intro'] + message
+    client = characters[character]["client"]
+    request = {'type': 'stream',
+               'content': message,
+               'stream': 'othello',
+               'subject': truncate(subject, 60)}
+    response = client.send_message(request)
+    if response['result'] != 'success':
+        print "Could not send message (API Key " + characters[character]["api-key"] + "):"
+        print "  Request:", request
+        print "  Response:", response
+        exit(1)
+    last_character = character
+
+def maybe_assign_character(character):
+    global characters
+    global users
+    if not character in characters:
+        user = next_user.next()
+        characters[character] = dict(users[user])
+        characters[character]['intro'] = '(as ' + character.title() + ")\n"
+
+play = file(os.path.dirname(__file__) + os.path.sep + "othello")
+next_character = None
+next_message = ""
+act = ""
+subject = ""
+
+for line in play:
+    if re.search('^\s*$', line):
+        continue
+    elif re.search('^ACT', line):
+        act = line.strip()
+    elif re.search('^SCENE', line):
+        subject = act + ", " + line.strip()
+        last_character = None
+    elif re.search('^\S', line):
+        next_character = line.strip().lower()
+        maybe_assign_character(next_character)
+        next_message = ""
+    elif re.search('^\s*\[\[', line):
+        maybe_assign_character('the narrator')
+        send_message('the narrator', line.strip().replace('[[', '').replace(']]', ''), subject)
+    else:
+        if next_message:
+            next_message += "\n"
+        next_message += line.strip()
+        if re.search('[\.|\?|!]$', next_message):
+            send_message(next_character, next_message, subject)
+            next_message = ''