더북(TheBook)

이 코드는 get_code_list_by_market으로 KOSPI, KOSDAQ 전체 종목 코드를 가져온 후 반복문을 수행하여 코드별로 get_master_code_name의 호출 결과인 종목명을 얻어 옵니다. 이 종목명이 universe_list에 속하는 종목명과 같을 때 ‘종목 코드-종목명’을 미리 만들어 놓은 universe 딕셔너리에 저장한 후 최종적으로 데이터프레임으로 만들어 데이터베이스에 저장합니다.

universe_list에 들어 있는 종목명 200개에 대한 종목 코드를 찾아내고 ‘종목 코드-종목명’을 키(key)-값(value)으로 한 딕셔너리 universe에 저장합니다. 여기에 생성 일자('created_at')를 추가하여 universe_df라는 데이터프레임을 만듭니다.

'created_at': [now] * len(universe.keys())

([now] * len(universe.keys())universe의 데이터 길이만큼 리스트에 현재 날짜를 채워 넣으라는 의미로, universe_df가 생성될 때 데이터끼리 길이를 맞추는 코드입니다.

이렇게 유니버스 정보를 담은 universe_df를 만들고 나면 데이터베이스에 저장하는 과정을 거쳐야 합니다. 데이터를 저장하는 기능은 insert_df_to_db 함수로 아직 구현하지 않았는데, db_helper.py 파일에서 구현하겠습니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.