ptrade移除当前ST股

def remove_st_stock(all_stock_list):

    st_dict = get_stock_status(all_stock_list, query_type='ST', query_date=None)

    st_list = []

    for k, v in st_dict.items():

        if v:

            st_list.append(k)

    return st_list



MARKET_DICT = {0: '科创板', 1: '创业板', }

IGNORE_MARKET = [0, 1]



def all_codes_in_market():

    all_stock_set = set(get_Ashares(date=None))

    for ignore_code in IGNORE_MARKET:

        market = MARKET_DICT.get(ignore_code)

        if market == '科创板':

            all_stock_set = all_stock_set - set(filter(lambda x: x.startswith('68'), all_stock_set))

        if market == '创业板':

            all_stock_set = all_stock_set - set(filter(lambda x: x.startswith('3'), all_stock_set))



    return all_stock_set





def create_target(context):

    all_stock_set = all_codes_in_market()

    st_list = remove_st_stock(list(all_stock_set))

    all_stock_set = all_stock_set - set(st_list)

    return all_stock_set