5.7.2 매도 조건 확인하기

그럼 매도 대상인지 확인하는 check_sell_signal 함수를 만들어 보겠습니다. 먼저 함수의 매개변수로 종목 코드를 전달받고 universe 딕셔너리에 접근하여 어떤 데이터를 가지고 있는지 확인해 보겠습니다. universe에 들어 있는 데이터를 universe_item 변수를 사용하여 확인할 수 있도록 다음과 같이 코드를 만들고 출력까지 해 보겠습니다. RSIStrategy.py 파일의 set_universe_real_time 함수에 이어서 다음 코드를 작성합니다.

RSIStrategy.py

(...)
def check_sell_signal(self, code):
    universe_item = self.universe[code]
    print(universe_item)
    print(universe_item.keys())

main.py 파일을 실행하여 check_sell_signal 함수를 호출하면 실행 결과는 다음과 같습니다. universe에 저장된 정보를 확인해 보면, 종목 코드 이름인 code_name과 일봉 데이터, price_df가 들어 있습니다. 여기서는 유니버스를 구성하는 종목 중 하나인 ‘메리츠화재’를 미리 매수했기 때문에 check_sell_signal 함수가 실행되어 보유 종목의 price_df를 확인할 수 있습니다. 하지만 아직 보유 중인 종목이 없으면 아래 결과를 확인하기 어려울 수 있습니다.