67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
|
package grpc
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"encoding/json"
|
|||
|
|
"log/slog"
|
|||
|
|
|
|||
|
|
pb "schedule_converter/proto/runner"
|
|||
|
|
|
|||
|
|
"schedule_converter/config"
|
|||
|
|
"schedule_converter/parser"
|
|||
|
|
apperr "schedule_converter/pkg/errors"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func (h *TaskHandler) getGrades(ctx context.Context, payload []byte) ([]byte, error) {
|
|||
|
|
var req pb.GetGradesPayload
|
|||
|
|
if err := unmarshalPayload(payload, &req); err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log := slog.With("username", req.Username, "semester", req.Semester)
|
|||
|
|
log.Info("获取成绩任务")
|
|||
|
|
|
|||
|
|
httpClient, err := createHTTPClient()
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
cfg := config.Get()
|
|||
|
|
setTWFIDCookie(httpClient, cfg.BaseURL, "")
|
|||
|
|
|
|||
|
|
if err := executeLogin(ctx, httpClient, req.Username, req.Password); err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var gpaSummary *pb.GpaSummary
|
|||
|
|
summary, grades, err := parser.FetchGPA(httpClient)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Warn("获取GPA页面失败，尝试成绩页面", "error", err)
|
|||
|
|
grades, err = parser.FetchGrades(httpClient, parser.ScoreTypeFinal)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, apperr.New("fetch_grades", err, "获取成绩失败")
|
|||
|
|
}
|
|||
|
|
} else if summary != nil {
|
|||
|
|
gpaSummary = &pb.GpaSummary{
|
|||
|
|
AverageGpa: summary.AverageGPA,
|
|||
|
|
Rank: summary.Rank,
|
|||
|
|
ExamTotalGpa: summary.ExamTotalGPA,
|
|||
|
|
ExamTotalCredits: summary.ExamTotalCredits,
|
|||
|
|
FailCredits: summary.FailCredits,
|
|||
|
|
Semesters: summary.Semesters,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
pbGrades := convertGrades(grades)
|
|||
|
|
|
|||
|
|
result := &pb.GradesResultData{
|
|||
|
|
StudentId: req.Username,
|
|||
|
|
Semester: req.Semester,
|
|||
|
|
Grades: pbGrades,
|
|||
|
|
GpaSummary: gpaSummary,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log.Info("成绩获取成功", "grade_count", len(pbGrades))
|
|||
|
|
return json.Marshal(result)
|
|||
|
|
}
|