私のauカブコム証券のkabuステーションで自動売買プログラムの説明(9) 現在の株価が損切り値に達していないかを監視(websocket.py) その2

websocket.pyのソースコード
80行目 def websocketA1():すべての入り口
32行目 def on_open(ws):WebSoketで損切監視開始
13行目 def on_message(ws, message):株価が通知されるたびに損切りチェック

websocket.pyのソースコード

import sys
import websocket
import _thread
import pprint
import json
import settings
import rsi1570i_cancelorder
import datetime
import urllib.request
import board
import time

def on_message(ws, message):
    printWithTime('--- RECV MSG. --- ')
    content = json.loads(message)
    curPrice = content["CurrentPrice"]

    if(curPrice >= settings.orderPrice + settings.lossCutMargin):
        # キャンセルからの損切り注文
        cancelorder.cancelorder()
        ws.close()

def on_error(ws, error):
    if len(error) != 0:
        printWithTime('--- ERROR --- ')
        print(error)

def on_close(ws):
    printWithTime('--- DISCONNECTED --- ')


def on_open(ws):
    printWithTime('--- CONNECTED --- ')
    def run():
        while(True):

            # 指定時間sleep
            time.sleep(settings.intervalCloseCheck)

            url = 'http://localhost:' + settings.port + '/kabusapi/orders'
            params = { 'product': 0, }
            req = urllib.request.Request('{}?{}'.format(url, urllib.parse.urlencode(params)), method='GET')
            req.add_header('Content-Type', 'application/json')
            req.add_header('X-API-KEY', settings.token)

            try:
                with urllib.request.urlopen(req) as res:
                    content = json.loads(res.read())

                    # 注文情報を取得
                    lastOrder = content[len(content)-1]
                            
                    # 全約定の場合(売りが完了している場合)
                    if lastOrder['State'] == 5:

                        # まずはwebsoketを停止
                        printWithTime('closing...')
                        ws.close()
                        break
                    
                    # 手仕舞い時刻
                    nowtime = datetime.datetime.now()
                    if nowtime > settings.forceCloseTime:
                        # キャンセルからの損切り注文
                        rsi1570i_cancelorder.cancelorder()
                        ws.close()
                        break

            except urllib.error.HTTPError as e:
                print(e)
                content = json.loads(e.read())
                pprint.pprint(content)
            except Exception as e:
                print(e)



    _thread.start_new_thread(run, ())

def websocketA1():
    printWithTime('--- websocketA1 Start--- ')
    url = 'ws://localhost:' + settings.port + '/kabusapi/websocket'
    # 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()

    printWithTime('--- websocketA1 End --- ')



def printWithTime(message):
    print(str(datetime.datetime.now()) + ' ' + message)

if __name__ == "__main__":
    import sys
    websocketA1()

前回説明した、kabuステーションのwebsoketのサンプルコードに、このトレードプログラムでやりたいことを当てはめたのが、上記のコードです。

80行目 def websocketA1():すべての入り口

kabuステーションのサンプルは、それだけで実行するモジュールなので、ソースコードの地の部分(関数外)に最初に動く処理が書かれていましたが、このモジュールは前の処理から数珠繋ぎ方式で呼ばれるため、プログラムの入り口を関数にしています。

以前は、1日何回もエントリしていたため、「ws.run_forever()」の次に、次のエントリへつなげる処理を書いていたのですが、今は最大1日1回しかエントリしていないため、この後は何もしていません。WebSoketが閉じると、プログラム自体が終了します。

逆に言えば、1日何回もエントリしたい人は、「ws.run_forever()」の次に、「board.board()」を呼びだし、最初に戻る処理を入れてください。

32行目 def on_open(ws):WebSoketで損切監視開始

このWebSoketのモジュールは、外側のWebSocket自体ではkabuステーションのサーバから、株価のリアルタイム通知を受けて損切り価格に達していないかの監視をしています。

同時にdef run(): 無限ループで10秒ことに利食い決済したかも監視

しかし、WebSocketが開始した「def on_open(ws):」の中で、「def run():」関数を定義し、「thread.start_new_thread(run, ())」で別スレッドでその関数を呼ぶことによって、同時に以下の条件を監視し、条件に合う場合はWebSocketを終了させる処理をしています。