Add api token flag

Currently the `demo` API token is hardcoded, which is not usable for
real API calls. Add a new CLI flag to allow users to supply their own
registered API token.
This commit is contained in:
Jon Banafato
2020-04-06 00:21:17 -04:00
parent 124fa37be8
commit cf884263c8

View File

@@ -13,6 +13,7 @@ import (
) )
func main() { func main() {
apiToken := flag.String("a", "demo", "World Trading Data API Token")
ledgerBinary := flag.String("b", "ledger", "Ledger Binary") ledgerBinary := flag.String("b", "ledger", "Ledger Binary")
ledgerFile := flag.String("f", "ledger.ledger", "Ledger File") ledgerFile := flag.String("f", "ledger.ledger", "Ledger File")
priceDbFile := flag.String("p", "prices.db", "Price Database File") priceDbFile := flag.String("p", "prices.db", "Price Database File")
@@ -27,7 +28,7 @@ func main() {
defer pricedb.Close() defer pricedb.Close()
for _, c := range commodities { for _, c := range commodities {
priceString, err := GetPriceString(c) priceString, err := GetPriceString(c, *apiToken)
if err != nil { if err != nil {
continue continue
} }
@@ -35,8 +36,8 @@ func main() {
} }
} }
func GetPriceString(ticker string) (string, error) { func GetPriceString(ticker string, apiToken string) (string, error) {
resp, err := http.Get("https://api.worldtradingdata.com/api/v1/stock?symbol=" + ticker + "&api_token=demo") resp, err := http.Get("https://api.worldtradingdata.com/api/v1/stock?symbol=" + ticker + "&api_token=" + apiToken)
if err != nil { if err != nil {
return "", err return "", err
} }