yg070520@sina.com
1 year ago
parent
a9f4de28d0
commit
56e6e0a711
@ -0,0 +1,297 @@ |
|||||||
|
package mop |
||||||
|
|
||||||
|
import ( |
||||||
|
"encoding/json" |
||||||
|
"fmt" |
||||||
|
"io/ioutil" |
||||||
|
"net/http" |
||||||
|
"strconv" |
||||||
|
"strings" |
||||||
|
"time" |
||||||
|
|
||||||
|
//"github.com/markcheno/go-talib"
|
||||||
|
) |
||||||
|
|
||||||
|
type KLineData struct { |
||||||
|
Day string `json:"day"` |
||||||
|
Open float64 `json:"open,string"` |
||||||
|
High float64 `json:"high,string"` |
||||||
|
Low float64 `json:"low,string"` |
||||||
|
Close float64 `json:"close,string"` |
||||||
|
Volume float64 `json:"volume,string"` |
||||||
|
} |
||||||
|
|
||||||
|
func getURL(code string, ts int, count int) string { |
||||||
|
return fmt.Sprintf("http://money.finance.sina.com.cn/quotes_service/api/json_v2.php/CN_MarketData.getKLineData?symbol=%s&scale=%d&ma=5&datalen=%d", code, ts, count) |
||||||
|
} |
||||||
|
|
||||||
|
func getKLineData(url string) ([]KLineData, error) { |
||||||
|
resp, err := http.Get(url) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
defer resp.Body.Close() |
||||||
|
|
||||||
|
body, err := ioutil.ReadAll(resp.Body) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
var data []KLineData |
||||||
|
err = json.Unmarshal(body, &data) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
return data, nil |
||||||
|
} |
||||||
|
func get_price_sina(code string, end_date string, count int, frequency string, startdate string) ([]KLineData, error) { |
||||||
|
frequency = strings.Replace(frequency, "1d", "240m", 1) |
||||||
|
frequency = strings.Replace(frequency, "1w", "1200m", 1) |
||||||
|
frequency = strings.Replace(frequency, "1M", "7200m", 1) |
||||||
|
|
||||||
|
ts, err := strconv.Atoi(strings.TrimSuffix(frequency, "m")) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
if end_date != "" && (frequency == "240m" || frequency == "1200m" || frequency == "7200m") { |
||||||
|
endDate, err := time.Parse("2006-01-02", end_date) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
unit := 1 |
||||||
|
if frequency == "1200m" { |
||||||
|
unit = 4 |
||||||
|
} else if frequency == "7200m" { |
||||||
|
unit = 29 |
||||||
|
} |
||||||
|
|
||||||
|
days := int(time.Since(endDate).Hours() / 24) |
||||||
|
count = count + days/unit |
||||||
|
} |
||||||
|
|
||||||
|
url := getURL(code, ts, count) |
||||||
|
//fmt.Printf("url: %s\n", url)
|
||||||
|
data, err := getKLineData(url) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
//fmt.Printf("data: %v\n", data)
|
||||||
|
if startdate != "" && (frequency == "240m" || frequency == "1200m" || frequency == "7200m") { |
||||||
|
endDate := time.Now() |
||||||
|
//endDate, err := time.Parse("2006-01-02", end_date)
|
||||||
|
//if err != nil {
|
||||||
|
// return nil, err
|
||||||
|
//}
|
||||||
|
|
||||||
|
start_date, err := time.Parse("2006-01-02", startdate) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
filteredData := []KLineData{} |
||||||
|
for _, d := range data { |
||||||
|
day, err := time.Parse("2006-01-02", d.Day) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
if day.After(start_date) && (day.Before(endDate) || day.Equal(endDate)) { |
||||||
|
filteredData = append(filteredData, d) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return filteredData, nil |
||||||
|
} |
||||||
|
|
||||||
|
return data, nil |
||||||
|
} |
||||||
|
|
||||||
|
func findMaxHighAndMinLow(data []KLineData) (float64, float64) { |
||||||
|
maxHigh := data[0].High |
||||||
|
minLow := data[0].Low |
||||||
|
|
||||||
|
for _, d := range data { |
||||||
|
if d.High > maxHigh { |
||||||
|
maxHigh = d.High |
||||||
|
} |
||||||
|
if d.Low < minLow { |
||||||
|
minLow = d.Low |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return maxHigh, minLow |
||||||
|
} |
||||||
|
|
||||||
|
// 将interface{}类型转换为float64类型
|
||||||
|
func convertToFloat64(value interface{}) float64 { |
||||||
|
if floatValue, ok := value.(float64); ok { |
||||||
|
return floatValue |
||||||
|
} |
||||||
|
if stringValue, ok := value.(string); ok { |
||||||
|
floatValue, err := strconv.ParseFloat(stringValue, 64) |
||||||
|
if err == nil { |
||||||
|
return floatValue |
||||||
|
} |
||||||
|
} |
||||||
|
return 0 |
||||||
|
} |
||||||
|
|
||||||
|
func getURLTencent(code string, start string, count int) string { |
||||||
|
// 给定的日期
|
||||||
|
datestr := "" |
||||||
|
date, err := time.Parse("2006-01-02", start) |
||||||
|
if err != nil { |
||||||
|
fmt.Println("解析日期失败：", err) |
||||||
|
}else{ |
||||||
|
futureDate := date.AddDate(0, 0, 10) |
||||||
|
if futureDate.Before(time.Now()){ |
||||||
|
datestr = futureDate.Format("2006-01-02") |
||||||
|
fmt.Println(datestr) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return fmt.Sprintf("https://web.ifzq.gtimg.cn/appstock/app/fqkline/get?param=%s,day,,%s,%d,qfq", code, datestr, count) |
||||||
|
} |
||||||
|
|
||||||
|
func getKLineTencent(url string, code string)([]KLineData, error){ |
||||||
|
resp, err := http.Get(url) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
defer resp.Body.Close() |
||||||
|
|
||||||
|
body, err := ioutil.ReadAll(resp.Body) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
var dataMap map[string]interface{} |
||||||
|
err = json.Unmarshal(body, &dataMap) |
||||||
|
if err != nil { |
||||||
|
fmt.Println("解析JSON字符串失败", err) |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
klineData := make([]KLineData, 0) |
||||||
|
// 获取K线数据
|
||||||
|
if shData, ok := dataMap["data"].(map[string]interface{})[code].(map[string]interface{}); ok { |
||||||
|
if qfqday, ok := shData["qfqday"].([]interface{}); ok { |
||||||
|
|
||||||
|
for _, item := range qfqday { |
||||||
|
if data, ok := item.([]interface{}); ok { |
||||||
|
klineData = append(klineData, KLineData{ |
||||||
|
Day: data[0].(string), |
||||||
|
Open: convertToFloat64(data[1]), |
||||||
|
High: convertToFloat64(data[3]), |
||||||
|
Low: convertToFloat64(data[4]), |
||||||
|
Close: convertToFloat64(data[2]), |
||||||
|
Volume: convertToFloat64(data[5]), |
||||||
|
}) |
||||||
|
} |
||||||
|
} |
||||||
|
//fmt.Println(klineData)
|
||||||
|
return klineData,nil |
||||||
|
}else if day, ok := shData["day"].([]interface{}); ok { |
||||||
|
for _, item := range day { |
||||||
|
if data, ok := item.([]interface{}); ok { |
||||||
|
klineData = append(klineData, KLineData{ |
||||||
|
Day: data[0].(string), |
||||||
|
Open: convertToFloat64(data[1]), |
||||||
|
High: convertToFloat64(data[3]), |
||||||
|
Low: convertToFloat64(data[4]), |
||||||
|
Close: convertToFloat64(data[2]), |
||||||
|
Volume: convertToFloat64(data[5]), |
||||||
|
}) |
||||||
|
} |
||||||
|
} |
||||||
|
return klineData,nil |
||||||
|
} |
||||||
|
} |
||||||
|
return klineData,nil |
||||||
|
} |
||||||
|
|
||||||
|
func get_price_tencent(code string, end_date string, count int, frequency string, startdate string) ([]KLineData, error) { |
||||||
|
frequency = strings.Replace(frequency, "1d", "240m", 1) |
||||||
|
frequency = strings.Replace(frequency, "1w", "1200m", 1) |
||||||
|
frequency = strings.Replace(frequency, "1M", "7200m", 1) |
||||||
|
|
||||||
|
if end_date != "" && (frequency == "240m" || frequency == "1200m" || frequency == "7200m") { |
||||||
|
endDate, err := time.Parse("2006-01-02", end_date) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
unit := 1 |
||||||
|
if frequency == "1200m" { |
||||||
|
unit = 4 |
||||||
|
} else if frequency == "7200m" { |
||||||
|
unit = 29 |
||||||
|
} |
||||||
|
|
||||||
|
days := int(time.Since(endDate).Hours() / 24) |
||||||
|
count = count + days/unit |
||||||
|
} |
||||||
|
/* |
||||||
|
url := getURL(code, ts, count) |
||||||
|
fmt.Printf("url: %s\n", url) |
||||||
|
data, err := getKLineData(url) |
||||||
|
*/ |
||||||
|
url := getURLTencent(code, startdate, 10) |
||||||
|
//fmt.Printf("url: %s\n", url)
|
||||||
|
data, err := getKLineTencent(url, code) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
//fmt.Printf("data: %v\n", data)
|
||||||
|
if startdate != "" && (frequency == "240m" || frequency == "1200m" || frequency == "7200m") { |
||||||
|
endDate := time.Now() |
||||||
|
//endDate, err := time.Parse("2006-01-02", end_date)
|
||||||
|
//if err != nil {
|
||||||
|
// return nil, err
|
||||||
|
//}
|
||||||
|
start_date, err := time.Parse("2006-01-02", startdate) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
filteredData := []KLineData{} |
||||||
|
for _, d := range data { |
||||||
|
day, err := time.Parse("2006-01-02", d.Day) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
//fmt.Println(start_date, endDate, day)
|
||||||
|
if day.After(start_date) && (day.Before(endDate) || day.Equal(endDate)) { |
||||||
|
filteredData = append(filteredData, d) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return filteredData, nil |
||||||
|
} |
||||||
|
|
||||||
|
return data, nil |
||||||
|
} |
||||||
|
|
||||||
|
func getnextdaysHL(code string, start_date string) (float64, float64, float64, float64 ,int) { |
||||||
|
//data, err := get_price_sina(code, "", 30, "240m", start_date)
|
||||||
|
data, err := get_price_tencent(code, "", 30, "240m", start_date) |
||||||
|
if err != nil { |
||||||
|
return 0, 0, 0, 0, 0 |
||||||
|
} |
||||||
|
|
||||||
|
pre := 3 |
||||||
|
if len(data) < 3 { |
||||||
|
pre = len(data) |
||||||
|
} |
||||||
|
|
||||||
|
//fmt.Printf("data: %v\n", data[:3])
|
||||||
|
caldata := data[:pre] |
||||||
|
maxHigh, minLow := findMaxHighAndMinLow(caldata) |
||||||
|
//fmt.Println(maxHigh, minLow)
|
||||||
|
|
||||||
|
return caldata[0].Open, caldata[pre-1].Close, maxHigh, minLow, pre |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
@ -0,0 +1,102 @@ |
|||||||
|
// Copyright (c) 2013-2023 by Michael Dvorkin and contributors. All Rights Reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style license that can
|
||||||
|
// be found in the LICENSE file.
|
||||||
|
|
||||||
|
package mop |
||||||
|
|
||||||
|
import ( |
||||||
|
"io/ioutil" |
||||||
|
"net/http" |
||||||
|
"strings" |
||||||
|
"net/url" |
||||||
|
) |
||||||
|
|
||||||
|
const crumbURL = "https://query1.finance.yahoo.com/v1/test/getcrumb" |
||||||
|
const cookieURL = "https://login.yahoo.com" |
||||||
|
const userAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/113.0" |
||||||
|
|
||||||
|
func fetchCrumb(cookies string) string { |
||||||
|
proxyUrl, err := url.Parse("http://172.29.100.107:28888") |
||||||
|
if err != nil { |
||||||
|
panic(err) |
||||||
|
} |
||||||
|
|
||||||
|
transport := &http.Transport{ |
||||||
|
Proxy: http.ProxyURL(proxyUrl), |
||||||
|
} |
||||||
|
|
||||||
|
client := &http.Client{ |
||||||
|
Transport: transport, |
||||||
|
} |
||||||
|
//client := http.Client{}
|
||||||
|
request, err := http.NewRequest("GET", crumbURL, nil) |
||||||
|
if err != nil { |
||||||
|
panic(err) |
||||||
|
} |
||||||
|
|
||||||
|
request.Header = http.Header{ |
||||||
|
"Accept": {"*/*"}, |
||||||
|
"Accept-Encoding": {"gzip, deflate, br"}, |
||||||
|
"Accept-Language": {"en-US,en;q=0.5"}, |
||||||
|
"Connection": {"keep-alive"}, |
||||||
|
"Content-Type": {"text/plain"}, |
||||||
|
"Cookie": {cookies}, |
||||||
|
"Host": {"query1.finance.yahoo.com"}, |
||||||
|
"Sec-Fetch-Dest": {"empty"}, |
||||||
|
"Sec-Fetch-Mode": {"cors"}, |
||||||
|
"Sec-Fetch-Site": {"same-site"}, |
||||||
|
"TE": {"trailers"}, |
||||||
|
"User-Agent": {userAgent}, |
||||||
|
} |
||||||
|
|
||||||
|
response, err := client.Do(request) |
||||||
|
if err != nil { |
||||||
|
panic(err) |
||||||
|
} |
||||||
|
defer response.Body.Close() |
||||||
|
|
||||||
|
body, err := ioutil.ReadAll(response.Body) |
||||||
|
if err != nil { |
||||||
|
panic(err) |
||||||
|
} |
||||||
|
|
||||||
|
return string(body[:]) |
||||||
|
} |
||||||
|
|
||||||
|
func fetchCookies() string { |
||||||
|
client := http.Client{} |
||||||
|
request, err := http.NewRequest("GET", cookieURL, nil) |
||||||
|
if err != nil { |
||||||
|
panic(err) |
||||||
|
} |
||||||
|
|
||||||
|
request.Header = http.Header{ |
||||||
|
"Accept": {"*/*"}, |
||||||
|
"Accept-Encoding": {"gzip, deflate, br"}, |
||||||
|
"Accept-Language": {"en-US,en;q=0.5"}, |
||||||
|
"Connection": {"keep-alive"}, |
||||||
|
"Host": {"login.yahoo.com"}, |
||||||
|
"Sec-Fetch-Dest": {"document"}, |
||||||
|
"Sec-Fetch-Mode": {"navigate"}, |
||||||
|
"Sec-Fetch-Site": {"none"}, |
||||||
|
"Sec-Fetch-User": {"?1"}, |
||||||
|
"TE": {"trailers"}, |
||||||
|
"Update-Insecure-Requests": {"1"}, |
||||||
|
"User-Agent": {userAgent}, |
||||||
|
} |
||||||
|
|
||||||
|
response, err := client.Do(request) |
||||||
|
if err != nil { |
||||||
|
panic(err) |
||||||
|
} |
||||||
|
defer response.Body.Close() |
||||||
|
|
||||||
|
var result string |
||||||
|
for _, cookie := range response.Cookies() { |
||||||
|
if cookie.Name != "AS" { |
||||||
|
result += cookie.Name + "=" + cookie.Value + "; " |
||||||
|
} |
||||||
|
} |
||||||
|
result = strings.TrimSuffix(result, "; ") |
||||||
|
return result |
||||||
|
} |
||||||
Loading…
Reference in new issue