You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
246 lines
7.0 KiB
246 lines
7.0 KiB
// Copyright (c) 2013 by Michael Dvorkin. All Rights Reserved.
|
|
//=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
|
|
package mop
|
|
|
|
import (
|
|
`bytes`
|
|
`fmt`
|
|
`regexp`
|
|
`strings`
|
|
`text/template`
|
|
`time`
|
|
)
|
|
|
|
type Formatter struct {}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func (self *Formatter) Format(entity interface{}) string {
|
|
switch entity.(type) {
|
|
case *Market:
|
|
return self.format_market(entity.(*Market))
|
|
case *Quotes:
|
|
return self.format_quotes(entity.(*Quotes))
|
|
}
|
|
return ``
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func (self *Formatter) format_market(m *Market) string {
|
|
markup := `{{.Dow.name}}: `
|
|
if m.Dow[`change`][0:1] != `-` {
|
|
markup += `<green>{{.Dow.change}} ({{.Dow.percent}})</green> at {{.Dow.latest}}, `
|
|
} else {
|
|
markup += `{{.Dow.change}} ({{.Dow.percent}}) at {{.Dow.latest}}, `
|
|
}
|
|
markup += `{{.Sp500.name}}: `
|
|
if m.Sp500[`change`][0:1] != `-` {
|
|
markup += `<green>{{.Sp500.change}} ({{.Sp500.percent}})</green> at {{.Sp500.latest}}, `
|
|
} else {
|
|
markup += `{{.Sp500.change}} ({{.Sp500.percent}}) at {{.Sp500.latest}}, `
|
|
}
|
|
markup += `{{.Nasdaq.name}}: `
|
|
if m.Nasdaq[`change`][0:1] != `-` {
|
|
markup += `<green>{{.Nasdaq.change}} ({{.Nasdaq.percent}})</green> at {{.Nasdaq.latest}}`
|
|
} else {
|
|
markup += `{{.Nasdaq.change}} ({{.Nasdaq.percent}}) at {{.Nasdaq.latest}}`
|
|
}
|
|
markup += "\n"
|
|
markup += `{{.Advances.name}}: {{.Advances.nyse}} ({{.Advances.nysep}}) on NYSE and {{.Advances.nasdaq}} ({{.Advances.nasdaqp}}) on Nasdaq. `
|
|
markup += `{{.Declines.name}}: {{.Declines.nyse}} ({{.Declines.nysep}}) on NYSE and {{.Declines.nasdaq}} ({{.Declines.nasdaqp}}) on Nasdaq`
|
|
if !m.Open {
|
|
markup += `<right>U.S. markets closed</right>`
|
|
}
|
|
markup += "\n"
|
|
markup += `New highs: {{.Highs.nyse}} on NYSE and {{.Highs.nasdaq}} on Nasdaq. `
|
|
markup += `New lows: {{.Lows.nyse}} on NYSE and {{.Lows.nasdaq}} on Nasdaq.`
|
|
template, err := template.New(`market`).Parse(markup)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
buffer := new(bytes.Buffer)
|
|
err = template.Execute(buffer, m)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return buffer.String()
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func (self *Formatter) format_quotes(quotes *Quotes) string {
|
|
vars := struct {
|
|
Now string
|
|
Header string
|
|
Stocks []Stock
|
|
}{
|
|
time.Now().Format(`3:04:05pm PST`),
|
|
header(),
|
|
prettify(quotes),
|
|
}
|
|
|
|
markup := `<right><white>{{.Now}}</white></right>
|
|
|
|
|
|
|
|
{{.Header}}
|
|
{{range .Stocks}}{{.Color}}{{.Ticker}} {{.LastTrade}} {{.Change}} {{.ChangePercent}} {{.Open}} {{.Low}} {{.High}} {{.Low52}} {{.High52}} {{.Volume}} {{.AvgVolume}} {{.PeRatio}} {{.Dividend}} {{.Yield}} {{.MarketCap}}
|
|
{{end}}`
|
|
//markup += fmt.Sprintf("[%v]", quotes.profile.Grouped)
|
|
template, err := template.New(`quotes`).Parse(markup)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
buffer := new(bytes.Buffer)
|
|
err = template.Execute(buffer, vars)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return buffer.String()
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func header() string {
|
|
str := fmt.Sprintf(`<u>%-7s `, `Ticker`)
|
|
str += fmt.Sprintf(`%9s `, `Last`)
|
|
str += fmt.Sprintf(`%9s `, `Change`)
|
|
str += fmt.Sprintf(`%9s `, `%Change`)
|
|
str += fmt.Sprintf(`%9s `, `Open`)
|
|
str += fmt.Sprintf(`%9s `, `Low`)
|
|
str += fmt.Sprintf(`%9s `, `High`)
|
|
str += fmt.Sprintf(`%9s `, `52w Low`)
|
|
str += fmt.Sprintf(`%9s `, `52w High`)
|
|
str += fmt.Sprintf(`%10s `, `Volume`)
|
|
str += fmt.Sprintf(`%10s `, `AvgVolume`)
|
|
str += fmt.Sprintf(`%9s `, `P/E`)
|
|
str += fmt.Sprintf(`%9s `, `Dividend`)
|
|
str += fmt.Sprintf(`%9s `, `Yield`)
|
|
str += fmt.Sprintf(`%10s</u>`, `MktCap`)
|
|
|
|
return str
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func prettify(quotes *Quotes) []Stock {
|
|
pretty := make([]Stock, len(quotes.stocks))
|
|
for i, q := range group(quotes) {
|
|
pretty[i].Ticker = pad(q.Ticker, -7)
|
|
pretty[i].LastTrade = pad(with_currency(q.LastTrade), 9)
|
|
pretty[i].Change = pad(with_currency(q.Change), 9)
|
|
pretty[i].ChangePercent = pad(last_of_pair(q.ChangePercent), 9)
|
|
pretty[i].Open = pad(with_currency(q.Open), 9)
|
|
pretty[i].Low = pad(with_currency(q.Low), 9)
|
|
pretty[i].High = pad(with_currency(q.High), 9)
|
|
pretty[i].Low52 = pad(with_currency(q.Low52), 9)
|
|
pretty[i].High52 = pad(with_currency(q.High52), 9)
|
|
pretty[i].Volume = pad(q.Volume, 10)
|
|
pretty[i].AvgVolume = pad(q.AvgVolume, 10)
|
|
pretty[i].PeRatio = pad(nullify(q.PeRatioX), 9)
|
|
pretty[i].Dividend = pad(with_currency(q.Dividend), 9)
|
|
pretty[i].Yield = pad(with_percent(q.Yield), 9)
|
|
pretty[i].MarketCap = pad(with_currency(q.MarketCapX), 10)
|
|
}
|
|
return pretty
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func group(quotes *Quotes) []Stock {
|
|
if !quotes.profile.Grouped {
|
|
return quotes.stocks
|
|
} else {
|
|
grouped := make([]Stock, len(quotes.stocks))
|
|
current := 0
|
|
for _,stock := range quotes.stocks {
|
|
if strings.Index(stock.Change, "-") == -1 {
|
|
grouped[current] = stock
|
|
current++
|
|
}
|
|
}
|
|
for _,stock := range quotes.stocks {
|
|
if strings.Index(stock.Change, "-") != -1 {
|
|
grouped[current] = stock
|
|
current++
|
|
}
|
|
}
|
|
return grouped
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func nullify(str string) string {
|
|
if len(str) == 3 && str[0:3] == `N/A` {
|
|
return `-`
|
|
} else {
|
|
return str
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func last_of_pair(str string) string {
|
|
if len(str) >= 6 && str[0:6] != `N/A - ` {
|
|
return str
|
|
} else {
|
|
return str[6:]
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func with_currency(str string) string {
|
|
if str == `N/A` {
|
|
return `-`
|
|
} else {
|
|
switch str[0:1] {
|
|
case `+`, `-`:
|
|
return str[0:1] + `$` + str[1:]
|
|
default:
|
|
return `$` + str
|
|
}
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func with_percent(str string) string {
|
|
if str == `N/A` {
|
|
return `-`
|
|
} else {
|
|
return str + `%`
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func colorize(str string) string {
|
|
if str == `N/A` {
|
|
return `-`
|
|
} else if str[0:1] == `-` {
|
|
return `<red>` + str + `</red>`
|
|
} else {
|
|
return `<green>` + str + `</green>`
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func ticker(str string, change string) string {
|
|
if change[0:1] == `-` {
|
|
return `<red>` + str + `</red>`
|
|
} else {
|
|
return `<green>` + str + `</green>`
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func pad(str string, width int) string {
|
|
re := regexp.MustCompile(`(\.\d+)[MB]?$`)
|
|
match := re.FindStringSubmatch(str)
|
|
if len(match) > 0 {
|
|
switch len(match[1]) {
|
|
case 2:
|
|
str = strings.Replace(str, match[1], match[1] + `0`, 1)
|
|
case 4, 5:
|
|
str = strings.Replace(str, match[1], match[1][0:3], 1)
|
|
}
|
|
}
|
|
|
|
return fmt.Sprintf(`%*s`, width, str)
|
|
}
|
|
|