4f51a511dd924f89fe9ec9ee3da944779cc0876f,QUANTAXIS/QAWeb/fetch_block.py,,get_block,#Any#,10

Before Change

def get_block(block):
    block_tdx = QA.DATABASE.stock_block
    block_ths = QA.DATABASE.stock_block_ths
    codelist = QA.QA_fetch_stock_list_adv()
    block = [block] if isinstance(block, str) else block

    code_tdx = pd.DataFrame(
        [item for item in block_tdx.find({"blockname": {"$in": block}})])
    code_ths = pd.DataFrame(
        [item for item in block_ths.find({"blockname": {"$in": block}})])

    return pd.concat([code_tdx, code_ths], axis=0).code.drop_duplicates().tolist()


def get_name(code):
    codelist = QA.QA_fetch_stock_list_adv()

After Change

    block = [block] if isinstance(block, str) else block
    block_df = pd.DataFrame(
        [item for item in block_db.find({"blockname": {"$in": block}})])
    return block_df.code.drop_duplicates().tolist()


def get_name(code):
    codelist = QA.QA_fetch_stock_list_adv()

Instances