반응형

이번에는 파이썬을 이용하여 특정 주식종목 목록을 입력할 수 있고 또 목록에서 주식종목 삭제도 할 수 있게 새롭게 코드를 작성했습니다.

 

지난번에는 일일이 주식종목 코드를 코드상에 직접 입력하다보니 새로운 주식종목을 조회하기 위해서는 엄청 불편합니다.

 

import requests
from bs4 import BeautifulSoup
 
stock_list = []   # 관심종목 목록
stock_item = ["", ""]   # 추가할 관심종목 입력
sel = ""
st_no = ""
st_name = ""   # 관심종목 이름 입력
st_code = ""   # 관심족옥 코드 입력
 
url = ""
url_temp = "https://finance.naver.com/item/main.nhn?code="
 
while True:
    print ("==========")
    print ("97 : 관심종목 추가")
    print ("98 : 관심종목 삭제")
    print ("99 : 종료")
    print ("종목코드 번호 : 선택한 종목 주가정보 조회")
    print ("==========")
 
    i = 0
    for i in range ( len(stock_list) ):
        print (i, " - ", stock_list[i][0], stock_list[i][1])
        i = i + 1
 
    print ("==========")
 
    sel = input ("원하는 항목을 입력하세요 : ")
    sel_no = int(sel)
 
    if sel == "97":   # 새로운 주식종목 추가
        st_name = input ("종목이름 : ")
        st_code = input ("종목코드 : ")
        stock_item[0] = st_name
        stock_item[1] = st_code
        stock_list.append(list(stock_item))
 
    elif sel == "98":   # 지정된 주식종목 삭제
        st_no = input ("종목항목번호 : ")
        del stock_list[ int(st_no) ]
 
    elif sel == "99":   # 종료
        break
 
    else:   # 선택한 주식종목 주가정보 보여주기
        url = url_temp + stock_list[sel_no][1]
 
        response = requests.get (url)
 
        if response.status_code == 200:
            html = response.text
            soup = BeautifulSoup (html, 'html.parser')
            print ("----------")
            print (soup.dl.text)
            print ("----------")

 

여기에 작성한 코드는 파이썬에서 기본적인 키워드를 주로 사용하여 작성하였으므로 보기에 어렵지 않을 것입니다.

 

처음 실행하면 등록된 관심종목이 없으므로 아래와 같이 나올 것입니다.

 

==========
98 : 관심종목 추가
99 : 종료
종목코드 번호 : 선택한 종목 주가정보 조회
==========
==========
원하는 항목을 입력하세요 :

 

여기서 종목추가를 위해 “97”을 입력하고, 추가할 종목이름과 코드를 입력합니다.

 

그러면 아래와 같이 화면이 나오는데, 예제로 네이버를 입력해보았습니다.

 

원하는 항목을 입력하세요 : 97
종목이름 : 네이버
종목코드 : 035420
==========
97 : 관심종목 추가
98 : 관심종목 삭제
99 : 종료
종목코드 번호 : 선택한 종목 주가정보 조회
==========
0  -  네이버 035420
==========

 

다시 추가로 삼성전자 현대차를 입력해보겠습니다.

 

==========
0  -  네이버 035420
1  -  삼성전자 005930
2  -  현대차 005380
==========

 

여기에서 삼성전자를 삭제하면 아래 그림과 같이 나옵니다.

 

원하는 항목을 입력하세요 : 98
종목항목번호 : 1
==========
97 : 관심종목 추가
98 : 관심종목 삭제
99 : 종료
종목코드 번호 : 선택한 종목 주가정보 조회
==========
0  -  네이버 035420
1  -  현대차 005380
==========

 

어디서든 해당 종목에 대한 목록번호를 입력하면 해당 종목의 주가정보를 볼 수 있습니다.

반응형
Posted by HLIFEINFO
,