mirror of
https://github.com/aclindsa/ofxgo.git
synced 2024-11-16 09:50:04 -05:00
Aaron Lindsay
4521bb377a
This adds a python script to generate constants.go when `go generate` is called, and updates the structs, tests, and command-line client to all use the new named constants.
99 lines
2.6 KiB
Go
99 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"github.com/aclindsa/ofxgo"
|
|
"os"
|
|
)
|
|
|
|
var bankTransactionsCommand = Command{
|
|
Name: "transactions-bank",
|
|
Description: "Print bank transactions and balance",
|
|
Flags: flag.NewFlagSet("transactions-bank", flag.ExitOnError),
|
|
CheckFlags: checkServerFlags,
|
|
Do: bankTransactions,
|
|
}
|
|
|
|
func init() {
|
|
defineServerFlags(bankTransactionsCommand.Flags)
|
|
bankTransactionsCommand.Flags.StringVar(&bankId, "bankid", "", "BankId (from `get-accounts` subcommand)")
|
|
bankTransactionsCommand.Flags.StringVar(&acctId, "acctid", "", "AcctId (from `get-accounts` subcommand)")
|
|
bankTransactionsCommand.Flags.StringVar(&acctType, "accttype", "CHECKING", "AcctType (from `get-accounts` subcommand)")
|
|
}
|
|
|
|
func bankTransactions() {
|
|
client, query := NewRequest()
|
|
|
|
acctTypeEnum, err := ofxgo.NewAcctType(acctType)
|
|
if err != nil {
|
|
fmt.Println("Error parsing accttype:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
uid, err := ofxgo.RandomUID()
|
|
if err != nil {
|
|
fmt.Println("Error creating uid for transaction:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
statementRequest := ofxgo.StatementRequest{
|
|
TrnUID: *uid,
|
|
BankAcctFrom: ofxgo.BankAcct{
|
|
BankId: ofxgo.String(bankId),
|
|
AcctId: ofxgo.String(acctId),
|
|
AcctType: acctTypeEnum,
|
|
},
|
|
Include: true,
|
|
}
|
|
|
|
query.Bank = append(query.Bank, &statementRequest)
|
|
|
|
response, err := client.Request(query)
|
|
if err != nil {
|
|
fmt.Println("Error requesting account statement:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if response.Signon.Status.Code != 0 {
|
|
meaning, _ := response.Signon.Status.CodeMeaning()
|
|
fmt.Printf("Nonzero signon status (%d: %s) with message: %s\n", response.Signon.Status.Code, meaning, response.Signon.Status.Message)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if len(response.Bank) < 1 {
|
|
fmt.Println("No banking messages received")
|
|
return
|
|
}
|
|
|
|
if stmt, ok := response.Bank[0].(*ofxgo.StatementResponse); ok {
|
|
fmt.Printf("Balance: %s %s (as of %s)\n", stmt.BalAmt, stmt.CurDef, stmt.DtAsOf)
|
|
fmt.Println("Transactions:")
|
|
for _, tran := range stmt.BankTranList.Transactions {
|
|
printTransaction(stmt.CurDef, &tran)
|
|
}
|
|
}
|
|
}
|
|
|
|
func printTransaction(defCurrency ofxgo.String, tran *ofxgo.Transaction) {
|
|
currency := defCurrency
|
|
if len(tran.Currency) > 0 {
|
|
currency = tran.Currency
|
|
} else if len(tran.OrigCurrency) > 0 {
|
|
currency = tran.Currency
|
|
}
|
|
|
|
var name string
|
|
if len(tran.Name) > 0 {
|
|
name = string(tran.Name)
|
|
} else if tran.Payee != nil {
|
|
name = string(tran.Payee.Name)
|
|
}
|
|
|
|
if len(tran.Memo) > 0 {
|
|
name = name + " - " + string(tran.Memo)
|
|
}
|
|
|
|
fmt.Printf("%s %-15s %-11s %s\n", tran.DtPosted, tran.TrnAmt.String()+" "+string(currency), tran.TrnType, name)
|
|
}
|