2021년 3월 19일 금요일

[파이썬] 관심주식종목 목록에 종목 추가/삭제가능 및 선택한 주가정보을 조회하기 (네이버 증권 사이트 이용)

 

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

 

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

 

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

==========

 

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

 

댓글 없음:

댓글 쓰기