Skip to main content

HuoBi Trading Framwork(python)

Project description

火币API的Python版

  • websocket封装成HBWebsocket类,用run开启连接线程
  • HBWebsocket通过注册Handler的方式来处理数据,消息通过pub_msg来分发到个各topic下的Handler线程来处理
  • 火币的鉴权WS是与普通WS独立的,所以同时使用需要开启两个WS
  • restful api基本参照火币网的demo封装成HBRestAPI
  • 兼容win,mac,linux,python版本必须3.6或以上,因为使用了大量的f***
  • 目前已经稳定使用,后续会基于框架提供如行情持久化,交易数据持久化等handler
  • 有疑问或者需要支持和交流的小伙伴可以联系我, QQ:137150224
  • 鉴于小伙伴数量也越来越多,所以建个小群:859745469 , 方便大家交流

QQ Group

Notice

  • 该封装的函数命名跟火币本身的请求命名表达不太一致
  • 包含open, close, high, low的数据命名是kline(其中部分有ask和bid,都纳入这类命名)
  • 当且仅当数据只有一条逐笔tick(没有ohlc),命名是ticker
  • 深度数据则命名为depth

Lastest

  • 合约与现货已经进行了部分测试,保证可用性
  • 优化相关datatype

PyPI GitHub forks GitHub stars build license PyPI - Python Version

1. Installation

pip install huobitrade

2. Usage

  • 实现长连订阅策略最核心的部分是实现handler里面的handle函数
    1. 通过HBWebsocket实例的sub开头的函数订阅需要的topic
    2. 通过继承BaseHandler的实例的初始化或者add_topic来增加对相关topic,实现handle函数来处理相关topic的消息
    3. 通过HBWebsocket实例的register_handler来注册handler
    4. handler初始化中有个latest,建议使用depth或者ticker数据来做处理,且策略性能不高的时候使用它
  • 基于websocket的接口都是用异步回调的方式来处理
    1. 首先需要用HBWebsocket的装饰器register_onRsp来绑定实现一个处理相关的topic消息的函数
    2. 再用req开头的函数来请求相关topic数据,回调的数据将会交给回调函数处理
  • 交易相关的都是用的restful api(因为火币还没推出websocket的交易接口)
    1. setKey是必要的,如果需要用到交易相关请求的话,只是请求行情的话可以不设。
    2. HBRestAPI是单例类,所以多次初始化也木有啥大问题,如在handler里面初始化
    3. 每个请求都有一个_async参数来提供异步请求,建议尽量使用它,具体用法是先初始化数个请求到一个list,再用async_request一次性向服务器发起请求
    4. 子账户体系没用过,可能会有问题,有bug欢迎pr
  • 另外还提供了几个简单易用的封装类
    1. HBMarket, HBAccount, HBMargin分别是行情,账户和借贷账户类,里面提供了大部分属性调用请求,均基于HBRestAPI
    2. 一般情景下应该是可以替代HBRestAPI的使用的
  • 最后还提供了数个运营管理的工具
    1. 微信推送handler,可以实现一些交易信息推送之类的,但是建议朋友们慎用,因为鄙人有试过一天推送几千条信息被封禁了半个月微信web端登陆的经历
    2. rpc模块,具体用法就不细说了,懂的应该都懂的,看下源码就知道咋用啦
  • 最后的最后,其实基于这个项目,还有附带的另外一个可视化web的项目没有放出来
    1. 基于flask写的一个用于查询当日成交明细和成交分布图,很丑很简陋
    2. 有兴趣的小伙伴可以联系我

2.1 huobitrade CLI Tool

  • huobitrade run -f strategy.py -a access-key -s secret-key用于启用一个基本简单的策略,其中strategy里应该可以包含一个init和handle_func用于初始化或处理相关topic.连接和鉴权成功后,会进入交互环境,提供6个命名空间来进行交互,包括restapi ws auth_ws account data margin,分别都是huobitrade几个主要类的实例huobi
  • huobitrade test_conn用于测试是否可以正常连接, huobitrade doc打开huobitrade文档
  • huobitrade --help通过该命令获取帮助

2.2.1 WebSocket API

from huobitrade.service import HBWebsocket

hb = HBWebsocket()  # 可以填入url参数,默认是api.huobi.br.com
@hb.after_open  # 使用装饰器注册函数,当ws连接之后会调用函数,可以实现订阅之类的
def sub_depth():
    hb.sub_depth('ethbtc')

hb.run()  # 开启websocket进程

# --------------------------------------------
hb.sub_kline('ethbtc', '1min')  # 订阅数据
@hb.register_handle_func('market.ethbtc.kline.1min')  # 注册一个处理函数,最好的处理方法应该是实现一个handler
def handle(msg):
    print('handle:', msg)

hb.unregister_handle_func(handle, 'market.ethbtc.kline.1min')  # 释放处理函数

# --------------------------------------------
# websocket请求数据是异步请求回调,所以先注册一个回调处理函数,再请求
@hb.register_onRsp('market.btcusdt.kline.1min')
def OnRsp_print(msg):
    print(msg)

hb.req_kline('btcusdt', '1min')
hb.unregister_onRsp('market.btcusdt.kline.1min')  # 注销某topic的请求回调处理

2.2.2 Auth WebSocket API

from huobitrade import setKey
from huobitrade.service import HBWebsocket
setKey('your acess_key', 'you secret_key')
hb = HBWebsocket(auth=True)  # 可以填入url参数,默认是api.huobi.br.com
@hb.after_auth  # 会再鉴权成功通过之后自动调用
def sub_accounts():
    hb.sub_accounts()

hb.run()  # 开启websocket进程

@hb.register_handle_func('accounts')  # 注册一个处理函数,最好的处理方法应该是实现一个handler
def auth_handle(msg):
    print('auth_handle:', msg)

2.3 Restful API

  • restapi需要先用setKey设置密钥
  • 默认交易和行情url都是https://api.huobi.br.com (调试用),实盘要用from huobitrade import setUrl设置url
from huobitrade.service import HBRestAPI
from huobitrade import setKey
# setUrl('', '')
setKey('your acess_key', 'you secret_key')  # setKey很重要,最好在引入其他模块之前先setKey,鉴权ws和restapi的部分函数是基于密钥
api = HBRestAPI(get_acc=True)  # get_acc参数默认为False,初始化不会取得账户ID,需要ID的函数无法使用.也可用api.set_acc_id('you_account_id')
print(api.get_timestamp())

api = HBRestAPI(get_acc=True) # 异步请求
klines = api.get_kline('omgeth', '1min', _async=True)
symbols = api.get_symbols(_async=True)
results = api.async_request([klines, symbols])
for r in results:
    print(r)

2.4 Message Handler

  • handler是用来处理websocket的原始返回消息的,通过继承basehandler实现handle函数以及注册进HBWebsocket相关的topic来使用
from huobitrade.handler import BaseHandler
from huobitrade.utils import handler_profiler
from huobitrade import setKey
from huobitrade.service import HBWebsocket
setKey('your acess_key', 'you secret_key')
hb = HBWebsocket(auth=True)  # 可以填入url参数,默认是api.huobi.br.com

class MyHandler(BaseHandler):
    def __init__(self, topic, *args, **kwargs):
        BaseHandler.__init__(self, 'just Thread name', topic)

    @handler_profiler('profiler.csv')  #  可以加上这个装饰器来测试handle函数的执行性能,加参数会输出到单独文件
    def handle(self, topic, msg):  # 实现handle来处理websocket推送的msg
        print(topic, msg)


handler = MyHandler('market.ethbtc.kline.1min')  # topic为str或者list
handler.add_topic('market.ethbtc.kline.5min')  # 为handler增加处理topic(remove_topic来删除)
hb.register_handler(handler)  # 通过register来把handler注册到相应的topic
  • 内置实现了一个mongodb的DBHandler
from huobitrade.handler import DBHandler
from huobitrade import setKey
from huobitrade.service import HBWebsocket
setKey('your acess_key', 'you secret_key')
hb = HBWebsocket(auth=True)  # 可以填入url参数,默认是api.huobi.br.com
handler = DBHandler()  # topic为空的话,会对所有topic的msg做处理
hb.register_handler(handler)

2.5 Latest Message Handler

  • 基于handler函数根据策略复杂度和性能的的不同造成对message的处理时间不一样,可能造成快生产慢消费的情况,增加lastest参数,每次都是handle最新的message
from huobitrade.handler import BaseHandler
from huobitrade.utils import handler_profiler
class MyLatestHandler(BaseHandler):
    def __init__(self, topic, *args, **kwargs):
        BaseHandler.__init__(self, 'just Thread name', topic, latest=True)

    @handler_profiler()  #  可以加上这个装饰器来测试handle函数的执行性能
    def handle(self, topic, msg):  # 实现handle来处理websocket推送的msg
        print(topic, msg)

2.6 HBData

  • 使用类似topic的方式来取数据,topic的表达方式与火币有不同
from huobitrade import setKey
setKey('acess_key', 'secret_key')
from huobitrade.datatype import HBMarket, HBAccount, HBMargin

data = HBMarket()  # 行情接口类
account = HBAccount()  # 交易接口类
margin = HBMargin()  # 借贷接口类

data.omgeth
 # <Symbol:omgeth-{'base-currency': 'omg', 'quote-currency': 'eth', 'price-precision': 6, 'amount-precision': 4, 'symbol-partition': 'main'}>
data.omgeth.kline
 # <<class 'huobitrade.datatype.HBKline'> for omgeth>
data.omgeth.depth
 # <<class 'huobitrade.datatype.HBDepth'> for omgeth>
data.omgeth.ticker
 # <<class 'huobitrade.datatype.HBTicker'> for omgeth>
data.omgeth.kline._1min_200  # period前面加'_', 后面加数量最大值为2000
data.omgeth.kline.last
data.omgeth.kline.last_24_hour
data.omgeth.depth.step0  # step0,1,2,3,4,5
data.omgeth.ticker.last  # 最新的一条tick
data.omgeth.ticker.last_20  # last_1至last_2000
data.all_24h_kline  # 当前所有交易对的ticker
account.Detail  # 所有账户明细
account.balance_XXXX  # XXXX为account_id,某账户的结余, 引用结余信息会自动更新
account.order  # 账户的订单类
account.order['order_id']  # 查询某order明细,或者用get方法
account.order.send('account_id', 1, 'omgeth', 'buy-limit', 0.001666)  # 发送订单
account.order.batchcancel(['order_id1', 'order_id2'])
account.order + [1, 'omgeth', 'buy-limit', 0.001666] # 发送订单
account.order + {'acc_id': 'your_account_id', 'amount': 1, 'symbol': 'omgeth', 'type': 'buy-limit', 'price': 0.001666}
account.order - 'order_id'  # 取消订单
account.order - ['order_id1', 'order_id2'] # 批量取消订单
account.trade.get_by_id('order_id')  # 某账户的成交类(即火币的matchresults),也可以直接索引
margin.transferIn('ethusdt', 'eth', 1)
ethusdt_margin_info = margin['ethusdt']  # 或者用getBalance
ethusdt_margin_info.balance  # ethusdt交易对的保证金结余信息

3. Extra

  • 交易策略运营相关的模块,wechat推送rpc远程订阅调用等 详见extra

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

huobitrade-0.5.5.tar.gz (31.3 kB view hashes)

Uploaded Source

Built Distribution

huobitrade-0.5.5-py3-none-any.whl (34.9 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page