watson_ws_test June 5, 2017 In [1]: #test websocket connection to watson

Transcription

1 watson_ws_test June 5, 2017 In [1]: #test websocket connection to watson In [2]: import websocket import thread import time import os import requests import certifi In [3]: from os.path import join, dirname from dotenv import load_dotenv Out[3]: True dotenv_path = join(dirname(" file "), '.env') load_dotenv(dotenv_path) In [4]: class def getauthenticationtoken(hostname, servicename, username, password): uri = hostname + "/authorization/api/v1/token?url=" + hostname + '/' \ + servicename + "/api" uri = uri.replace("wss://", " uri = uri.replace("ws://", " print (uri) resp = requests.get(uri, auth=(username, password), verify=certifi.where(), headers={'accept': 'application/json', timeout=(30, 30)) print (resp.text) jsonobject = resp.json() return jsonobject['token'] In [5]: hostname=" servicename='speech-to-text' username=os.environ['watson_username'] password=os.environ['watson_password'] url=os.environ['watson_url'] 1

2 In [15]: token = Utils.getAuthenticationToken(hostname, servicename, username, password) url = 'wss://stream.watsonplatform.net/speech-to-text/api/v1/recognize?watson-token=' + token + '&model=en-us_broadbandmodel' {"token":"ovni009okr%2btfk2cpxige5zmpyafqvd%2f6i%2fbpkqvvccrmenc3xtszvu2pfi7iy2xs7ugdkox798d3t09 In [9]: def read_in_chunks(file_object, chunk_size=1024): """Lazy function (generator) to read a file piece by piece. Default chunk size: 1k.""" while True: data = file_object.read(chunk_size) if not data: break yield data In [10]: def get_audio_filepath(): filepath = 'projects/sausalito_nov2016_5min_conv/audio-segments/ flac' return filepath In [11]: def get_json_filepath(): filepath = 'projects/sausalito_nov2016_5min_conv/transcripts/ _ws.json' return filepath In [12]: os.path.getsize(get_audio_filepath()) Out[12]: In [13]: # def on_message(ws, message): listenmessage = 0 print(u"text message received: {0".format(message)) jsonobject = json.loads(message) # if uninitialized, receive the initialization response # from the server if 'state' in jsonobject: listenmessage += 1 if (listenmessage == 2): print ("Sending results...") # if in streaming elif 'results' in jsonobject: jsonobject = json.loads(message) hypothesis = "" # empty hypothesis if (len(jsonobject['results']) == 0): print ("empty hypothesis!") 2

3 # regular hypothesis else: # dump the message to the output directory jsonobject = json.loads(message) json_file = get_json_filepath() with open(json_file, 'w') as f: json.dump(jsonobject, indent=4, sort_keys=true) print('transcript received {0:'. format(json_file)) #close the connection ws.close() def on_error(ws, error): print (error) def on_close(ws): print ("### closed ###") def on_open(ws): message = { 'action': 'start', 'content-type': 'audio/flac', 'inactivity_timeout': -1 ws.send(json.dumps(message).encode('utf8')) audio_file = get_audio_filepath() #file = open(get_audio_filepath(), 'rb') #for chunk in read_in_chunks(file): # ws.send(chunk) with open (audio_file, "rb") as file: data = file.read() ws.send(data) print("sending file...") file_sent = {'action': 'stop' ws.send (json.dumps(file_sent).encode('utf8')) In [14]: websocket.enabletrace(true) ws = websocket.websocketapp(url, on_message = on_message, on_error = on_error, on_close = on_close) ws.on_open = on_open ws.run_forever() error from callback <function on_open at 0x108622ae8>: [Errno 32] Broken pipe Text message received: { "state": "listening" 3

4 [SSL: BAD_LENGTH] bad length (_ssl.c:1844) ### closed ### In [16]: #from websocket import create_connection ws = websocket.create_connection("ws://echo.websocket.org/") print ("Sending 'Hello, World'...") ws.send("hello, World") print ("Sent") print ("Receiving...") result = ws.recv() print ("Received '%s'" % result) ws.close() Sending 'Hello, World'... Sent Receiving... Received 'Hello, World' In [14]: #from websocket import create_connection ws = websocket.create_connection(url) message = { 'action': 'start', 'content-type': 'audio/flac;rate=16000' ws.send(json.dumps(message).encode('utf8')) print("opening connection") result = ws.recv() print("result: {0".format(result)) f = open(get_audio_filepath(), encoding = 'latin-1', buffering = 3) try: data = f.read() ws.send(data) Print("Data sent...") except Exception as e: print("goddamnit. Closing connection: {0".format(e)) result = ws.recv() opening connection Result: { "state": "listening" Goddamnit. Closing connection: [Errno 32] Broken pipe

5 SSLError Traceback (most recent call last) <ipython-input-14-0aedab596ca2> in <module>() 16 except Exception as e: 17 print("goddamnit. Closing connection: {0".format(e)) ---> 18 result = ws.recv() /Users/kate/dev/pblc_dev/venv3_pblc/lib/python3.5/site-packages/websocket/_core.py in re 291 return value: string(byte array) value. 292 """ --> 293 opcode, data = self.recv_data() 294 if six.py3 and opcode == ABNF.OPCODE_TEXT: 295 return data.decode("utf-8") /Users/kate/dev/pblc_dev/venv3_pblc/lib/python3.5/site-packages/websocket/_core.py in re 308 return value: tuple of operation code and string(byte array) value. 309 """ --> 310 opcode, frame = self.recv_data_frame(control_frame) 311 return opcode, frame.data 312 /Users/kate/dev/pblc_dev/venv3_pblc/lib/python3.5/site-packages/websocket/_core.py in re elif frame.opcode == ABNF.OPCODE_CLOSE: --> 337 self.send_close() 338 return frame.opcode, frame 339 elif frame.opcode == ABNF.OPCODE_PING: /Users/kate/dev/pblc_dev/venv3_pblc/lib/python3.5/site-packages/websocket/_core.py in se 368 raise ValueError("code is invalid range") 369 self.connected = False --> 370 self.send(struct.pack('!h', status) + reason, ABNF.OPCODE_CLOSE) def close(self, status=status_normal, reason=six.b(""), timeout=3): /Users/kate/dev/pblc_dev/venv3_pblc/lib/python3.5/site-packages/websocket/_core.py in se frame = ABNF.create_frame(payload, opcode) --> 234 return self.send_frame(frame) def send_frame(self, frame): 5

6 /Users/kate/dev/pblc_dev/venv3_pblc/lib/python3.5/site-packages/websocket/_core.py in se 257 with self.lock: 258 while data: --> 259 l = self._send(data) 260 data = data[l:] 261 /Users/kate/dev/pblc_dev/venv3_pblc/lib/python3.5/site-packages/websocket/_core.py in _s def _send(self, data): --> 423 return send(self.sock, data) def _recv(self, bufsize): /Users/kate/dev/pblc_dev/venv3_pblc/lib/python3.5/site-packages/websocket/_socket.py in try: --> 116 return sock.send(data) 117 except socket.timeout as e: 118 message = extract_err_message(e) /usr/local/cellar/python3/3.5.2_2/frameworks/python.framework/versions/3.5/lib/python "non-zero flags not allowed in calls to send() on %s" % 860 self. class ) --> 861 return self._sslobj.write(data) 862 else: 863 return socket.send(self, data, flags) /usr/local/cellar/python3/3.5.2_2/frameworks/python.framework/versions/3.5/lib/python The 'data' argument must support the buffer interface. 585 """ --> 586 return self._sslobj.write(data) def getpeercert(self, binary_form=false): SSLError: [SSL: BAD_LENGTH] bad length (_ssl.c:1844) In [16]: def on_message(ws, message): print (message) def on_error(ws, error): 6

7 print (error) def on_close(ws): print ("### closed ###") def on_open(ws): def run(*args): for i in range(3): time.sleep(1) ws.send("hello %d" % i) time.sleep(1) ws.close() print ("thread terminating...") thread.start_new_thread(run, ()) if name == " main ": websocket.enabletrace(true) ws = websocket.websocketapp("ws://echo.websocket.org/", on_message = on_message, on_error = on_error, on_close = on_close) ws.on_open = on_open ws.run_forever() [Errno 61] Connection refused ### closed ### In [ ]: 7