ok

tushare/stock.go

@@ -0,0 +1,241 @@
+package tushare
+
+import (
+	"time"
+)
+
+/*
+TradeCal 获取交易日历
+
+	exchange: 交易所代码，SSE 上交所，SZSE 深交所，BSE 北交所，空字符串代表所有
+	start_date: 开始日期，格式:YYYYMMDD
+	end_date: 结束日期，格式:YYYYMMDD
+	is_open: 是否开盘，'0'休市，'1'交易，空字符串代表所有
+*/
+func (cli *TushareClient) TradeCal(exchange, start_date, end_date, is_open string) (*TushareRespData, error) {
+	params := map[string]any{}
+
+	if exchange != "" {
+		params["exchange"] = exchange
+	}
+	if start_date != "" {
+		params["start_date"] = start_date
+	}
+	if end_date != "" {
+		params["end_date"] = end_date
+	}
+	if is_open != "" {
+		params["is_open"] = is_open
+	}
+
+	req := TushareReq{
+		APIName: "trade_cal",
+		Params:  params,
+	}
+
+	fields := []map[string]string{
+		{"exchange": "交易所代码"},
+		{"cal_date": "日历日期"},
+		{"is_open": "是否开盘"},
+		{"pretrade_date": "上一交易日"},
+	}
+
+	return cli.Do(req, fields)
+}
+
+func (cli *TushareClient) ReturnLastTradeDay() string {
+	result, err := cli.TradeCal("SSE", time.Now().AddDate(0, 0, -15).Format("20060102"), time.Now().Format("20060102"), "1")
+	if err != nil {
+		return ""
+	}
+	cal := result.Map()
+	return cal[0]["cal_date"].(string)
+}
+
+/*
+StockBasic 获取股票列表
+
+	ts_code: 股票代码，支持多个，如"000002.SZ,000001.SZ"
+	exchange: 交易所代码，SSE 上交所，SZSE 深交所
+	is_shsc: 是否在沪股通或深港通范围内，0:否;1:沪股通;2:深股通
+	start_date: 开始日期，格式:YYYYMMDD
+	end_date: 结束日期，格式:YYYYMMDD
+*/
+func (cli *TushareClient) StockBasic(ts_code, exchange, is_shsc, start_date, end_date string) (*TushareRespData, error) {
+	params := map[string]any{}
+
+	if ts_code != "" {
+		params["ts_code"] = ts_code
+	}
+	if exchange != "" {
+		params["exchange"] = exchange
+	}
+	if is_shsc != "" {
+		params["is_shsc"] = is_shsc
+	}
+	if start_date != "" {
+		params["start_date"] = start_date
+	}
+	if end_date != "" {
+		params["end_date"] = end_date
+	}
+
+	req := TushareReq{
+		APIName: "stock_basic",
+		Params:  params,
+	}
+
+	fields := []map[string]string{
+		{"ts_code": "TS 代码"},
+		{"symbol": "股票交易代码"},
+		{"name": "证券简称"},
+		{"area": "地区"},
+		{"industry": "所属行业"},
+		{"market": "市场类别"},
+		{"list_status": "上市状态"},
+		{"list_date": "上市日期"},
+		{"delist_date": "退市日期"},
+		{"is_shsc": "是否沪股通或深股通"},
+	}
+
+	return cli.Do(req, fields)
+}
+
+/*
+Daily 获取日线行情
+
+	ts_code: 股票代码，格式：000001.SZ 或 600000.SH
+	trade_date: 交易日期，格式:YYYYMMDD
+	start_date: 开始日期，格式:YYYYMMDD
+	end_date: 结束日期，格式:YYYYMMDD
+*/
+func (cli *TushareClient) Daily(ts_code, trade_date, start_date, end_date string) (*TushareRespData, error) {
+	params := map[string]any{}
+
+	if ts_code != "" {
+		params["ts_code"] = ts_code
+	}
+	if trade_date != "" {
+		params["trade_date"] = trade_date
+	}
+	if start_date != "" {
+		params["start_date"] = start_date
+	}
+	if end_date != "" {
+		params["end_date"] = end_date
+	}
+
+	req := TushareReq{
+		APIName: "daily",
+		Params:  params,
+	}
+
+	fields := []map[string]string{
+		{"ts_code": "股票代码"},
+		{"trade_date": "交易日期"},
+		{"open": "开盘价"},
+		{"high": "最高价"},
+		{"low": "最低价"},
+		{"close": "收盘价"},
+		{"pre_close": "昨收价"},
+		{"change": "涨跌额"},
+		{"pct_chg": "涨跌幅%"},
+		{"vol": "成交量 (手)"},
+		{"amount": "成交额 (千元)"},
+	}
+
+	return cli.Do(req, fields)
+}
+
+/*
+Min 获取分钟线行情
+
+	ts_code: 股票代码，格式：000001.SZ 或 600000.SH
+	trade_date: 交易日期，格式:YYYYMMDD
+	minute: 分钟类型，1/5/15/30/60 分钟
+	start_time: 开始时间，格式:HHmmss
+	end_time: 结束时间，格式:HHmmss
+*/
+func (cli *TushareClient) Min(ts_code, trade_date, minute, start_time, end_time string) (*TushareRespData, error) {
+	params := map[string]any{}
+
+	if ts_code != "" {
+		params["ts_code"] = ts_code
+	}
+	if trade_date != "" {
+		params["trade_date"] = trade_date
+	}
+	if minute != "" {
+		params["minute"] = minute
+	}
+	if start_time != "" {
+		params["start_time"] = start_time
+	}
+	if end_time != "" {
+		params["end_time"] = end_time
+	}
+
+	req := TushareReq{
+		APIName: "min",
+		Params:  params,
+	}
+
+	fields := []map[string]string{
+		{"ts_code": "股票代码"},
+		{"trade_time": "交易时间"},
+		{"open": "开盘价"},
+		{"high": "最高价"},
+		{"low": "最低价"},
+		{"close": "收盘价"},
+		{"vol": "成交量 (手)"},
+		{"amount": "成交额 (千元)"},
+	}
+
+	return cli.Do(req, fields)
+}
+
+/*
+IndexDaily 获取大盘指数日线行情
+
+	ts_code: 指数代码，如 000001.SH(上证指数),399001.SZ(深证成指)
+	trade_date: 交易日期，格式:YYYYMMDD
+	start_date: 开始日期，格式:YYYYMMDD
+	end_date: 结束日期，格式:YYYYMMDD
+*/
+func (cli *TushareClient) IndexDaily(ts_code, trade_date, start_date, end_date string) (*TushareRespData, error) {
+	params := map[string]any{}
+
+	if ts_code != "" {
+		params["ts_code"] = ts_code
+	}
+	if trade_date != "" {
+		params["trade_date"] = trade_date
+	}
+	if start_date != "" {
+		params["start_date"] = start_date
+	}
+	if end_date != "" {
+		params["end_date"] = end_date
+	}
+
+	req := TushareReq{
+		APIName: "index_daily",
+		Params:  params,
+	}
+
+	fields := []map[string]string{
+		{"ts_code": "指数代码"},
+		{"trade_date": "交易日期"},
+		{"close": "收盘点位"},
+		{"open": "开盘点位"},
+		{"high": "最高点位"},
+		{"low": "最低点位"},
+		{"pre_close": "昨收点位"},
+		{"change": "涨跌额"},
+		{"pct_chg": "涨跌幅%"},
+		{"vol": "成交量 (手)"},
+		{"amount": "成交额 (千元)"},
+	}
+
+	return cli.Do(req, fields)
+}