Import records from a CSV file

suminb · Apr 30, 2017 · bdc0824 · bdc0824
1 parent 0419114
commit bdc0824
Show file tree

Hide file tree

Showing 4 changed files with 62 additions and 3 deletions.
diff --git a/goport/src/finance/importer.go b/goport/src/finance/importer.go
@@ -96,7 +96,37 @@ func ImportAccounts(filePath string) error {
 	return nil
 }
 
-func ImportStockValues(filePath string, symbol string) {
+// Imports trading records from a CSV file.
+func ImportRecords(db *DB, filePath string) error {
+	ch := make(chan []string)
+	go ReadCSV(filePath, ch)
+	for row := range ch {
+		// "Test Account","AMZN",DEPOSIT,2017-02-24,1
+		accountName := strings.TrimSpace(row[0])
+		assetName := strings.TrimSpace(row[1])
+		recordType := RecordType(strings.TrimSpace(row[2]))
+		createdAt, _ := time.Parse("2006-01-02", strings.TrimSpace(row[3]))
+		quantity, _ := strconv.ParseInt(strings.TrimSpace(row[4]), 10, 64)
+
+		account := db.GetAccountByName(accountName)
+		asset := db.GetAssetByName(assetName)
+
+		record := Record{
+			Account:   account,
+			Asset:     asset,
+			Type:      recordType,
+			CreatedAt: createdAt,
+			Quantity:  int(quantity),
+		}
+		res := db.Raw.Create(&record)
+		if res.Error != nil {
+			return res.Error
+		}
+	}
+	return nil
+}
+
+func ImportStockValues(filePath string, symbol string) error {
 	db := ConnectDatabase()
 	defer db.Raw.Close()
 
@@ -109,7 +139,10 @@ func ImportStockValues(filePath string, symbol string) {
 	ch := make(chan AssetValue)
 	go ReadStockValues(filePath, ch)
 	for v := range ch {
-		fmt.Println("Processing", v)
-		db.Raw.Create(&v)
+		res := db.Raw.Create(&v)
+		if res.Error != nil {
+			return res.Error
+		}
 	}
+	return nil
 }
diff --git a/goport/src/finance/importer_test.go b/goport/src/finance/importer_test.go
@@ -49,3 +49,15 @@ func TestImportAccounts(t *testing.T) {
 		}
 	}
 }
+
+func TestImportRecords(t *testing.T) {
+	db := ConnectDatabase()
+	defer db.Raw.Close()
+
+	filePath := "test-data/records.csv"
+
+	err := ImportRecords(db, filePath)
+	if err != nil {
+		t.Error(err)
+	}
+}
diff --git a/goport/src/finance/models.go b/goport/src/finance/models.go
@@ -166,10 +166,18 @@ type DB struct {
 // Returns an `Asset` instance matching the given name.
 func (db *DB) GetAssetByName(name string) Asset {
 	var asset Asset
+	// FIXME: Deal with cases where no matching record found
 	db.Raw.First(&asset, "name = ?", name)
 	return asset
 }
 
+func (db *DB) GetAccountByName(name string) Account {
+	var account Account
+	// FIXME: Deal with cases where no matching record found
+	db.Raw.First(&account, "name = ?", name)
+	return account
+}
+
 func (db *DB) InsertAsset(name string, description string) (Asset, []error) {
 	asset := Asset{
 		Name:        name,

diff --git a/goport/src/finance/test-data/records.csv b/goport/src/finance/test-data/records.csv
@@ -0,0 +1,6 @@
+"Test Account","AMZN",deposit,2017-02-24,1
+"Test Account","USD",withdraw,2017-02-24,800
+"Test Account","AMZN",deposit,2017-03-24,1
+"Test Account","USD",withdraw,2017-03-24,850
+"Test Account","AMZN",deposit,2017-04-24,1
+"Test Account","USD",withdraw,2017-04-24,900