print("3. 케라스 코스피 주식 예측") print("4. 텐서플로우 코스피 주식 예측") print("5. 텐서플로우 코스닥 주식 예측") print("6. 테스트") print("7. 종료") return int(input("메뉴선택: ")) if __name__ == "__main__": while 1: menu = print_menu() if menu == 1: i = 0 CodeMgr.collect_stock_codes() stock_codes = DbMgr.get_all_stock_code() for code, row in stock_codes.iterrows(): print(i, row["code_name"], code) i += 1 if DbMgr.exist_stock_info_db(code)[0][0] == 1: if DbMgr.get_stock_count(code) < 10: print("생성") StockMgr.collect_total_stock_info(code) else: StockMgr.collect_stock_info(code) else: print("생성") StockMgr.collect_total_stock_info(code) elif menu == 2:
"html.parser").select("td > a.tltle") for stock_code in stock_codes: # print(stock_code.attrs["href"][-6:], stock_code.getText(), stock_type) DbMgr.insert_into_stock_codes_db(stock_code.attrs["href"][-6:], stock_code.getText(), "P" if stock_type == 0 else "Q") @staticmethod def collect_stock_codes(): DbMgr.create_stock_codes_db() # KOSPI for i in range(1, kospi_page_len): print("KOSPI" + str(i)) CodeMgr._insert_stock_codes(0, i) # KOSDAQ for i in range(1, kosdaq_page_len): print("KOSDAQ" + str(i)) CodeMgr._insert_stock_codes(1, i) if __name__ == "__main__": i = 0 stock_codes = DbMgr.get_all_stock_code() for code, row in stock_codes.iterrows(): print(i, row["code_name"], code) i += 1