주식 알고리즘 만들기 - 7. 1) 보유종목 프로젝트

2022.10.11 - [Python] - 주식 알고리즘 만들기 - 6. 1) 파일 다루기 - 읽기

주식 알고리즘 만들기 - 6. 1) 파일 다루기 - 읽기

https://wikidocs.net/3523

• PyCharm: https://www.jetbrains.com/pycharm/download/

Download PyCharm: Python IDE for Professional Developers by JetBrains

 

 

이번 장에서는 지금까지 배운 파이썬 기초 문법을 활용해 간단한 프로젝트를 진행해보겠습니다.

 

프로그래밍을 처음 배우면 언어의 문법은 어느 정도 알 것 같은데, 막상 프로그램을 작성하려고 하면 어디서부터 어떻게 시작해야 할지 감이 오지 않습니다. 이러한 단계에서 더 발전하려면 꼭 코드를 직접 입력해 보면서 이번 장에서 소개하는 프로젝트를 완성해 봅시다.

 

전문 개발자는 보통 통합 개발 환경이라고 하는 IDE(Integrated Development Environment)를 이용해 프로그램을 개발합니다. 게임이나 음악 같은 것도 장비가 좋은 환경에서 발휘한 실력이 더 빛이 날 때가 있습니다.

이처럼 IDE는 변수나 함수 이름을 자동으로 완성하는 기능을 제공하고 있어 개발자가 더욱 빠르게 코드를 작성할 수 있게 도와 줍니다. 이제 여러분도 IDE를 사용해 파이썬 프로그램을 개발해봅시다.

 

아래 포스트에서 원하는 IDE를 선택해서 개발해봅시다.

• PTVS: http://microsoft.github.io/PTVS/
• PyDev: http://www.pydev.org/
• PyCharm: https://www.jetbrains.com/pycharm/download/

https://post.naver.com/viewer/postView.naver?volumeNo=33785623&memberNo=856655&vType=VERTICAL

 

위 포스트에서 초기설정까지 마치셧다면 이제 프로젝트를 시직하겠습니다. 

 

1) 설계

 

관심종목이라는 프로그램을 프로젝트할지 초기 설계를 해보도록 하겠습니다.

 

1. 관심종목이라는 클래스를 만들겠습니다.
2. 관심종목에서는 종목명, 현재가, 내평단, 수익률이라는 데이터를 넣겠습니다.
3. 이 데이터를 FILE로 저장하도록 하겠습니다.

 

2)실행

설계를 이제 실행할 수 있도록 코드를 작성해 보겠습니다.

 

처음에 클래스를 생성하하고 입력 값을 변수로 저장하는 함수를 만들어보겠습니다. 

기억을 되새겨보면서 만들어보죠

 

class itr_stock :
    def __init__(self, name, cur_price, my_price):
        self.name =name
        self.cur_price = cur_price
        self.my_price = my_price
        self.rate = (1-(my_price/cur_price))*100

기억이 나시나요? 자, 이제 입력값을 변수에 저장했으니 우리는 이것을 화면에 출력하는 함수를 만들도록 하겠습니다.

 

    def print_info(self):
        print("종목명: ", self.name)
        print("현재가: ", self.cur_price)
        print("평단가: ", self.my_price)
        print("수익률: ", self.rate)

여기까지는 여태까지 개념을 배우면서 코드를 작성한 기억이 있을겁니다. 이제 무엇이 필요할까요?

이게 정상적으로 작동하는지 인스턴스를 생성해서 입력값을 넣어 봐야겠죠?

 

def run():
    kakaopay = itr_stock('카카오페이', 36350 , 44019)
    kakaopay.print_info()

run()이라는 함수를 만들어서 간단히 run()만 입력하면 되도록 설계했습니다.

자, 이제 실행해봅시다.

 

잘 실행되는지 확인을 했습니다.  조금 채울만한 부분이 있는지 확인을 해보면, 화면 출력을 알아 보기 쉽게 단위를 설정하도록하겠습니다. 검토까지 끝마치고 이제는 input으로 사용자에게 직접 입력값을 하나씩 받아보도록 하겠습니다.

 

class itr_stock :
    def __init__(self, name, cur_price, my_price, rate):
        self.name =name
        self.cur_price = cur_price
        self.my_price = my_price
        self.rate = rate
        
    def print_info(self):
        print("종목명: ", self.name)
        print("현재가(원): ", self.cur_price)
        print("평단가(원): ", self.my_price)
        print("수익률(%): " , self.rate)

def user_input():
    name = input("종목명: ")
    cur_price = float(input("현재가(원): "))
    my_price = float(input("평단가(원): "))
    rate= (cur_price/my_price -1)*100
    myStock = itr_stock(name, cur_price, my_price, rate)
    return myStock

def print_menu():
    print("---보유잔고---")
    print("1. 종목 입력")
    print("2. 종목 출력")
    print("3. 종목 삭제")
    print("4. 종료")
    menu = input("메뉴선택: ")
    return int(menu)

def print_myStock(stock_list):
    for myStock in stock_list:
        myStock.print_info()

def run():
    stock_list = []
    while 1:
        menu = print_menu()
        if menu == 1:
            myStock = user_input()
            stock_list.append(myStock)
        elif menu == 2:
          print_myStock(stock_list)
        elif menu == 4:
            break
run()