send attendance code to all current courses when course not specified

1 files changed, 5 insertions, 3 deletions

diff --git a/aplus.go b/aplus.go
index 29d5033..2e2ab0a 100644
--- a/aplus.go
+++ b/aplus.go
@@ -7,12 +7,14 @@ import (
         "io"
         "net/http"
         "net/url"
-        "os"
         "strings"
 )
 
 func submit_code(course_code int, attendance_code string) bool {
         cur_body := launch_aplus(course_code)
+        if cur_body == "" {
+                return false
+        }
 
         if verbose {
                 fmt.Printf("%s", cur_body)
@@ -81,7 +83,7 @@ func launch_aplus(course_code int) string {
         if err != nil {
                 fmt.Println("Failed to get A+ base URL and tool ID. Try again with a different course code.")
                 fmt.Printf("Error: %s\n", err)
-                os.Exit(1)
+                return ""
         }
 
         aplus_link := fmt.Sprintf("%s/courses/%d/external_tools/sessionless_launch?id=%d&access_token=%s",
@@ -172,5 +174,5 @@ func submit_code_sans_course(attendance_code string) {
                 }
         }
 
-        fmt.Println("Could not submit code %s to any course\n", attendance_code)
+        fmt.Println("Could not submit code", attendance_code, "to any course")
 }