256 lines
7.3 KiB
Python
256 lines
7.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
import sys
|
|||
|
|
import time
|
|||
|
|
import hashlib
|
|||
|
|
import requests
|
|||
|
|
import json
|
|||
|
|
from queue import Queue, Empty
|
|||
|
|
from multiprocessing.dummy import Pool
|
|||
|
|
from threading import Thread
|
|||
|
|
|
|||
|
|
# 此处为APIID SECRET
|
|||
|
|
apiid = ' '
|
|||
|
|
secret = ' '
|
|||
|
|
|
|||
|
|
# 此处为API请求地址及参数
|
|||
|
|
REST_HOST = "http://api.coinbene.com"
|
|||
|
|
market_url = "http://api.coinbene.com/v1/market/"
|
|||
|
|
trade_url = "http://api.coinbene.com/v1/trade/"
|
|||
|
|
header_dict = {"User-Agent": "Mozilla/5.0 (Windows NT 6.1; Trident/7.0; rv:11.0) like Gecko",\
|
|||
|
|
"Content-Type":"application/json;charset=utf-8","Connection":"keep-alive"}
|
|||
|
|
|
|||
|
|
FUNCTIONCODE_GET_ORDERBOOK_COINBENE = "orderbook"
|
|||
|
|
FUNCTIONCODE_GET_TICKER_COINBENE = "ticker"
|
|||
|
|
FUNCTIONCODE_GET_TRADE_COINBENE = "trades"
|
|||
|
|
FUNCTIONCODE_POST_BALANCE_COINBENE = "balance"
|
|||
|
|
FUNCTIONCODE_POST_CANCEL_COINBENE = "order/cancel"
|
|||
|
|
FUNCTIONCODE_POST_INFO_COINBENE = "order/info"
|
|||
|
|
FUNCTIONCODE_POST_ORDERPLACE_COINBENE = "order/place"
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class coinbeneRestApi(object):
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
self.apiKey = ''
|
|||
|
|
self.apiSecret = ''
|
|||
|
|
|
|||
|
|
self.active = False
|
|||
|
|
self.reqid = 0
|
|||
|
|
self.queue = Queue()
|
|||
|
|
self.pool = None
|
|||
|
|
self.sessionDict = {} # 会话对象字典
|
|||
|
|
|
|||
|
|
self.header = {
|
|||
|
|
'Content-Type': 'application/x-www-form-urlencoded',
|
|||
|
|
'Accept': 'application/json'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# ----------------------------------------------------------------------
|
|||
|
|
def init(self, apiKey, apiSecret):
|
|||
|
|
"""初始化"""
|
|||
|
|
self.apiKey = apiKey
|
|||
|
|
self.apiSecret = apiSecret
|
|||
|
|
|
|||
|
|
# ----------------------------------------------------------------------
|
|||
|
|
def start(self, n=3):
|
|||
|
|
"""启动"""
|
|||
|
|
if self.active:
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
self.active = True
|
|||
|
|
self.pool = Pool(n)
|
|||
|
|
self.pool.map_async(self.run, range(n))
|
|||
|
|
|
|||
|
|
# ----------------------------------------------------------------------
|
|||
|
|
def close(self):
|
|||
|
|
"""关闭"""
|
|||
|
|
self.active = False
|
|||
|
|
|
|||
|
|
if self.pool:
|
|||
|
|
self.pool.close()
|
|||
|
|
self.pool.join()
|
|||
|
|
|
|||
|
|
# ----------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
def addReq(self, method, path, callback, params=None, postdict=None):
|
|||
|
|
"""添加请求"""
|
|||
|
|
self.reqid += 1
|
|||
|
|
req = (method, path, callback, params, postdict, self.reqid)
|
|||
|
|
self.queue.put(req)
|
|||
|
|
return self.reqid
|
|||
|
|
|
|||
|
|
# ----------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
def processReq(self, req, i):
|
|||
|
|
"""处理请求"""
|
|||
|
|
method, callback, params, postdict, reqid = req
|
|||
|
|
|
|||
|
|
# 使用长连接的session，比短连接的耗时缩短80%
|
|||
|
|
session = self.sessionDict[i]
|
|||
|
|
resp = session.request(method, url, headers=header, params=params, data=postdict)
|
|||
|
|
|
|||
|
|
# resp = requests.request(method, url, headers=header, params=params, data=postdict)
|
|||
|
|
|
|||
|
|
code = resp.status_code
|
|||
|
|
d = resp.json()
|
|||
|
|
|
|||
|
|
if code == 200:
|
|||
|
|
callback(d, reqid)
|
|||
|
|
else:
|
|||
|
|
self.onError(code, d)
|
|||
|
|
|
|||
|
|
# ----------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
def run(self, i):
|
|||
|
|
"""连续运行"""
|
|||
|
|
self.sessionDict[i] = requests.Session()
|
|||
|
|
|
|||
|
|
while self.active:
|
|||
|
|
try:
|
|||
|
|
req = self.queue.get(timeout=1)
|
|||
|
|
self.processReq(req, i)
|
|||
|
|
except Empty:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# ----------------------------------------------------------------------
|
|||
|
|
def onError(self, code, error):
|
|||
|
|
"""错误回调"""
|
|||
|
|
print(u'on error {} {}'.format(code, error),file=sys.stderr)
|
|||
|
|
|
|||
|
|
# ----------------------------------------------------------------------
|
|||
|
|
def onData(self, data, reqid):
|
|||
|
|
"""通用回调"""
|
|||
|
|
print(u'on data:{} reqid:{}'.format(data, reqid))
|
|||
|
|
|
|||
|
|
# 生成签名sign
|
|||
|
|
def sign(self,**kwargs):
|
|||
|
|
"""
|
|||
|
|
将传入的参数生成列表形式，排序后用＆拼接成字符串，用hashbli加密成生sign
|
|||
|
|
"""
|
|||
|
|
sign_list = []
|
|||
|
|
for key, value in kwargs.items():
|
|||
|
|
sign_list.append("{}={}".format(key, value))
|
|||
|
|
sign_list.sort()
|
|||
|
|
sign_str = "&".join(sign_list)
|
|||
|
|
mysecret = sign_str.upper().encode()
|
|||
|
|
m = hashlib.md5()
|
|||
|
|
m.update(mysecret)
|
|||
|
|
return m.hexdigest()
|
|||
|
|
|
|||
|
|
# 生成时间戳
|
|||
|
|
def create_timestamp(self):
|
|||
|
|
timestamp = int(round(time.time() * 1000))
|
|||
|
|
return timestamp
|
|||
|
|
|
|||
|
|
#timestamp = create_timestamp()
|
|||
|
|
|
|||
|
|
def http_get_nosign(self, url, i=0):
|
|||
|
|
return self.http_request(url, data=None,i=i)
|
|||
|
|
|
|||
|
|
def http_post_sign(self,url,dic,i=0):
|
|||
|
|
mysign = self.sign(**dic)
|
|||
|
|
del dic['secret']
|
|||
|
|
dic['sign'] = mysign
|
|||
|
|
return self.http_request(url,data=dic,i=i)
|
|||
|
|
|
|||
|
|
def http_request(self, url, data,i=0):
|
|||
|
|
session = self.sessionDict[i]
|
|||
|
|
if data == None:
|
|||
|
|
reponse = session.request(method, url, headers=header_dict, params=params, data=postdict)
|
|||
|
|
reponse = requests.get(url,headers=header_dict)
|
|||
|
|
else:
|
|||
|
|
reponse = requests.post(url,data=json.dumps(data),headers=header_dict)
|
|||
|
|
try:
|
|||
|
|
if reponse.status_code == 200:
|
|||
|
|
return json.loads(reponse.text)
|
|||
|
|
else:
|
|||
|
|
return None
|
|||
|
|
except Exception as e:
|
|||
|
|
print('http failed : %s' % e)
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 获取最新价
|
|||
|
|
def get_ticker(self, symbol):
|
|||
|
|
"""
|
|||
|
|
symbol必填，为all或交易对代码:btcusdt
|
|||
|
|
"""
|
|||
|
|
url = market_url + "ticker?symbol=" + str(symbol)
|
|||
|
|
return self.http_get_nosign(url)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 获取挂单
|
|||
|
|
def get_orderbook(self, symbol, depth=200):
|
|||
|
|
"""
|
|||
|
|
depth为选填项，默认为200
|
|||
|
|
"""
|
|||
|
|
url = market_url + "orderbook?symbol=" + symbol + "&depth=" + str(depth)
|
|||
|
|
return self.http_get_nosign(url)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 获取成交记录
|
|||
|
|
def get_trade(self, symbol, size=300):
|
|||
|
|
"""
|
|||
|
|
size:获取记录数量，按照时间倒序传输。默认300
|
|||
|
|
"""
|
|||
|
|
url = market_url + "trades?symbol=" + symbol + "&size=" + str(size)
|
|||
|
|
return self.http_get_nosign(url)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 查询账户余额
|
|||
|
|
def post_balance(self, dic):
|
|||
|
|
"""
|
|||
|
|
以字典形式传参
|
|||
|
|
apiid:可在coinbene申请,
|
|||
|
|
secret:个人密钥(请勿透露给他人),
|
|||
|
|
timestamp:时间戳,
|
|||
|
|
account:默认为exchange，
|
|||
|
|
"""
|
|||
|
|
url = trade_url + "balance"
|
|||
|
|
return self.http_post_sign(url, dic)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 下单
|
|||
|
|
def post_order_place(self, dic):
|
|||
|
|
"""
|
|||
|
|
以字典形式传参
|
|||
|
|
apiid,symbol,timestamp
|
|||
|
|
type:可选 buy-limit/sell-limit
|
|||
|
|
price:购买单价
|
|||
|
|
quantity:购买数量
|
|||
|
|
"""
|
|||
|
|
url = trade_url + "order/place"
|
|||
|
|
return self.http_post_sign(url, dic)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 查询委托
|
|||
|
|
def post_info(self, dic):
|
|||
|
|
"""
|
|||
|
|
以字典形式传参
|
|||
|
|
apiid,timestamp,secret,orderid
|
|||
|
|
"""
|
|||
|
|
url = trade_url + "order/info"
|
|||
|
|
return self.http_post_sign(url, dic)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 查询当前委托
|
|||
|
|
def post_open_orders(self, dic):
|
|||
|
|
"""
|
|||
|
|
以字典形式传参
|
|||
|
|
apiid,timestamp,secret,symbol
|
|||
|
|
"""
|
|||
|
|
url = trade_url + "order/open-orders"
|
|||
|
|
return self.http_post_sign(url, dic)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 撤单
|
|||
|
|
def post_cancel(self, dic):
|
|||
|
|
"""
|
|||
|
|
以字典形式传参
|
|||
|
|
apiid,timestamp,secret,orderid
|
|||
|
|
"""
|
|||
|
|
url = trade_url + "order/cancel"
|
|||
|
|
return self.http_post_sign(url, dic)
|