|
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
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
|
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
from matplotlib.ticker import FormatStrFormatter
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
def computeRSI (data, time_window):
diff = data.diff(1).dropna() # diff in one field(one day)
#this preservers dimensions off diff values
up_chg = 0 * diff
down_chg = 0 * diff
# up change is equal to the positive difference, otherwise equal to zero
up_chg[diff > 0] = diff[ diff>0 ]
# down change is equal to negative deifference, otherwise equal to zero
down_chg[diff < 0] = diff[ diff < 0 ]
# check pandas documentation for ewm
# https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.ewm.html
# values are related to exponential decay
# we set com=time_window-1 so we get decay alpha=1/time_window
up_chg_avg = up_chg.ewm(com=time_window-1 , min_periods=time_window).mean()
down_chg_avg = down_chg.ewm(com=time_window-1 , min_periods=time_window).mean()
rs = abs(up_chg_avg/down_chg_avg)
rsi = 100 - 100/(1+rs)
return rsi
class MainWidget(QWidget):
def __init__(self):
super().__init__()
self.df = yf.download('TQQQ', start='2022-01-01', end='2022-03-18') #yfinance 에서 TQQQ 데이터 다운로드
self.df['RSI'] = computeRSI(self.df['Close'], 14)
print(self.df)
#iday = intraday.loc['2019-11-06 15:00':'2019-11-06 16:00',:]
fig, axlist = mpf.plot(self.df, datetime_format='%b %d', type='candle', volume=True, mav=(20), style='starsandstripes', returnfig=True) # 그래프 그리기
axlist[0].yaxis.set_major_formatter(FormatStrFormatter('$%.0f')) #숫자 뒤에 $ 붙이기
canvas = FigureCanvas(fig)
canvas.draw()
self.text_edit = QTextEdit(self)
self.text_edit.setGeometry(0, 0, 1200, 600)
# 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, 1200, 600)
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, 1200, 600)
self.show()
if __name__ == "__main__":
app = QApplication(sys.argv)
myWindow = MyWindow()
myWindow.show()
app.exec_()
|
cs |
self.df['RSI'] = computeRSI(self.df['Close'], 14)
RSI 계산하여 데이터 프레임 컬럼을 하나 다시 추가하여 넣는다
반응형
'미국 주식 프로그램 만들기(eFriend expert)' 카테고리의 다른 글
| 7. 미국 주식 자동매매 프로그램 만들기 - 실시간 체결가 조회 (0) | 2022.04.03 |
|---|---|
| 6. 미국 주식 자동매매 프로그램 만들기 - 이동평균선, 일목균형표 추가 (0) | 2022.04.03 |
| 4. 미국 주식 자동매매 프로그램 만들기 - 그래프 옵션 추가 (0) | 2022.04.03 |
| 3. 미국 주식 자동매매 프로그램 만들기 - 데이터 받아오기, 그래프 그리기 (0) | 2022.04.03 |
| 2. 미국 주식 자동매매 프로그램 만들기 - 레이아웃 추가 (0) | 2022.04.02 |