v1 of console

1 files changed, 326 insertions, 0 deletions

diff --git a/python/examples/console.py b/python/examples/console.py
new file mode 100644
index 0000000..bd364f7
--- /dev/null
+++ b/python/examples/console.py
@@ -0,0 +1,326 @@
+## www.pubnub.com - PubNub Real-time push service in the cloud.
+# coding=utf8
+
+## PubNub Real-time Push APIs and Notifications Framework
+## Copyright (c) 2010 Stephen Blum
+## http://www.pubnub.com/
+
+import sys
+from Pubnub import PubnubAsync as Pubnub
+import threading
+from datetime import datetime
+
+from cmd2 import Cmd, make_option, options, Cmd2TestCase
+import optparse
+
+
+of=sys.stdout
+
+
+class color(object):
+    PURPLE = '\033[95m'
+    CYAN = '\033[96m'
+    DARKCYAN = '\033[36m'
+    BLUE = '\033[94m'
+    GREEN = '\033[92m'
+    YELLOW = '\033[93m'
+    RED = '\033[91m'
+    BOLD = '\033[1m'
+    UNDERLINE = '\033[4m'
+    END = '\033[0m'
+
+def print_ok(msg, channel=None):
+    chstr = color.PURPLE + "[" + datetime.now().strftime(
+        '%Y-%m-%d %H:%M:%S') + "] " + color.END
+    chstr += color.CYAN + "[Channel : " + channel + \
+        "] " if channel is not None else "" + color.END
+    try:
+        print >>of, ("\n")
+        print >>of, (chstr + color.GREEN + str(msg) + color.END)
+    except UnicodeEncodeError as e:
+        print >>of, (msg)
+
+def print_error(msg, channel=None):
+    chstr = color.PURPLE + "[" + datetime.now().strftime(
+        '%Y-%m-%d %H:%M:%S') + "] " + color.END
+    chstr += color.CYAN + "[Channel : " + channel + \
+        "] " if channel is not None else "" + color.END
+    try:
+        print >>of, ("\n")
+        print >>of, (chstr + color.RED + color.BOLD + str(msg) + color.END)
+    except UnicodeEncodeError as e:
+        print >>of, (msg)
+
+class DefaultPubnub(object):
+    def handlerFunctionClosure(self,name):
+        def handlerFunction(*args,**kwargs):
+            print_error("Pubnub not initialized. Use init command to initialize")
+        return handlerFunction
+    def __getattr__(self,name):
+        return self.handlerFunctionClosure(name)
+
+pubnub=DefaultPubnub()
+
+
+
+def kill_all_threads():
+    for thread in threading.enumerate():
+        if thread.isAlive():
+            thread._Thread__stop()
+
+def get_input(message, t=None):
+    while True:
+        try:
+            try:
+                command = raw_input(message)
+            except NameError:
+                command = input(message)
+            except KeyboardInterrupt:
+                return None
+
+            command = command.strip()
+
+            if command is None or len(command) == 0:
+                raise ValueError
+
+            if t is not None and t == bool:
+                valid = ["True", "true", "1", 1, "y", "Y", "yes", "Yes", "YES"]
+                if command in valid:
+                    return True
+                else:
+                    return False
+            if t is not None:
+                command = t(command)
+            else:
+                command = eval("'" + command + "'")
+
+            return command
+        except ValueError:
+            print_error("Invalid input : " + command)
+
+def _publish_command_handler(channel, message):
+    def _callback(r):
+        print_ok(r)
+
+    def _error(r):
+        print_error(r)
+    pubnub.publish(channel, message, _callback, _error)
+
+
+def _subscribe_command_handler(channel):
+
+    def _callback(r):
+        print_ok(r, channel)
+
+    def _error(r):
+        print_error(r, channel)
+    pubnub.subscribe(channel, _callback, _error)
+
+
+def _unsubscribe_command_handler(channel):
+
+    def _callback(r):
+        print_ok(r)
+
+    def _error(r):
+        print_error(r)
+    pubnub.unsubscribe(channel)
+
+
+def _grant_command_handler(channel, auth_key, read, write, ttl):
+    def _callback(r):
+        print_ok(r)
+
+    def _error(r):
+        print_error(r)
+
+    pubnub.grant(channel, auth_key, read, write, ttl, _callback)
+
+
+def _revoke_command_handler(channel, auth_key, ttl):
+    def _callback(r):
+        print_ok(r)
+
+    def _error(r):
+        print_error(r)
+
+    pubnub.revoke(channel, auth_key, ttl, _callback)
+
+
+def _audit_command_handler(channel, auth_key):
+    def _callback(r):
+        print_ok(r)
+
+    def _error(r):
+        print_error(r)
+
+    pubnub.audit(channel, auth_key, _callback)
+
+
+def _history_command_handler(channel, count):
+    def _callback(r):
+        print_ok(r)
+
+    def _error(r):
+        print_error(r)
+
+    pubnub.history(channel, count, callback=_callback, error=_error)
+
+
+def _here_now_command_handler(channel):
+    def _callback(r):
+        print_ok(r)
+
+    def _error(r):
+        print_error(r)
+
+    pubnub.here_now(channel, callback=_callback, error=_error)
+
+
+class DevConsole(Cmd):
+    multilineCommands = ['orate']
+    Cmd.shortcuts.update({'&': 'speak'})
+    maxrepeats = 3
+    Cmd.settable.append('maxrepeats')
+
+    def __init__(self):
+        Cmd.__init__(self)
+        self.prompt = "(PubNub Console) > "
+        self.intro  = "Welcome to PubNub Developer Console!"  ## defaults to None
+
+    def cmdloop(self):
+        try:
+            Cmd.cmdloop(self)
+        except KeyboardInterrupt as e:
+            self.cmdloop()
+
+
+    @options([make_option('-p', '--publish-key', action="store", default="demo", help="Publish Key"),
+            make_option('-s', '--subscribe-key', action="store", default="demo", help="Subscribe Key"),
+            make_option('-k', '--secret-key', action="store", default="demo", help="cipher Key"),
+            make_option('-c', '--cipher-key', action="store", default="", help="Secret Key"),
+            make_option('-a', '--auth-key', action="store", default=None, help="Auth Key"),
+            make_option('--ssl-on',  dest='ssl', action='store_true', default=False, help="SSL Enabled ?"),
+            make_option('-o', '--origin', action="store", default="pubsub.pubnub.com", help="Origin"),   
+            make_option('-u', '--uuid', action="store", default=None, help="UUID")
+     ])
+    def do_init(self, command, opts):
+        global pubnub
+        pubnub = Pubnub(opts.publish_key,
+                opts.subscribe_key,
+                opts.secret_key,
+                opts.cipher_key,
+                opts.auth_key,
+                opts.ssl,
+                opts.origin,
+                opts.uuid)
+
+    @options([make_option('-c', '--channel', action="store", help="Channel for here now data")
+         ])
+    def do_here_now(self, command, opts):
+
+        if opts.channel is None:
+            print_error("Missing channel")
+            return
+
+        _here_now_command_handler(opts.channel)
+
+    @options([make_option('-c', '--channel', action="store", help="Channel for history data"),
+            make_option('-n', '--count', action="store", default=5, help="Number of messages")
+         ])
+    def do_history(self, command, opts):
+
+        if opts.channel is None:
+            print_error("Missing channel")
+            return
+
+        _history_command_handler(opts.channel, opts.count)
+
+
+    @options([make_option('-c', '--channel', action="store", help="Channel on which to publish"),
+                make_option('-m', '--message', action="store", help="Message to be published")
+         ])
+    def do_publish(self, command, opts):
+
+        if opts.channel is None:
+            print_error("Missing channel")
+            return
+
+        if opts.message is None:
+            print_error("Missing message")
+            return
+
+        _publish_command_handler(opts.channel,opts.message)
+
+    @options([make_option('-c', '--channel', action="store", help="Channel on which to grant"),
+                make_option('-a', '--auth-key', dest="auth_key", action="store",
+                    help="Auth Key"),
+                make_option('-r', '--read-enabled', dest='read', action='store_true', default=False, help="Read ?"),
+                make_option('-w', '--write-enabled', dest='write', action='store_true', default=False, help="Write ?"),
+                make_option('-t', '--ttl', action="store", default=5, help="TTL")
+         ])
+    def do_grant(self, command, opts):
+        if opts.channel is None:
+            print_error("Missing channel")
+            return
+
+        opts.auth_key = pubnub.auth_key if opts.auth_key is None else opts.auth_key
+
+        _grant_command_handler(opts.channel,opts.auth_key, opts.read, opts.write, opts.ttl)
+
+    @options([make_option('-c', '--channel', action="store", help="Channel on which to revoke"),
+                make_option('-a', '--auth-key', dest="auth_key", action="store",
+                    help="Auth Key"),
+                make_option('-t', '--ttl', action="store", default=5, help="TTL")
+         ])
+    def do_revoke(self, command, opts):
+        if opts.channel is None:
+            print_error("Missing channel")
+            return
+
+        opts.auth_key = pubnub.auth_key if opts.auth_key is None else opts.auth_key
+
+        _revoke_command_handler(opts.channel,opts.auth_key, opts.ttl)
+
+    @options([make_option('-a', '--auth-key', dest="auth_key", action="store",
+                    help="Auth Key")
+         ])
+    def do_audit(self, command, opts):
+
+        opts.auth_key = pubnub.auth_key if opts.auth_key is None else opts.auth_key
+
+        _audit_command_handler(opts.auth_key)
+
+
+    @options([make_option('-c', '--channel', action="store", help="Channel for unsubscribe")
+         ])
+    def do_unsubscribe(self, command, opts):
+        if opts.channel is None:
+            print_error("Missing channel")
+            return
+        _unsubscribe_command_handler(opts.channel)
+
+
+    @options([make_option('-c', '--channel', action="store", help="Channel for subscribe")
+         ])
+    def do_subscribe(self, command, opts):
+
+        if opts is None:
+            print_error("Missing argument")
+            return
+
+        if opts.channel is None:
+            print_error("Missing channel")
+            return
+
+        _subscribe_command_handler(opts.channel)
+
+
+
+def main():
+    app = DevConsole()
+    app.cmdloop()
+  
+if __name__ == "__main__":
+    main()    
+