303 lines
11 KiB
Python
303 lines
11 KiB
Python
|
# encoding: UTF-8
|
|||
|
|
|
|||
|
|
'''
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
from __future__ import print_function
|
|||
|
|
|
|||
|
|
import base64
|
|||
|
|
import hashlib
|
|||
|
|
import hmac
|
|||
|
|
import json
|
|||
|
|
import re
|
|||
|
|
import urllib
|
|||
|
|
import zlib
|
|||
|
|
|
|||
|
|
from vnpy.api.rest import Request, RestClient
|
|||
|
|
from vnpy.api.websocket import WebsocketClient
|
|||
|
|
from vnpy.trader.vtGateway import *
|
|||
|
|
|
|||
|
|
REST_HOST = 'https://api.huobipro.com'
|
|||
|
|
WEBSOCKET_MARKET_HOST = 'wss://api.huobi.pro/ws' # Global站行情
|
|||
|
|
WEBSOCKET_ASSETS_HOST = 'wss://api.huobi.pro/ws/v1' # 资产和订单
|
|||
|
|
WEBSOCKET_CONTRACT_HOST = 'wss://www.hbdm.com/ws' # 合约站行情
|
|||
|
|
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def _split_url(url):
|
|||
|
|
"""
|
|||
|
|
将url拆分为host和path
|
|||
|
|
:return: host, path
|
|||
|
|
"""
|
|||
|
|
m = re.match('\w+://([^/]*)(.*)', url)
|
|||
|
|
if m:
|
|||
|
|
return m.group(1), m.group(2)
|
|||
|
|
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def createSignature(apiKey, method, host, path, secretKey):
|
|||
|
|
"""创建签名"""
|
|||
|
|
sortedParams = (
|
|||
|
|
("AccessKeyId", apiKey),
|
|||
|
|
("SignatureMethod", 'HmacSHA256'),
|
|||
|
|
("SignatureVersion", "2"),
|
|||
|
|
("Timestamp", datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%S'))
|
|||
|
|
)
|
|||
|
|
encodeParams = urllib.urlencode(sortedParams)
|
|||
|
|
|
|||
|
|
payload = [method, host, path, encodeParams]
|
|||
|
|
payload = '\n'.join(payload)
|
|||
|
|
payload = payload.encode(encoding='UTF8')
|
|||
|
|
|
|||
|
|
secretKey = secretKey.encode(encoding='UTF8')
|
|||
|
|
|
|||
|
|
digest = hmac.new(secretKey, payload, digestmod=hashlib.sha256).digest()
|
|||
|
|
|
|||
|
|
signature = base64.b64encode(digest)
|
|||
|
|
params = dict(sortedParams)
|
|||
|
|
params["Signature"] = signature
|
|||
|
|
return params
|
|||
|
|
|
|||
|
|
|
|||
|
|
########################################################################
|
|||
|
|
class HuobiRestApi(RestClient):
|
|||
|
|
|
|||
|
|
def __init__(self, gateway): # type: (VtGateway)->HuobiRestApi
|
|||
|
|
super(HuobiRestApi, self).__init__()
|
|||
|
|
self.gateway = gateway
|
|||
|
|
self.gatewayName = gateway.gatewayName
|
|||
|
|
|
|||
|
|
self.apiKey = ""
|
|||
|
|
self.apiSecret = ""
|
|||
|
|
self.signHost = ""
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def sign(self, request):
|
|||
|
|
request.headers = {
|
|||
|
|
"User-Agent":
|
|||
|
|
"Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.71 Safari/537.36"}
|
|||
|
|
additionalParams = createSignature(self.apiKey,
|
|||
|
|
request.method,
|
|||
|
|
self.signHost,
|
|||
|
|
request.path,
|
|||
|
|
self.apiSecret)
|
|||
|
|
if not request.params:
|
|||
|
|
request.params = additionalParams
|
|||
|
|
else:
|
|||
|
|
request.params.update(additionalParams)
|
|||
|
|
if request.method == "POST":
|
|||
|
|
request.headers['Content-Type'] = 'application/json'
|
|||
|
|
return request
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def connect(self, apiKey, apiSecret, sessionCount=3):
|
|||
|
|
"""连接服务器"""
|
|||
|
|
self.apiKey = apiKey
|
|||
|
|
self.apiSecret = apiSecret
|
|||
|
|
|
|||
|
|
host, path = _split_url(REST_HOST)
|
|||
|
|
self.init(REST_HOST)
|
|||
|
|
self.signHost = host
|
|||
|
|
self.start(sessionCount)
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def qeuryAccount(self):
|
|||
|
|
self.addRequest('GET', '/v1/account/accounts', self.onAccount)
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def onAccount(self, data, request): # type: (dict, Request)->None
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def cancelWithdraw(self, id):
|
|||
|
|
self.addRequest('POST',
|
|||
|
|
"/v1/dw/withdraw-virtual/" + str(id) + "/cancel",
|
|||
|
|
self.onWithdrawCanceled
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def onWithdrawCanceled(self, data, request): # type: (dict, Request)->None
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
########################################################################
|
|||
|
|
class HuobiWebsocketApiBase(WebsocketClient):
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def __init__(self, gateway):
|
|||
|
|
"""Constructor"""
|
|||
|
|
super(HuobiWebsocketApiBase, self).__init__()
|
|||
|
|
|
|||
|
|
self.gateway = gateway
|
|||
|
|
self.gatewayName = gateway.gatewayName
|
|||
|
|
|
|||
|
|
self.apiKey = ''
|
|||
|
|
self.apiSecret = ''
|
|||
|
|
self.signHost = ''
|
|||
|
|
self.path = ''
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def connect(self, apiKey, apiSecret, url):
|
|||
|
|
""""""
|
|||
|
|
self.apiKey = apiKey
|
|||
|
|
self.apiSecret = apiSecret
|
|||
|
|
|
|||
|
|
host, path = _split_url(url)
|
|||
|
|
|
|||
|
|
self.init(url)
|
|||
|
|
self.signHost = host
|
|||
|
|
self.path = path
|
|||
|
|
self.start()
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def login(self):
|
|||
|
|
params = {
|
|||
|
|
'op': 'auth',
|
|||
|
|
}
|
|||
|
|
params.update(
|
|||
|
|
createSignature(self.apiKey,
|
|||
|
|
'GET',
|
|||
|
|
self.signHost,
|
|||
|
|
self.path,
|
|||
|
|
self.apiSecret)
|
|||
|
|
)
|
|||
|
|
return self.sendPacket(params)
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def onLogin(self, packet):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
@staticmethod
|
|||
|
|
def unpackData(data):
|
|||
|
|
return json.loads(zlib.decompress(data, 31))
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def onPacket(self, packet):
|
|||
|
|
"""
|
|||
|
|
这里我新增了一个onHuobiPacket的函数，也可以让子类重写这个函数，然后调用super.onPacket
|
|||
|
|
"""
|
|||
|
|
if 'ping' in packet:
|
|||
|
|
self.sendPacket({'pong': packet['ping']})
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# todo: use another error handing method
|
|||
|
|
if 'err-msg' in packet:
|
|||
|
|
return self.onHuobiErrorPacket(packet)
|
|||
|
|
|
|||
|
|
if "op" in packet and packet["op"] == "auth":
|
|||
|
|
return self.onLogin(packet)
|
|||
|
|
|
|||
|
|
self.onHuobiPacket(packet)
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def onHuobiPacket(self, packet): # type: (dict)->None
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def onHuobiErrorPacket(self, packet): # type: (dict)->None
|
|||
|
|
print("error : {}".format(packet))
|
|||
|
|
|
|||
|
|
|
|||
|
|
########################################################################
|
|||
|
|
class HuobiAssetsWebsocketApi(HuobiWebsocketApiBase):
|
|||
|
|
|
|||
|
|
def connect(self, apiKey, apiSecret, host=WEBSOCKET_ASSETS_HOST):
|
|||
|
|
"""
|
|||
|
|
这里我使用重写connect，添加了默认参数。这样写感觉~~不太好~~，不过目前想到的比较好的方式就是这样了
|
|||
|
|
虽然在Python中可以直接把这个connect()写成不接收host和path的形式，但是PyCharm会提示重载错误，所以不接收host和path似乎不太好？
|
|||
|
|
|
|||
|
|
我觉得最好的写法应该是这个函数不接收host和path。同时为了让PyCharm不提示重载错误(减少歧义)，应该给
|
|||
|
|
HuobiWebsocketApiBase.connect起另外一个名字。
|
|||
|
|
"""
|
|||
|
|
return super(HuobiAssetsWebsocketApi, self). \
|
|||
|
|
connect(apiKey, apiSecret, host)
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def onConnected(self):
|
|||
|
|
self.login()
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def subscribeAccount(self):
|
|||
|
|
"""
|
|||
|
|
:param symbol: str ethbtc, ltcbtc, etcbtc, bchbtc
|
|||
|
|
:param period: str 1min, 5min, 15min, 30min, 60min, 1day, 1mon, 1week, 1year
|
|||
|
|
"""
|
|||
|
|
self.sendPacket({
|
|||
|
|
"op": "sub",
|
|||
|
|
"cid": "any thing you want",
|
|||
|
|
"topic": "accounts"
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def onHuobiPacket(self, packet): # type: (dict)->None
|
|||
|
|
if 'op' in packet:
|
|||
|
|
if packet['op'] == 'sub':
|
|||
|
|
timestamp = packet['ts']
|
|||
|
|
topic = packet['topic']
|
|||
|
|
"""
|
|||
|
|
"data": {
|
|||
|
|
"event": "order.match|order.place|order.refund|order.cancel|order.fee-refund|margin.transfer|margin.loan|margin.interest|margin.repay|other",
|
|||
|
|
"list": [
|
|||
|
|
{
|
|||
|
|
"account-id": 419013,
|
|||
|
|
"currency": "usdt",
|
|||
|
|
"type": "trade",
|
|||
|
|
"balance": "500009195917.4362872650"
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"account-id": 419013,
|
|||
|
|
"currency": "btc",
|
|||
|
|
"type": "frozen",
|
|||
|
|
"balance": "9786.6783000000"
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
"""
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
########################################################################
|
|||
|
|
class HuobiMarketWebsocketApi(HuobiWebsocketApiBase):
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def connect(self, apiKey, apiSecret, host=WEBSOCKET_MARKET_HOST):
|
|||
|
|
"""
|
|||
|
|
这里我使用重写connect，添加了默认参数。这样写感觉~~不太好~~，不过目前想到的比较好的方式就是这样了
|
|||
|
|
虽然在Python中可以直接把这个connect()写成不接收host和path的形式，但是PyCharm会提示重载错误，所以不接收host和path似乎不太好？
|
|||
|
|
|
|||
|
|
我觉得最好的写法应该是这个函数不接收host和path。同时为了让PyCharm不提示重载错误(减少歧义)，应该给
|
|||
|
|
HuobiWebsocketApiBase.connect起另外一个名字。
|
|||
|
|
"""
|
|||
|
|
return super(HuobiMarketWebsocketApi, self). \
|
|||
|
|
connect(apiKey, apiSecret, host)
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def subscribeKLine(self, symbol, period): # type:(str, str)->None
|
|||
|
|
"""
|
|||
|
|
:param symbol: str ethbtc, ltcbtc, etcbtc, bchbtc
|
|||
|
|
:param period: str 1min, 5min, 15min, 30min, 60min, 1day, 1mon, 1week, 1year
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
self.sendPacket({
|
|||
|
|
"sub": "market." + symbol + ".kline." + period,
|
|||
|
|
"id": "any thing you want"
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def onHuobiPacket(self, packet): # type: (dict)->None
|
|||
|
|
# code for test purpose only
|
|||
|
|
if 'ch' in packet:
|
|||
|
|
if packet['ch'] == 'market.btcusdt.kline.1min':
|
|||
|
|
timestamp = packet['ts']
|
|||
|
|
data = packet['tick']
|
|||
|
|
id = data['id']
|
|||
|
|
amount = data['amount']
|
|||
|
|
count = data['count']
|
|||
|
|
open = data['open']
|
|||
|
|
close = data['close']
|
|||
|
|
low = data['low']
|
|||
|
|
high = data['high']
|
|||
|
|
vol = data['vol']
|
|||
|
|
pass
|