자동매매프로그램 UI 정리 및 종목명 변환코드 추가

오늘은 자동매매 프로그램의 UI를 정리하고 종목명을 종목코드로 변환하는 코드를 추가했다.

 

 

 

UI 정리

---

UI정리

'종목'부분을 '종목 코드'와 '종목명'으로 나누어서 설정했으며 수동 주문 탭을 묶어서 정리했다.

 

 

종목명 변환코드 작성

---

총 3가지 고민을 했는데

• 새로운 창을 만들어서 pushButton을 활용하여 종목명을 종목코드로 변환하는 코드를 만들기
• 새로운 창을 만들어서 text가 변할 시 자동으로 변환되는 코드 만들기
• 기존의 창에 종목명 입력시 종목코드로 변환되고 종목코드 입력 시 종목명으로 변환되는 코드 만들기

그중에서 3번이 가장 깔끔할 것 같아서 3번으로 진행했다.

 

pytrader.py 부분에 다음과 같은 코드를 추가했다.

   code_name_list = []
   code_list = self.kiwoom.get_code_list_by_market("0")
   for i in range(len(code_list)):
	   code_name_list.insert(i, self.kiwoom.get_master_code_name(code_list[i]))
   self.code_name_dict = dict(zip(code_name_list, code_list))

kiwoom을 통해 만들어놓은 get_master_code_name을 활용하여서 code_name_list(코드 이름)을 만들었고

 

zip함수를 이용하여서 dict로 바꾸어주었다.

self.lineEdit.textChanged.connect(self.code_changed)
def code_changed_3(self):
  code = self.lineEdit_2.text()
  name = self.code_name_dict.get(str(code))
  self.lineEdit.setText(name)

그리고 textChanged를 code_changed_3과 연결하여 종목코드를 반환하는 함수를 작성하였다.

 

삼성전자 예시

...

 

처음으로 책에 없는 내용을 혼자서 해결했다!

 

아직 많이 부족하지만 점점 완성시켜야겠다.