Python DbMgr.get_all_stock_code示例

示例#1

    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:

示例#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