Merge pull request #9 from ViolaPioggia/main

fix: fix the function of getting all the Renwen and Ziran courses
LgoLgo · May 6, 2023 · 413b7ce · 413b7ce
2 parents 5121330 + 76cabba
commit 413b7ce
Showing 1 changed file with 14 additions and 12 deletions.
diff --git a/query/queryer.go b/query/queryer.go
@@ -50,8 +50,8 @@ func (q *Queryer) AllRenWen(cookie string) {
 	if err != nil {
 		log.Fatal(err)
 	}
+	var builder strings.Builder
 	for _, item := range classInfo.Data {
-		var builder strings.Builder
 		SRsLimit := strconv.Itoa(item.RsLimit)
 		SRwType := strconv.Itoa(item.RwType)
 
@@ -64,11 +64,12 @@ func (q *Queryer) AllRenWen(cookie string) {
 		for _, str := range strs {
 			builder.WriteString(str)
 		}
-		loads := builder.String()
-		err = ioutil.WriteFile("./output_renwen.txt", []byte(loads), 0o666) // 写入文件(字节数组)
-		if err != nil {
-			log.Fatal(err.Error())
-		}
+		builder.WriteString("\n")
+	}
+	loads := builder.String()
+	err = ioutil.WriteFile("./output_renwen.txt", []byte(loads), 0o666) // 写入文件(字节数组)
+	if err != nil {
+		log.Fatal(err.Error())
 	}
 	return
 }
@@ -80,8 +81,8 @@ func (q *Queryer) AllZiRan(cookie string) {
 	if err != nil {
 		log.Fatal(err)
 	}
+	var builder strings.Builder
 	for _, item := range classInfo.Data {
-		var builder strings.Builder
 		SRsLimit := strconv.Itoa(item.RsLimit)
 		SRwType := strconv.Itoa(item.RwType)
 
@@ -94,11 +95,12 @@ func (q *Queryer) AllZiRan(cookie string) {
 		for _, str := range strs {
 			builder.WriteString(str)
 		}
-		loads := builder.String()
-		err = ioutil.WriteFile("./output_ziran.txt", []byte(loads), 0o666) // 写入文件(字节数组)
-		if err != nil {
-			log.Fatal(err.Error())
-		}
+		builder.WriteString("\n")
+	}
+	loads := builder.String()
+	err = ioutil.WriteFile("./output_ziran.txt", []byte(loads), 0o666) // 写入文件(字节数组)
+	if err != nil {
+		log.Fatal(err.Error())
 	}
 	return
 }