start working on submitting course without user-provided course code

1 files changed, 27 insertions, 1 deletions

diff --git a/aplus.go b/aplus.go
index dfc73e1..29d5033 100644
--- a/aplus.go
+++ b/aplus.go
@@ -11,17 +11,27 @@ import (
         "strings"
 )
 
-func submit_code(course_code int, attendance_code string) {
+func submit_code(course_code int, attendance_code string) bool {
         cur_body := launch_aplus(course_code)
 
         if verbose {
                 fmt.Printf("%s", cur_body)
         }
 
+        if strings.Contains(cur_body, "Student not known") {
+                fmt.Println("Student not known. This course must first be visited by an instructor for students to have access.")
+                return false
+        }
+
         current_url := fmt.Sprintf("%s/student/", base_aplus_link)
         // the links to submit the code for a class is under the dayPanel div
         daypanel_start := strings.Index(cur_body, "<div class=\"dayPanel\"")
 
+        if !strings.Contains(cur_body[daypanel_start:], "<a href=\"./?module") {
+                fmt.Println("No code submitted. Did the professor generate the code or is the class in session?")
+                return false
+        }
+
         // link_start - link_end is to extract the URL
         // TODO: make this more elegant to allow multiple of these links
         //       (maybe use an html parser)
@@ -62,6 +72,8 @@ func submit_code(course_code int, attendance_code string) {
         } else if strings.Index(string(body), "ctl02_codeSuccessMessage") != -1 {
                 fmt.Println("Code successfully recorded")
         }
+
+        return true
 }
 
 func launch_aplus(course_code int) string {
@@ -148,3 +160,17 @@ func extract_attribute(input string, attribute string) string {
 
         return input[start : start+end]
 }
+
+func submit_code_sans_course(attendance_code string) {
+        courses := list_all_courses(true)
+
+        for _, course := range courses {
+                fmt.Println("Checking with course", course.ID%10000)
+                if submit_code(int(course.ID%10000), attendance_code) == true {
+                        fmt.Println("Submitted code to course", course.ID%10000)
+                        return
+                }
+        }
+
+        fmt.Println("Could not submit code %s to any course\n", attendance_code)
+}