mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-26 16:55:15 +00:00
improve balances command
This commit is contained in:
parent
00109eb464
commit
22a9809327
|
@ -3,37 +3,72 @@ package cmd
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"github.com/pkg/errors"
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
|
|
||||||
|
"github.com/c9s/bbgo/pkg/bbgo"
|
||||||
)
|
)
|
||||||
|
|
||||||
//godotenv -f .env.local go run ./cmd/bbgo balances --session=ftx
|
func init() {
|
||||||
|
balancesCmd.Flags().String("session", "", "the exchange session name for querying balances")
|
||||||
|
RootCmd.AddCommand(balancesCmd)
|
||||||
|
}
|
||||||
|
|
||||||
|
// go run ./cmd/bbgo balances --session=ftx
|
||||||
var balancesCmd = &cobra.Command{
|
var balancesCmd = &cobra.Command{
|
||||||
Use: "balances",
|
Use: "balances",
|
||||||
SilenceUsage: true,
|
SilenceUsage: true,
|
||||||
RunE: func(cmd *cobra.Command, args []string) error {
|
RunE: func(cmd *cobra.Command, args []string) error {
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
session, err := cmd.Flags().GetString("session")
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("can't get session from flags: %w", err)
|
|
||||||
}
|
|
||||||
ex, err := newExchange(session)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
b, err := ex.QueryAccountBalances(ctx)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
log.Infof("balances: %+v", b)
|
|
||||||
|
|
||||||
|
configFile, err := cmd.Flags().GetString("config")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(configFile) == 0 {
|
||||||
|
return errors.New("--config option is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := os.Stat(configFile); os.IsNotExist(err) {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
userConfig, err := bbgo.Load(configFile, false)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
environ := bbgo.NewEnvironment()
|
||||||
|
if err := environ.ConfigureDatabase(ctx) ; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := environ.ConfigureExchangeSessions(userConfig); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
sessionName, err := cmd.Flags().GetString("session")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
session, ok := environ.Session(sessionName)
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("session %s not found", sessionName)
|
||||||
|
}
|
||||||
|
|
||||||
|
b, err := session.Exchange.QueryAccountBalances(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Infof("balances: %+v", b)
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
func init() {
|
|
||||||
balancesCmd.Flags().String("session", "", "the exchange session name for sync")
|
|
||||||
|
|
||||||
RootCmd.AddCommand(balancesCmd)
|
|
||||||
}
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user