openstack/deb-python-websocket-client
Code Issues Proposed changes

console decoration of wsdump.py inside InteractiveConsole

Browse Source
This commit is contained in:
Simó Albert i Beltran
2015-05-14 17:18:43 +02:00
parent d342049ee4
commit d6168cfed1
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bin/wsdump.py
View File

@@ -59,7 +59,7 @@ class InteractiveConsole(code.InteractiveConsole):
def write(self, data): def write(self, data):
sys.stdout.write("\033[2K\033[E") sys.stdout.write("\033[2K\033[E")
# sys.stdout.write("\n") # sys.stdout.write("\n")
sys.stdout.write("\033[34m" + data + "\033[39m") sys.stdout.write("\033[34m< " + data + "\033[39m")
sys.stdout.write("\n> ") sys.stdout.write("\n> ")
sys.stdout.flush() sys.stdout.flush()
@@ -117,9 +117,9 @@ def main():
opcode, data = recv() opcode, data = recv()
msg = None msg = None
if not args.verbose and opcode in OPCODE_DATA: if not args.verbose and opcode in OPCODE_DATA:
msg = "< %s" % data msg = data
elif args.verbose: elif args.verbose:
msg = "< %s: %s" % (websocket.ABNF.OPCODE_MAP.get(opcode), data) msg = "%s: %s" % (websocket.ABNF.OPCODE_MAP.get(opcode), data)
if msg is not None: if msg is not None:
console.write(msg) console.write(msg)