Python 프로젝트 » 비트코인 자동매매

h1. 실시간 호가창

h3. 소스코드 

<pre>

import sys

import time

import pyupbit

import pybithumb

from PyQt5 import uic

from PyQt5.QtWidgets import QWidget, QApplication, QTableWidgetItem, QProgressBar

from PyQt5.QtCore import Qt, QThread, pyqtSignal

class OrderbookWorker(QThread):

    dataSent = pyqtSignal(dict)

    def __init__(self, ticker):

        super().__init__()

        self.ticker = ticker

        self.alive = True

    def run(self):

        while self.alive:

            # data = pybithumb.get_orderbook(self.ticker, limit=10)

            data = pyupbit.get_orderbook(self.ticker)

            time.sleep(0.1)

            self.dataSent.emit(data)

    def close(self):

        self.alive = False

class OrderbookWidget(QWidget):

    def __init__(self, parent=None, ticker="KRW-BTC"): # 업비트 : KRW-BTC | 빗썸 : BTC

        super().__init__(parent)

        uic.loadUi("resource/orderbook.ui", self)

        self.setWindowTitle("실시간 호가창")

        self.ticker = ticker

        for i in range(self.tableBids.rowCount()):

            # 매도 호가

            item_0 = QTableWidgetItem(str(""))

            item_0.setTextAlignment(Qt.AlignRight | Qt.AlignVCenter)

            self.tableAsks.setItem(i, 0, item_0)

            item_1 = QTableWidgetItem(str(""))

            item_1.setTextAlignment(Qt.AlignRight | Qt.AlignVCenter)

            self.tableAsks.setItem(i, 1, item_1)

            item_2 = QProgressBar(self.tableAsks)

            item_2.setAlignment(Qt.AlignRight | Qt.AlignVCenter)

            item_2.setStyleSheet("""

            QProgressBar {background-color : rgba(0, 0, 0, 0%); border : 1}

            QProgressBar::Chunk {background-color : rgba(255, 0, 0, 50%); border : 1}

            """)

            self.tableAsks.setCellWidget(i, 2, item_2)

            #매수 호가

            item_0 = QTableWidgetItem(str(""))

            item_0.setTextAlignment(Qt.AlignRight | Qt.AlignVCenter)

            self.tableBids.setItem(i, 0, item_0)

            item_1 = QTableWidgetItem(str(""))

            item_1.setTextAlignment(Qt.AlignRight | Qt.AlignVCenter)

            self.tableBids.setItem(i, 1, item_1)

            item_2 = QProgressBar(self.tableBids)

            item_2.setAlignment(Qt.AlignRight | Qt.AlignVCenter)

            item_2.setStyleSheet("""

            QProgressBar {background-color : rgba(0, 0, 0, 0%); border : 1}

            QProgressBar::Chunk {background-color : rgba(0, 255, 0, 40%); border : 1}

            """)

            self.tableBids.setCellWidget(i, 2, item_2)

        self.ow = OrderbookWorker(self.ticker)

        # self.ow.dataSent.connect(self.updateBithumbData) # 빗썸 실시간 호가

        self.ow.dataSent.connect(self.updateUpbitData) # 업비트 실시간 호가

        self.ow.start()

    def updateBithumbData(self, data):

        # print(data)

        tradingBidValues = []

        for v in data['bids']:

            tradingBidValues.append(int(v['price'] * v['quantity']))

        tradingAskValues = []

        for v in data['asks'][::-1]:

            tradingAskValues.append(int(v['price'] * v['quantity']))

            # print(int(v['price'] * v['quantity']))

        maxtradingValue = max(tradingBidValues + tradingAskValues)

        for i, v in enumerate(data['asks'][::-1]):

            item_0 = self.tableAsks.item(i, 0)

            item_0.setText(f"{v['price']:,}")

            item_1 = self.tableAsks.item(i, 1)

            item_1.setText(f"{v['quantity']:,}")

            item_2 = self.tableAsks.cellWidget(i, 2)

            item_2.setRange(0, maxtradingValue)

            item_2.setFormat(f"{tradingAskValues[i]:,}")

            item_2.setValue(tradingAskValues[i])

        for i, v in enumerate(data['bids'][::-1]):

            item_0 = self.tableBids.item(i, 0)

            item_0.setText(f"{v['price']:,}")

            item_1 = self.tableBids.item(i, 1)

            item_1.setText(f"{v['quantity']:,}")

            item_2 = self.tableBids.cellWidget(i, 2)

            item_2.setRange(0, maxtradingValue)

            item_2.setFormat(f"{tradingBidValues[i]:,}")

            item_2.setValue(tradingBidValues[i])

    def updateUpbitData(self, data):

        tradingBidValues = []

        tradingBidList = []

        tradingAskValues = []

        tradingAskList = []

        # 값 10개만 생성

        for i in range(0, 10):

            tradingBidValues.append(int(data['orderbook_units'][i]['bid_price'] * data['orderbook_units'][i]['bid_size']))

            tradingAskValues.append(int(data['orderbook_units'][i]['ask_price'] * data['orderbook_units'][i]['ask_size']))

            # 빗썸 예제와 동일한 구조를 가지도록 데이터 가공(매수)

            tradingBidsDict = {"price" : data['orderbook_units'][i]['bid_price'], "size" : data['orderbook_units'][i]['bid_size']}

            tradingBidList.append(tradingBidsDict)

            # 빗썸 예제와 동일한 구조를 가지도록 데이터 가공(매도)

            tradingAsksDict = {"price" : data['orderbook_units'][i]['ask_price'], "size" : data['orderbook_units'][i]['ask_size']}

            tradingAskList.append(tradingAsksDict)

            maxtradingValue = max(tradingBidValues + tradingAskValues)

        # 빗썸 예제와 동일한 구조를 가지도록 데이터 가공(Dictionary)

        resultData = {"bids" : tradingBidList, "asks" : tradingAskList}

        for i, v in enumerate(resultData['asks'][::-1]):

            item_0 = self.tableAsks.item(i, 0)

            item_0.setText(f"{v['price']:,}")

            item_1 = self.tableAsks.item(i, 1)

            # item_1.setText(f"{v['size']:,}") - 해당방식은 e형식 출력이 됨.

            item_1.setText(f"{format(v['size'],'f')}") # format을 f로 설정해야 소수로 표기해줌

            item_2 = self.tableAsks.cellWidget(i, 2)

            item_2.setRange(0, maxtradingValue)

            item_2.setFormat(f"{tradingAskValues[i]:,}")

            item_2.setValue(tradingAskValues[i])

        for i, v in enumerate(resultData['bids']):

            item_0 = self.tableBids.item(i, 0)

            item_0.setText(f"{v['price']:,}")

            item_1 = self.tableBids.item(i, 1)

            # item_1.setText(f"{v['size']:,}") - 해당방식은 e형식 출력이 됨.

            item_1.setText(f"{format(v['size'],'f')}") # format을 f로 설정해야 소수로 표기해줌

            item_2 = self.tableBids.cellWidget(i, 2)

            item_2.setRange(0, maxtradingValue)

            item_2.setFormat(f"{tradingBidValues[i]:,}")

            item_2.setValue(tradingBidValues[i])

    def closeEvent(self, data):

        self.ow.close()

if __name__ == "__main__":

    app = QApplication(sys.argv)

    ow = OrderbookWidget()

    ow.show()

    exit(app.exec_())

</pre>

h3. 결과

!clipboard-202502121730-ktzkj.png!