统一了一下函数风格

LeafYeeXYZ · Jun 9, 2024 · 84b630b · 84b630b
1 parent 2dab7ae
commit 84b630b
Show file tree

Hide file tree

Showing 2 changed files with 171 additions and 127 deletions.
diff --git a/catch.go b/catch.go
@@ -12,8 +12,11 @@ func (a *App) CatchCoursePub(speed int, studentID string, password string, cours
 
 	runtime.EventsEmit(a.ctx, "currentStatus", "开始抢课")
 
+	// 错误
+	var err error
+
 	// 安装浏览器
-	err := playwright.Install()
+	err = playwright.Install()
 	if err != nil { return err }
 
 	// 创建 Playwright 实例
@@ -34,124 +37,137 @@ func (a *App) CatchCoursePub(speed int, studentID string, password string, cours
 	// 为每个课程创建一个协程
 	for i := 0; i < len(courseID); i++ {
 		go func(speed int, studentID string, password string, courseID string, classID string) {
+			// 当前元素
+			var ele playwright.Locator
+			// 错误
+			var err error
+
+			// 创建页面实例
+			page, err := browser.NewPage()
+			if err != nil { errCh <- err; return }
+			runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("为课程 %s 创建新页面", courseID))
 
-		// 创建页面实例
-		page, err := browser.NewPage()
-		if err != nil { errCh <- err; return }
-		runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("为课程 %s 创建新页面", courseID))
+			// 浏览器出现 confirm 时, 点击 "确定"
+			page.On("dialog", func(dialog playwright.Dialog) {
+				dialog.Accept()
+			})
 
-		// 浏览器出现 confirm 时, 点击 "确定"
-		page.On("dialog", func(dialog playwright.Dialog) {
-			dialog.Accept()
-		})
+			// 跳转到登录页面
+			_, err = page.Goto("https://cas.bnu.edu.cn/cas/login?service=http%3A%2F%2Fzyfw.bnu.edu.cn%2F")
+			if err != nil { errCh <- err; return }
 
-		// 跳转到登录页面
-		_, err = page.Goto("https://cas.bnu.edu.cn/cas/login?service=http%3A%2F%2Fzyfw.bnu.edu.cn%2F")
-		if err != nil { errCh <- err; return }
+			// 输入学号
+			ele = page.Locator("#un")
+			err = ele.Fill(studentID)
+			if err != nil { errCh <- err; return }
 
-		// 输入学号
-		err = page.Locator("#un").Fill(studentID)
-		if err != nil { errCh <- err; return }
+			// 输入密码
+			ele = page.Locator("#pd")
+			err = ele.Fill(password)
+			if err != nil { errCh <- err; return }
 
-		// 输入密码
-		err = page.Locator("#pd").Fill(password)
-		if err != nil { errCh <- err; return }
+			// 点击登录按钮
+			ele = page.Locator("#index_login_btn")
+			err = ele.Click()
+			if err != nil { errCh <- err; return }
 
-		// 点击登录按钮
-		err = page.Locator("#index_login_btn").Click()
-		if err != nil { errCh <- err; return }
+			// 等待加载
+			page.WaitForLoadState(playwright.PageWaitForLoadStateOptions{
+				State: playwright.LoadStateNetworkidle,
+			})
 
-		// 等待加载
-		page.WaitForLoadState(playwright.PageWaitForLoadStateOptions{
-			State: playwright.LoadStateNetworkidle,
-		})
+			// 如果有, 点击 "继续访问原地址"
+			ele = page.Locator("body > div > div.mid_container > div > div > div > div.select_login_box > div:nth-child(6) > a")
+			if exists, _ := ele.IsVisible(); exists {
+				err = ele.Click()
+				if err != nil { errCh <- err; return }
+			}
 
-		// 如果有, 点击 "继续访问原地址"
-		ele := page.Locator("body > div > div.mid_container > div > div > div > div.select_login_box > div:nth-child(6) > a")
-		if exists, _ := ele.IsVisible(); exists {
+			// 点击 "网上选课"
+			ele = page.Locator("li[data-code=\"JW1304\"]")
 			err = ele.Click()
 			if err != nil { errCh <- err; return }
-		}
 
-		// 点击 "网上选课"
-		err = page.Locator("li[data-code=\"JW1304\"]").Click()
-		if err != nil { errCh <- err; return }
-
-		// 获取 iframe
-		iframe := page.Frame(playwright.PageFrameOptions{
-			Name: playwright.String("frmDesk"),
-		})
-		runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("课程 %s 进入选课界面", courseID))
-
-		// 点击 "抢公共选修课"
-		err = iframe.Locator("#title1803").Click()
-		if err != nil { errCh <- err; return }
-
-		// 等待加载
-		iframe.WaitForLoadState(playwright.FrameWaitForLoadStateOptions{
-			State: playwright.LoadStateNetworkidle,
-		})
-
-		// 等待选课时间
-		for {
-			// 课程代码输入框
-			ele = iframe.Locator("#kcmc")
-			// 如果没到时间, 刷新
-			if disabled, _ := ele.IsDisabled(); disabled {
-				runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("未到选课时间, 刷新课程 %s 页面 (关闭小鸦抢课即可停止)", courseID))
-				page.Locator("#JW130415").Click()
-				// 等待加载
-				page.WaitForLoadState(playwright.PageWaitForLoadStateOptions{
-					State: playwright.LoadStateNetworkidle,
-				})
-				continue
-			// 如果到时间, 输入课程号
-			} else {
-				time.Sleep(time.Duration(speed) * time.Millisecond)
+			// 获取 iframe
+			iframe := page.Frame(playwright.PageFrameOptions{
+				Name: playwright.String("frmDesk"),
+			})
+			runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("课程 %s 进入选课界面", courseID))
+
+			// 点击 "抢公共选修课"
+			ele = iframe.Locator("#title1785")
+			err = ele.Click()
+			if err != nil { errCh <- err; return }
+
+			// 等待加载
+			iframe.WaitForLoadState(playwright.FrameWaitForLoadStateOptions{
+				State: playwright.LoadStateNetworkidle,
+			})
+
+			// 等待选课时间
+			for {
+				// 课程代码输入框
+				ele = iframe.Locator("#kcmc")
+				// 如果没到时间, 刷新
 				if disabled, _ := ele.IsDisabled(); disabled {
+					runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("未到选课时间, 刷新课程 %s 页面 (关闭小鸦抢课即可停止)", courseID))
+				  ele = page.Locator("#JW130415")
+					err = ele.Click()
+					if err != nil { errCh <- err; return }
+					// 等待加载
+					page.WaitForLoadState(playwright.PageWaitForLoadStateOptions{
+						State: playwright.LoadStateNetworkidle,
+					})
 					continue
+				// 如果到时间, 输入课程号
+				} else {
+					time.Sleep(time.Duration(speed) * time.Millisecond)
+					if disabled, _ := ele.IsDisabled(); disabled {
+						continue
+					}
+					err = ele.Fill(courseID)
+					if err != nil { errCh <- err; return }
+					break
 				}
-				err = ele.Fill(courseID)
-				if err != nil { errCh <- err; return }
-				break
-			}
-		}
-		// 输入班号
-		err = iframe.Locator("#t_skbh").Fill(classID)
-		if err != nil { errCh <- err; return }
-
-		// 点击 "检索"
-		err = iframe.Locator("#btnQry").Click()
-		if err != nil { errCh <- err; return }
-
-		// 等待加载
-		page.WaitForLoadState(playwright.PageWaitForLoadStateOptions{
-			State: playwright.LoadStateNetworkidle,
-		})
-
-		// 获取子 iframe
-		iiframe := iframe.FrameLocator("#frmReport")
-
-		// 点击 "选择"
-		runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("检索课程 %s", courseID))
-		ele = iiframe.Locator("#tr0_xz a")
-		count := 0
-		for {
-			if count > 10000 { 
-				runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("课程 %s 网路超时或可选人数为零", courseID))
-				errCh <- fmt.Errorf("课程 %s 网路超时或可选人数为零", courseID)
-				return
 			}
-			if exists, _ := ele.IsVisible(); exists {
-				err = ele.Click()
-				if err != nil { errCh <- err; return }
-				break
-			} else {
-				runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("等待检索课程 %s 结果...", courseID))
-				time.Sleep(time.Duration(speed) * time.Millisecond)
-				count += speed
+			// 输入班号
+			ele = iframe.Locator("#t_skbh")
+			err = ele.Fill(classID)
+			if err != nil { errCh <- err; return }
+
+			// 点击 "检索"
+			ele = iframe.Locator("#btnQry")
+			err = ele.Click()
+			if err != nil { errCh <- err; return }
+
+			// 等待加载
+			page.WaitForLoadState(playwright.PageWaitForLoadStateOptions{
+				State: playwright.LoadStateNetworkidle,
+			})
+
+			// 获取子 iframe
+			iiframe := iframe.FrameLocator("#frmReport")
+
+			// 点击 "选择"
+			runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("检索课程 %s", courseID))
+			ele = iiframe.Locator("#tr0_xz a")
+			count := 0
+			for {
+				if count > 10000 { 
+					runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("课程 %s 网路超时或可选人数为零", courseID))
+					errCh <- fmt.Errorf("课程 %s 网路超时或可选人数为零", courseID)
+					return
+				}
+				if exists, _ := ele.IsVisible(); exists {
+					err = ele.Click()
+					if err != nil { errCh <- err; return }
+					break
+				} else {
+					runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("等待检索课程 %s 结果...", courseID))
+					time.Sleep(time.Duration(speed) * time.Millisecond)
+					count += speed
+				}
 			}
-		}
 
 			// 等待加载
 			runtime.EventsEmit(a.ctx, "currentStatus", fmt.Sprintf("成功选择课程 %s", courseID))