开源Python做的火币和ZB搬砖差价监控程序

Posted 2020-12-30 诺飞

火币和ZB网站域名经常换，页面也经常改，只保证发布文章时能用，将来要用自己改改URL地址一类的。

代码涉及到websocket和requests采集数据，使用pyquery解析HTML页面，colorama输出彩色文本。

功能：
自动采集火币和ZB同种货币，实时采集USDT和QC的OTC法币买卖价格，支持过滤异常价格OTC订单。
将数字货币价格换算成法币买卖价格后计算差价，差价超过1%时候，箭头提示搬砖方向。
可以自行配置监控的货币种类，黄色突出显示，达到指定差价时声音报警。
例如：

alert = {
    \'hsr\': {\'hb\': {\'enable\': True, \'profit\': 2}, \'zb\': {\'enable\': True, \'profit\': 2}},
    \'eth\': {\'hb\': {\'enable\': True, \'profit\': 2}, \'zb\': {\'enable\': True, \'profit\': 2}},
    }

可增加多行，按格式复制，enable设置True表示报警，设置False表示不报警。profit指定报警的差价。

安装方法：
下载Python 3安装
https://www.python.org/ftp/python/3.7.0/python-3.7.0-amd64.exe

然后命令行依次执行下列指令安装依赖包

pip install websocket-client
pip install requests
pip install pyquery
pip install colorama

最后执行

python bitcoin.py

即可运行，关闭的话，按Ctrl+C。

代码如下，比较简单，就不注释了。

from websocket import create_connection
from pyquery import PyQuery
from colorama import init, Fore

import requests
import socket
import gzip
import json

alert = {
    \'hsr\': {\'hb\': {\'enable\': True, \'profit\': 2}, \'zb\': {\'enable\': True, \'profit\': 2}},
    \'eth\': {\'hb\': {\'enable\': True, \'profit\': 2}, \'zb\': {\'enable\': True, \'profit\': 2}},
    }

otc = {\'usdt\': {\'sell\': 0.0, \'buy\': 0.0}, \'qc\': {\'sell\': 0.0, \'buy\': 0.0}}
markets = dict()

def run_client():
    ws = create_connection("wss://www.hbg.com/-/s/pro/ws")
    ws.send(json.dumps({\'sub\': \'market.overview\'}))
    while True:
        data = ws.recv()
        result = gzip.decompress(data)
        obj = json.loads(result)
        if \'ping\' in obj:
            ws.send(json.dumps({\'pong\': obj[\'ping\']}))
            refresh()
            output()
        elif \'ch\' in obj:
            for data in obj[\'data\']:
                if data[\'symbol\'].endswith(\'usdt\'):
                    key = data[\'symbol\'].replace(\'usdt\', \'\')
                    if not key in markets:
                        markets[key] = dict()
                    markets[key][\'hb\'] = {\'last\': data[\'close\']}

def run_forever():
    while True:
        try:
            run_client()
        except Exception as e:
            print(type(e), e.args)
        except KeyboardInterrupt:
            break

def refresh():
    try:
        url = "http://api.zb.cn/data/v1/allTicker"
        r = requests.get(url)
        obj = r.json()
        for key in obj:
            if key.endswith(\'qc\'):
                currency = key.replace(\'qc\', \'\')
                if currency == \'bcc\':
                    currency = \'bch\'
                if currency in markets:
                    markets[currency][\'zb\'] = {\'last\': float(obj[key][\'last\']), \'sell\': float(obj[key][\'sell\']), \'buy\': float(obj[key][\'buy\'])}
    except Exception as e:
        print(type(e), e.args)

def output():
    get_hb_price(1)
    get_hb_price(2)
    get_zb_price(1)
    get_zb_price(2)

    beep = \'\'
    print(\'┌───────┬─────────────────────────────────────┬───────┬─────────────────────────────────────┐\')
    print(\'│  USDT │  SELL: {0:.2f}    BUY: {1:.2f}            │   QC  │  SELL: {2:.3f}    BUY: {3:.3f}          │\'.format(\\
        otc[\'usdt\'][\'sell\'], otc[\'usdt\'][\'buy\'], otc[\'qc\'][\'sell\'], otc[\'qc\'][\'buy\']))
    print(\'├───────┼─────────────────────────────────────┴───────┴───────────────┬─────────────────────┤\')
    for key in markets:
        if \'zb\' in markets[key]:
            hb = markets[key][\'hb\'][\'last\']
            zb = markets[key][\'zb\'][\'last\']
            profit_hb = 100 / hb / otc[\'usdt\'][\'sell\'] * zb * otc[\'qc\'][\'buy\'] - 100
            profit_zb = 100 / otc[\'qc\'][\'sell\'] / zb * hb * otc[\'usdt\'][\'buy\'] - 100
            if key in alert:
                if (alert[key][\'hb\'][\'enable\'] and profit_hb > alert[key][\'hb\'][\'profit\']) or (alert[key][\'zb\'][\'enable\'] and profit_zb > alert[key][\'zb\'][\'profit\']):
                    beep = \'\\a\'
                color = Fore.YELLOW
            else:
                color = Fore.RESET
            if profit_hb > 1:
                dir = \'>>\'
            elif profit_zb > 1:
                dir = \'<<\'
            else:
                dir = \'  \'
            str = "│{0}  {1:^4} \\033[0m│{0}  {2:>6.2f}  {3:>8}  {4:>8}  {5:}  {6:>6.2f}  {7:>8}  {8:>8} \\033[0m│{0}  {9:>8}  {10:>8} \\033[0m│".format(\\
                color, key.upper(), profit_hb, format_price(hb * otc[\'usdt\'][\'sell\']), format_price(zb * otc[\'qc\'][\'buy\']), dir, \\
                profit_zb, format_price(hb * otc[\'usdt\'][\'buy\']), format_price(zb * otc[\'qc\'][\'sell\']), format_price(hb), format_price(zb))
            print(str)
    print(\'└───────┴─────────────────────────────────────────────────────────────┴─────────────────────┘\' + beep)

def get_hb_price(trade=1):
    if trade == 1:
        tradeType = \'buy\'
    else:
        tradeType = \'sell\'
    try:
        url = "https://otc-api.hbg.com/v1/data/trade-market?country=37&currency=1&payMethod=0&currPage=1&coinId=2&tradeType={0}&blockType=general&online=1".format(tradeType)
        r = requests.get(url)
        obj = r.json()
        for data in obj[\'data\']:
            if data[\'minTradeLimit\'] <= 20000 and data[\'maxTradeLimit\'] >= 5000:
                otc[\'usdt\'][tradeType] = data[\'price\']
                break
    except Exception as e:
        print(type(e), e.args)

def get_zb_price(trade=1):
    if trade == 1:
        tradeType = \'buy\'
    else:
        tradeType = \'sell\'
    try:
        url = "https://vip.zb.cn/otc/trade/qc_cny?type={0}".format(trade)
        s = requests.session()
        s.headers[\'Accept-Language\'] = \'zh-Hans-CN, zh-Hans; q=0.5\'
        s.headers[\'User-Agent\'] = \'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/64.0.3282.140 Safari/537.36 Edge/17.17134\'
        r = s.get(url)
        if len(r.text):
            d = PyQuery(r.text)
            tr = d(\'table.c2c-table\')(\'tr:gt(0) td.price\')
            for td in tr:
                price = float(td.text[:-5])
                l = td.find(\'span\').text.split(\'-\')
                if float(l[0]) <= 20000 and float(l[1]) >= 5000:
                    otc[\'qc\'][tradeType] = price
                    break
    except Exception as e:
        print(type(e), e.args)

def format_price(price):
    price = str(price)[:8]
    return float(price)

def main():
    init()
    socket.setdefaulttimeout(10)
    run_forever()

if __name__ == \'__main__\':
    main()