（okex量化）python打通api最简单的一个程序（inflate(data)函数是官方提供转码的，必须使用否则乱码解决不了！！）

落日映苍穹つ 2021-09-18 00:30 162阅读 0赞

import websocket
    import sys
    import zlib
    
    
    def on_open(self):
        #subscribe okcoin.com spot ticker
        self.send(
            '{"channel":"ok_sub_futureusd_btc_depth_quarter","event":"addChannel"}'
        )
        print("数据连接成功")
    
    
    #okex专用转码函数(如果不转码，会乱码无法解决)
    def inflate(data):
        decompress = zlib.decompressobj(-zlib.MAX_WBITS  # see above
                                        )
        inflated = decompress.decompress(data)
        inflated += decompress.flush()
        return inflated
    
    
    def on_message(self, evt):
        #这里必须用官方提供的inflate函数转码，否则是乱码！
        data = inflate(evt)
        print(data)
    
    
    def on_error(self, evt):
        print(evt)
    
    
    def on_close(self, evt):
        print('DISCONNECT')
    
    
    if __name__ == "__main__":
        url = "wss://real.okex.com:10441/websocket?compress=true"  #if okcoin.cn  change url wss://real.okcoin.cn:10440/websocket/okcoinapi
    
        websocket.enableTrace(False)
        if len(sys.argv) < 2:
            host = url
        else:
            host = sys.argv[1]
        ws = websocket.WebSocketApp(
            host, on_message=on_message, on_error=on_error, on_close=on_close)
        ws.on_open = on_open
        ws.run_forever()