import setting
import gm.api as gmapi


class JueJinApi:
    __previous_trading_date_cache = {}

    # 获取掘金参数
    @classmethod
    def getJueJinAccountInfo(cls):
        # strategy_id, token = setting.get_juejin_params()
        return "", "95a982ce-fc2d-11ec-8ff3-0a0027000010", "77fa387b37e9bc1586ae1135827e2ccb3e7cd4e4"

    @classmethod
    def get_juejin_code_list_with_prefix(cls, codes):
        list = []
        for d in codes:
            if d.find(".") > 0:
                list.append(d)
            elif d[0:2] == '00':
                list.append("SZSE.{}".format(d))
            elif d[0:2] == '60':
                list.append("SHSE.{}".format(d))
        return list

    @classmethod
    def get_gp_latest_info(cls, codes, fields=None):
        if not codes:
            return []
        symbols = cls.get_juejin_code_list_with_prefix(codes)

        account_id, s_id, token = cls.getJueJinAccountInfo()
        gmapi.set_token(token)
        data = gmapi.get_instruments(symbols=",".join(symbols), fields=fields)
        return data

    @classmethod
    def get_history_tick_n(cls, code, count, fields=None, frequency="1d"):
        symbols = cls.get_juejin_code_list_with_prefix([code])

        account_id, s_id, token = cls.getJueJinAccountInfo()
        gmapi.set_token(token)
        # 前除权
        results = gmapi.history_n(symbol=symbols[0], frequency=frequency, count=count, adjust=1, fields=fields)
        return results

    @classmethod
    def get_history_tick(cls, code, start_time, end_time, fields=None, frequency="1d"):
        symbols = cls.get_juejin_code_list_with_prefix([code])

        account_id, s_id, token = cls.getJueJinAccountInfo()
        gmapi.set_token(token)
        # 前除权
        results = gmapi.history(symbol=symbols[0], frequency=frequency, start_time=start_time, end_time=end_time,
                                adjust=1, fields=fields)
        return results

    @classmethod
    def get_gp_current_info(cls, codes, fields=''):
        if not codes:
            return []
        symbols = cls.get_juejin_code_list_with_prefix(codes)

        account_id, s_id, token = cls.getJueJinAccountInfo()
        gmapi.set_token(token)
        data = gmapi.current(symbols=",".join(symbols), fields=fields)
        return data

    # 获取交易所的代码
    @classmethod
    def get_exchanges_codes(cls, exchanges):

        account_id, s_id, token = cls.getJueJinAccountInfo()
        gmapi.set_token(token)
        return gmapi.get_instruments(exchanges=exchanges, sec_types=[1], skip_suspended=True, skip_st=True,
                                     fields="symbol,sec_type,sec_id,sec_name,listed_date,sec_level,is_suspended,pre_close")

    @classmethod
    def get_previous_trading_date(cls, date):
        account_id, s_id, token = cls.getJueJinAccountInfo()
        gmapi.set_token(token)
        return gmapi.get_previous_trading_date("SHSE", date)

    @classmethod
    def get_previous_trading_date_cache(cls, date):
        if date not in cls.__previous_trading_date_cache:
            pre_date = cls.get_previous_trading_date(date)
            if pre_date:
                cls.__previous_trading_date_cache[date] = pre_date
        return cls.__previous_trading_date_cache.get(date)

    # 返回指定日期的下个交易日
    @classmethod
    def get_next_trading_date(cls, date):

        account_id, s_id, token = cls.getJueJinAccountInfo()
        gmapi.set_token(token)
        return gmapi.get_next_trading_date("SHSE", date)

    @classmethod
    def get_trading_dates(cls, start_date, end_date):
        account_id, s_id, token = cls.getJueJinAccountInfo()
        gmapi.set_token(token)
        return gmapi.get_trading_dates("SHSE", start_date, end_date)

    @classmethod
    def get_codes(cls, sec_type):
        """
        代码获取
        :param sec_type: 8：可转债 1：股票
        :return:
        """
        account_id, s_id, token = cls.getJueJinAccountInfo()
        gmapi.set_token(token)
        return gmapi.get_instrumentinfos(exchanges="SHSE,SZSE", sec_types=[sec_type])

    @classmethod
    def get_underlying_code(cls, symbols):
        """
        获取可转债正股代码
        :param symbols: 如:[SZSE.128144]
        :return: 正股代码,如:[SZSE.002734]
        """
        account_id, s_id, token = JueJinApi.getJueJinAccountInfo()
        gmapi.set_token(token)
        datas = gmapi.get_instruments(symbols=",".join(symbols), sec_types="8",
                                      fields="symbol, sec_type, sec_id,sec_name, underlying_symbol")
        fdatas = {}
        for d in datas:
            fdatas[d['symbol']] = d['underlying_symbol']
        return fdatas


if __name__ == "__main__":
    print(JueJinApi.get_previous_trading_date("2024-12-31"))