Better log output, fix writing to file.

antoniaelek · Mar 9, 2019 · 348024c · 348024c
1 parent cdd4e0e
commit 348024c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/export/artists.go b/export/artists.go
@@ -89,7 +89,8 @@ func GetArtists(user string, apiKey string) (artists []Artist, err error) {
 		for _, a := range ts {
 			cnt, err := strconv.Atoi(a.Playcount)
 			if err != nil {
-				continue
+				log.Printf("Error fetching play count for aritst %s\n", a.Name)
+				cnt = 0
 			}
 			artist := Artist{
 				Name:      a.Name,

diff --git a/main.go b/main.go
@@ -1,8 +1,8 @@
 package main
 
 import (
-	"bufio"
 	"flag"
+	"fmt"
 	"lastfmuserexport/export"
 	"log"
 	"os"
@@ -154,9 +154,11 @@ func SaveCsvFile(csv []string, filename string) error {
 		return err
 	}
 	defer f.Close()
-	w := bufio.NewWriter(f)
+
 	for _, line := range csv {
-		w.WriteString(line + "\n")
+		fmt.Fprintln(f, line)
 	}
+	defer f.Close()
+
 	return nil
 }