반응형
39강에서 간단히 매수/매도에 관련된 GUI를 업데이트하였습니다. 관련 내용을 종목 추가 클릭 시 Qtable Widget에 업데이트되도록 코딩해 보겠습니다. 참고로 DB 관련된 내용도 다 업데이트되어야 합니다. ㅜ.ㅜ
[유튜브 강의, 링크]와 같이 보시면 많은 도움이 되실 겁니다.
1. 39강 GUI에 매수수량 업데이트
매수가격/익절가격/손절가격만 업데이트하고 매수 가격 대비 매수수량을 넣지 않아 아래와 같이 추가하였습니다. objecName은 n_o_stock이라고 하였습니다.
2. Text Edit 내용 QTable Widget에 추가하기
32강에서 자동매매 선정시 수행했던 searchItem2 함수에 아래와 같은 내용을 추가합니다.
열 이름에 4 : 매수가, 5 : 매수수량, 6 : 익절가, 7 : 손절가를 입력합니다. 그리고 각 열에 Text Edit 내용을 업데이트합니다.
- buylast라는 Qtable Widget의 4열에 다음과 같은 내용을 입력합니다. buy_price라는 Text Widget에 입력된 매수가를 toPlainText()라는 함수를 사용하여 Qtable Widget 입력합니다.
- 나머지 n_o_stock, profit, loss 동일하게 입력합니다.(동일 문구이므로 설명 생략하겠습니다.)
self.buylast.setItem(row_count, 4, QTableWidgetItem(str(self.buy_price.toPlainText())))
3. DB 관련 내용 똑같이 업데이트 하기
37강에서 선정 종목 데이터 베이스를 하였는데 4가지 항목이 더 늘어났죠? 37강에서 자세히 설명된 부분이므로 각 코드만 간략히 설명드리겠습니다.
- 종목 저장 : Save_selected_code(self)에 mesu부터 loss까지 4가지 내용을 아래와 같이 저장합니다.
- 저장 시 f.write에 %s\t도 4개 추가해 주세요.
- 종목 로드 : Load_code
- 테스트 파일 읽을 때 mesu부터 loss까지 4개의 파일을 더 읽고 Getanal_code에 입력합니다.
- 앞서 와 동일하게 열 이름을 매수가부터 손절가까지 4개 더 추가합니다.
- 각 열에 관련 데이터를 Getanal_code에서 가져와 입력합니다.
4. 요약 및 코드
앞선 강의에서 GUI 업로드에 관하여 자세히 설명드렸기에 추가되는 부분은 간략히 설명하고 넘어가겠습니다. 이해되지 않으시는 분들은 질문하시거나 차 후에 업로드되는 유튜브 강의를 시청해 주십시오.
이번 강의에서는 매수/매도에 관련된 내용을 Qtable Widget에 입력하고, DB 저장 및 로드 시에도 늘어난 내용만큼 데이터를 수정해줬습니다. 관련 코드는 아래와 같습니다.
반응형
import os # 현재 디렉토리 확인 기능
import sys # system specific parameters and functions : 파이썬 스크립트 관리
from PyQt5.QtWidgets import * # GUI의 그래픽적 요소를 제어 하단의 terminal 선택, activate py37_32, pip install pyqt5, 전부다 y
from PyQt5 import uic # ui 파일을 가져오기위한 함수
from PyQt5.QtCore import * # eventloop/스레드를 사용 할 수 있는 함수 가져옴.
################# 부가 기능 수행(일꾼) #####################################
from kiwoom import Kiwoom # 키움증권 함수/공용 방 (싱글턴)
from Qthread_1 import Thread1 # 계좌평가잔고내역 가져오기
from Qthread_2 import Thread2 # 계좌 관리
#=================== 프로그램 실행 프로그램 =========================#
form_class = uic.loadUiType("ALBA.ui")[0] # 만들어 놓은 ui 불러오기
class Login_Machnine(QMainWindow, QWidget, form_class): # QMainWindow : PyQt5에서 윈도우 생성시 필요한 함수
def __init__(self, *args, **kwargs): # Main class의 self를 초기화 한다.
print("Login Machine 실행합니다.")
super(Login_Machnine, self).__init__(*args, **kwargs)
form_class.__init__(self) # 상속 받은 from_class를 실행하기 위한 초기값(초기화)
self.setUI() # UI 초기값 셋업 반드시 필요
### 초기 셋팅 : 계좌평가잔고내역
self.label_11.setText(str("총매입금액"))
self.label_12.setText(str("총평가금액"))
self.label_13.setText(str("추정예탁자산"))
self.label_14.setText(str("총평가손익금액"))
self.label_15.setText(str("총수익률(%)"))
#### 기타 함수
self.login_event_loop = QEventLoop() # 이때 QEventLoop()는 block 기능을 가지고 있다.
####키움증권 로그인 하기
self.k = Kiwoom() # Kiwoom()을 실행하며 상속 받는다. Kiwoom()은 전지적인 아이다.
self.set_signal_slot() # 키움로그인을 위한 명령어 전송 시 받는 공간을 미리 생성한다.
self.signal_login_commConnect()
#####이벤트 생성 및 진행
self.call_account.clicked.connect(self.c_acc) # 계좌정보가져오기
self.acc_manage.clicked.connect(self.a_manage) # 계좌정보가져오기
################# 부가기능 1 : 종목선택하기 새로운 종목 추가 및 삭제
self.k.kiwoom.OnReceiveTrData.connect(self.trdata_slot) # 키움서버 데이터 받는 곳
self.additmelast.clicked.connect(self.searchItem2) # 종목 추가
self.Deletcode.clicked.connect(self.deltecode) # 종목 삭제
################# 부가기능 2 : 데이터베이스화 하기, 저장, 삭제, 불러오기
self.Getanal_code = [] # 불러온 파일 저장
self.Save_Stock.clicked.connect(self.Save_selected_code) # 종목 저장
self.Del_Stock.clicked.connect(self.delet_code) # 종목 삭제
self.Load_Stock.clicked.connect(self.Load_code) # 종목 불러오기
####################
def Load_code(self):
if os.path.exists("dist/Selected_code.txt"):
f = open("dist/Selected_code.txt", "r", encoding="utf8")
lines = f.readlines() # 여러 종목이 저장되어 있다면 모든 항목을 가져온다.
for line in lines:
if line != "": # 만약에 line이 비어 있지 않다면
ls = line.split("\t") # \t(tap)로 구분을 지어 놓는다.
t_code = ls[0]
t_name = ls[1]
curren_price = ls[2]
dept = ls[3]
mesu = ls[4]
n_o_stock = ls[5]
profit = ls[6]
loss = ls[7].split("\n")[0]
self.Getanal_code.append([t_code, t_name, curren_price, dept, mesu, n_o_stock, profit, loss])
f.close()
column_head = ["종목코드", "종목명", "현재가", "신용비율", "매수가", "매수수량", "익절가", "손절가"]
colCount = len(column_head)
rowCount = len(self.Getanal_code)
self.buylast.setColumnCount(colCount) # 행 갯수
self.buylast.setRowCount(rowCount) # 열 갯수 (종목 수)
self.buylast.setHorizontalHeaderLabels(column_head) # 행의 이름 삽입
self.buylast.setSelectionMode(QAbstractItemView.SingleSelection)
for index in range(rowCount):
self.buylast.setItem(index, 0, QTableWidgetItem(str(self.Getanal_code[index][0])))
self.buylast.setItem(index, 1, QTableWidgetItem(str(self.Getanal_code[index][1])))
self.buylast.setItem(index, 2, QTableWidgetItem(str(self.Getanal_code[index][2])))
self.buylast.setItem(index, 3, QTableWidgetItem(str(self.Getanal_code[index][3])))
self.buylast.setItem(index, 4, QTableWidgetItem(str(self.Getanal_code[index][4])))
self.buylast.setItem(index, 5, QTableWidgetItem(str(self.Getanal_code[index][5])))
self.buylast.setItem(index, 6, QTableWidgetItem(str(self.Getanal_code[index][6])))
self.buylast.setItem(index, 7, QTableWidgetItem(str(self.Getanal_code[index][7])))
def Save_selected_code(self):
for row in range(self.buylast.rowCount()):
code_n = self.buylast.item(row, 0).text()
name = self.buylast.item(row, 1).text().strip()
price = self.buylast.item(row, 2).text()
dept = self.buylast.item(row, 3).text()
mesu = self.buylast.item(row, 4).text()
n_o_stock = self.buylast.item(row, 5).text()
profit = self.buylast.item(row, 6).text()
loss = self.buylast.item(row, 7).text()
f = open("dist/Selected_code.txt", "a",encoding="utf8") # "a" 달아 쓴다. "w" 덮어 쓴다. files라느 파이썬 페키지 볼더를 만든다.
f.write("%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\n" % (code_n, name, price, dept, mesu, n_o_stock, profit, loss)) # t는 tap을 의미한다.
f.close()
def delet_code(self):
if os.path.exists("dist/Selected_code.txt"):
os.remove("dist/Selected_code.txt")
def deltecode(self):
x = self.buylast.selectedIndexes() # 리스트로 선택된 행번호와 열번호가 x에 입력된다.
self.buylast.removeRow(x[0].row())
def searchItem2(self): # 종목추가시 사용됨.
itemName = self.searchItemTextEdit2.toPlainText()
if itemName != "":
for code in self.k.All_Stock_Code.keys(): # 포트폴리오에 저장된 코드들을 실시간 등록
# 주식체결 정보 가져오기(틱 데이터) : 현재가, 전일대비, 등락률, 매도호가, 매수호가, 거래량, 누적거래량, 고가, 시가, 저가
if itemName == self.k.All_Stock_Code[code]['종목명']:
self.new_code = code
column_head = ["종목코드", "종목명", "현재가", "신용비율", "매수가", "매수수량", "익절가", "손절가"]
colCount = len(column_head)
row_count = self.buylast.rowCount()
self.buylast.setColumnCount(colCount) # 행 갯수
self.buylast.setRowCount(row_count+1) # colum_haed가 한 행을 잡아 먹는다. 실제 입력 되는 값은 1행 부터이다.
self.buylast.setHorizontalHeaderLabels(column_head) # 행의 이름 삽입
self.buylast.setItem(row_count, 0, QTableWidgetItem(str(self.new_code))) # 실제 입력값은 1행부터이나 0부터 들어가야 된다.
self.buylast.setItem(row_count, 1, QTableWidgetItem(str(itemName)))
self.buylast.setItem(row_count, 4, QTableWidgetItem(str(self.buy_price.toPlainText())))
self.buylast.setItem(row_count, 5, QTableWidgetItem(str(self.n_o_stock.toPlainText())))
self.buylast.setItem(row_count, 6, QTableWidgetItem(str(self.profit_price.toPlainText())))
self.buylast.setItem(row_count, 7, QTableWidgetItem(str(self.loss_price.toPlainText())))
self.getItemInfo(self.new_code)
def getItemInfo(self, new_code):
self.k.kiwoom.dynamicCall("SetInputValue(QString, QString)", "종목코드", new_code)
self.k.kiwoom.dynamicCall("CommRqData(QString, QString, int, QString)", "주식기본정보요청", "opt10001", 0, "100")
def setUI(self):
self.setupUi(self) # UI 초기값 셋업
def set_signal_slot(self):
self.k.kiwoom.OnEventConnect.connect(self.login_slot) # 내가 알고 있는 login_slot에다가 특정 값을 던져 준다.
def signal_login_commConnect(self):
self.k.kiwoom.dynamicCall("CommConnect()") # 네트워크적 서버 응용프로그램에 데이터를 전송할 수 있게 만든 함수
self.login_event_loop.exec_() # 로그인이 완료될 때까지 계속 반복됨. 꺼지지 않음.
def login_slot(self, errCode):
if errCode == 0:
print("로그인 성공")
self.statusbar.showMessage("로그인 성공")
self.get_account_info() # 로그인시 계좌정보 가져오기
elif errCode == 100:
print("사용자 정보교환 실패")
elif errCode == 101:
print("서버접속 실패")
elif errCode == 102:
print("버전처리 실패")
self.login_event_loop.exit() # 로그인이 완료되면 로그인 창을 닫는다.
def get_account_info(self):
account_list = self.k.kiwoom.dynamicCall("GetLoginInfo(String)", "ACCNO")
for n in account_list.split(';'):
self.accComboBox.addItem(n)
def c_acc(self):
print("선택 계좌 정보 가져오기")
##### 1번 일꾼 실행
h1 = Thread1(self)
h1.start()
def a_manage(self):
print("계좌 관리")
h2 = Thread2(self)
h2.start()
def trdata_slot(self, sScrNo, sRQName, sTrCode, sRecordName, sPrevNext):
if sTrCode == "opt10001":
if sRQName == "주식기본정보요청":
currentPrice = abs(int(self.k.kiwoom.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, "현재가")))
D_R = (self.k.kiwoom.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, "신용비율")).strip()
row_count = self.buylast.rowCount()
self.buylast.setItem(row_count - 1, 2, QTableWidgetItem(str(currentPrice)))
self.buylast.setItem(row_count - 1, 3, QTableWidgetItem(str(D_R)))
if __name__=='__main__': # import된 것들을 실행시키지 않고 __main__에서 실행하는 것만 실행 시킨다.
# 즉 import된 다른 함수의 코드를 이 화면에서 실행시키지 않겠다는 의미이다.
app = QApplication(sys.argv) # PyQt5로 실행할 파일명을 자동으로 설정, PyQt5에서 자동으로 프로그램 실행
CH = Login_Machnine() # Main 클래스 myApp으로 인스턴스화
CH.show() # myApp에 있는 ui를 실행한다.
app.exec_() # 이벤트 루프
반응형
'주식 자동매매 강의 > 기초반(모든 코딩의 뿌리)' 카테고리의 다른 글
| 주식자동매매 41강. 실시간 자동매매 구현(3), 자동 매매 시작 GUI(Push Button) 및 스레드 스크립트 생성 (9) | 2022.06.07 |
|---|---|
| [특강] 주식자동매매 프로그램 GUI 우측 정렬, 천(1000) 단위 구분 콤마(comma) 설정하기 (10) | 2022.06.05 |
| 주식자동매매 39강. 실시간 자동매매 구현(1), KOA Studio등 실시간 매매 개념 잡기,GUI 구성하기 (5) | 2022.06.01 |
| 주식자동매매 38강. 선정 종목 데이터베이스화 하기(3), 저장된 텍스트 파일 삭제 및 코드 공개 (0) | 2022.05.31 |
| 주식자동매매 38강. 선정 종목 데이터베이스화 하기(2), 저장된 Text 불러와 GUI에 전시하기 (4) | 2022.05.29 |
