python-kabusapi でWebsocketをクローズする

時間が無いので動作未検証だが、イメージとしてはこんな感じでソケットをクローズできるのではないかと考えている。時間取れた時に検証する。

@api.websocket
def recieve(msg):
    # ここで処理を行う msgはdict形式である。
    print("{} / {} {} {}".format(
        msg,
        msg['Symbol'],
        msg['SymbolName'],
        msg['CurrentPrice'],
    ))

try:
  # 受信開始
  api.websocket.run()
except KeyboardInterrupt:
  api.websocket.loop.close()
  #api.websocket.close()