|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
import sys
from PyQt5.QtWidgets import * #pip install PyQt5
from PyQt5.QtGui import *
from PyQt5.QAxContainer import *
from PyQt5.QtCore import *
import yfinance as yf #pip install yfinance
import mplfinance as mpf #pip install mplfinance
from matplotlib.backends.backend_qt5agg import FigureCanvas as FigureCanvas
koreanInvest_id = ""
koreanInvest_account = ""
koreanInvest_account_code = ""
koreanInvest_account_password = ""
koreanInvest_stock_buy_exchange = "NASD"
koreanInvest_stock_buy_code = "TQQQ"
koreanInvest_stock_buy_count = "1"
koreanInvest_stock_buy_price = "6.11"
koreanInvest_stock_buy_type = "00" # 00:지정가, 01:시장가, LOO:32, LOC:34
koreanInvest_stock_sell_exchange = "NASD"
koreanInvest_stock_sell_code = "TQQQ"
koreanInvest_stock_sell_count = "1"
koreanInvest_stock_sell_price = "6.11"
koreanInvest_stock_sell_type = "00" # 00:지정가, 01:시장가, LOO:32, LOC:34
class MainWidget(QWidget):
def __init__(self):
super().__init__()
self.df = yf.download('TQQQ', start='2020-04-01', end='2022-03-18') #yfinance 에서 TQQQ 데이터 다운로드
fig, axlist = mpf.plot(self.df, returnfig=True) # 그래프 그리기
canvas = FigureCanvas(fig)
canvas.draw()
self.text_edit = QTextEdit(self)
self.text_edit.setGeometry(0, 0, 800, 100)
# self.text_edit.setEnabled(False)
# 매수 버튼
self.btn_buy = QPushButton("Buy", self) # 버튼 텍스트
self.btn_buy.clicked.connect(self.btn_buy_clicked) # 클릭 시 실행할
# 매도 버튼
self.btn_sell = QPushButton("Sell", self) # 버튼 텍스트
self.btn_sell.clicked.connect(self.btn_sell_clicked) # 클릭 시 실행할
mainLayout = QVBoxLayout() #레이아웃 추가
sub1Layout = QHBoxLayout()
sub2Layout = QHBoxLayout()
chartLayout = QVBoxLayout()
mainLayout.addLayout(sub1Layout)
mainLayout.addLayout(sub2Layout)
mainLayout.addLayout(chartLayout)
sub1Layout.addWidget(self.text_edit)
sub2Layout.addWidget(self.btn_buy)
sub2Layout.addWidget(self.btn_sell)
chartLayout.addWidget(canvas)
self.setLayout(mainLayout)
self.setGeometry(0, 0, 800, 400)
def btn_buy_clicked(self):
# 주문 입력 및 전송
self.tran_os_us_buy = QAxWidget("ITGExpertCtl.ITGExpertCtlCtrl.1")
self.tran_os_us_buy.SetSingleData(0, koreanInvest_account) #종합계좌번호 ##계좌일련번호[8],)
self.tran_os_us_buy.SetSingleData(1, koreanInvest_account_code) #계좌상품코드 ##계좌 상품코드, GetAccount() 함수 이용(뒤 2자리)
self.tran_os_us_buy.SetSingleData(2, self.tran_os_us_buy.GetEncryptPassword(koreanInvest_account_password)) #계좌비밀번호(반드시 암호화 처리 필요)
self.tran_os_us_buy.SetSingleData(3, koreanInvest_stock_buy_exchange) #해외거래소코드 ##거래소코드 나스닥 NASD 뉴욕 NVSE 아맥스 AMEX
self.tran_os_us_buy.SetSingleData(4, koreanInvest_stock_buy_code) #상품번호 ##종목코드
self.tran_os_us_buy.SetSingleData(5, koreanInvest_stock_buy_count) #주문수량
self.tran_os_us_buy.SetSingleData(6, koreanInvest_stock_buy_price) #해외주문단가 ##주문가격
self.tran_os_us_buy.SetSingleData(7, "") #연락전화번호 #전화번호 필수값 아님
self.tran_os_us_buy.SetSingleData(8, "") #운용사지정주문번호 입력하지 않음
self.tran_os_us_buy.SetSingleData(9, "0") #주문서버구분코드 고정값 0
self.tran_os_us_buy.SetSingleData(10, koreanInvest_stock_buy_type) # 주문구분, 00: 지정가, 01: 시장가 LOO:32 LOC:34
#항목 설정
self.tran_os_us_buy.RequestData("OS_US_BUY")
self.text_edit.append("미국 매수주문 전송 [" + str(self.tran_os_us_buy.GetSendRqID()) + "] ")
self.tran_os_us_buy.ReceiveData.connect(self.tran_os_us_buy_receivedata) #이벤트 연결
def tran_os_us_buy_receivedata(self):
# 주문응답 이벤트
self.text_edit.append("미국 매수주문 응답 [" + str(self.tran_os_us_buy.GetSendRqID()) + "] ")
#결과 메세지
self.text_edit.append("[" + str(self.tran_os_us_buy.GetSendRqID()) + "] " + self.tran_os_us_buy.GetReqMsgCode() + " " + self.tran_os_us_buy.GetReqMessage())
self.text_edit.append("주문번호 : " + self.tran_os_us_buy.GetSingleData(1, 0))
self.text_edit.append("주문시간 : " + self.tran_os_us_buy.GetSingleData(2, 0))
def btn_sell_clicked(self):
# 주문 입력 및 전송
self.tran_os_us_sell = QAxWidget("ITGExpertCtl.ITGExpertCtlCtrl.1")
self.tran_os_us_sell.SetSingleData(0, koreanInvest_account) #계좌일련번호[8], GetAccount() 함수 이용(앞 8자리)
self.tran_os_us_sell.SetSingleData(1, koreanInvest_account_code) #계좌 상품코드, GetAccount() 함수 이용(뒤 2자리)
self.tran_os_us_sell.SetSingleData(2, self.tran_os_us_sell.GetEncryptPassword(koreanInvest_account_password)) #계좌비밀번호(반드시 암호화 처리 필요)
self.tran_os_us_sell.SetSingleData(3, koreanInvest_stock_sell_exchange) #거래소코드
self.tran_os_us_sell.SetSingleData(4, koreanInvest_stock_sell_code) #종목코드
self.tran_os_us_sell.SetSingleData(5, koreanInvest_stock_sell_count) #주문수량
self.tran_os_us_sell.SetSingleData(6, koreanInvest_stock_sell_price) #주문가격
self.tran_os_us_sell.SetSingleData(7, "") #전화번호
self.tran_os_us_sell.SetSingleData(8, "") #운용사지정주문번호
self.tran_os_us_sell.SetSingleData(9, "0") #주문서버구분코드
self.tran_os_us_sell.SetSingleData(10, koreanInvest_stock_sell_type) # 주문구분, 지정가:00 MOO:31 MOC:33 LOO:32 LOC:34
#항목 설정
self.tran_os_us_sell.RequestData("OS_US_SELL")
self.text_edit.append("미국 매도주문 전송 [" + str(self.tran_os_us_sell.GetSendRqID()) + "] ")
self.tran_os_us_sell.ReceiveData.connect(self.tran_os_us_sell_receivedata) #이벤트 연결
def tran_os_us_sell_receivedata(self):
# 주문응답 이벤트
self.text_edit.append("미국 매도주문 응답 [" + str(self.tran_os_us_sell.GetSendRqID()) + "] ")
#결과 메세지
self.text_edit.append("[" + str(self.tran_os_us_sell.GetSendRqID()) + "] " + self.tran_os_us_sell.GetReqMsgCode() + " " + self.tran_os_us_sell.GetReqMessage())
self.text_edit.append("주문번호 : " + self.tran_os_us_sell.GetSingleData(1, 0))
self.text_edit.append("주문시간 : " + self.tran_os_us_sell.GetSingleData(2, 0))
class MyWindow(QMainWindow):
def __init__(self):
super().__init__()
self.initUI()
def initUI(self):
self.setWindowTitle("한국투자증권-API")
mainWidget = MainWidget()
self.setCentralWidget(mainWidget)
self.setGeometry(0, 0, 800, 400)
self.show()
if __name__ == "__main__":
app = QApplication(sys.argv)
myWindow = MyWindow()
myWindow.show()
app.exec_()
|
cs |
*** 필요설치 모듈
pip install yfinance
pip install mplfinance
self.df = yf.download('TQQQ', start='2020-04-01', end='2022-03-18')
요놈 한 줄로 날짜 변경하여 데이터 프레임 형식으로 자료 받기 가능
보고 싶으면 print(df) 해보면 됨
가져온 데이터를 캔버스에 그리고 위젯으로 추가한 뒤 레이아웃 얹힘
그리는 건 걍 fig, axlist = mpf.plot(self.df, returnfig=True)
옵션 조절해서 축 값 설정해야하는데 어렵게 생각할가봐 옵션 값 안 넣음
(그래서 축 값이 이상함)
대신 returnfig=True 옵션을 안 넣으면 그림 반환을 안하기 때문에 필수
반응형
'미국 주식 프로그램 만들기(eFriend expert)' 카테고리의 다른 글
| 5. 미국 주식 자동매매 프로그램 만들기 - RSI 함수 추가 및 계산 (0) | 2022.04.03 |
|---|---|
| 4. 미국 주식 자동매매 프로그램 만들기 - 그래프 옵션 추가 (0) | 2022.04.03 |
| 2. 미국 주식 자동매매 프로그램 만들기 - 레이아웃 추가 (0) | 2022.04.02 |
| 1. 미국 주식 자동매매 프로그램 만들기 - 사고 팔기 기본 (0) | 2022.04.02 |
| 0. 미국 주식 자동매매 프로그램 만들기 - 환경구축 (0) | 2022.03.27 |