63 行
1.5 KiB
Go
63 行
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"integrity-checkin/dao"
|
|
"integrity-checkin/data/po"
|
|
"io"
|
|
"os"
|
|
"sort"
|
|
)
|
|
|
|
type UserActivityWithRealName struct {
|
|
RealName string `json:"real_name"`
|
|
po.UserActivity
|
|
}
|
|
|
|
func main() {
|
|
dao.Setup()
|
|
var userActivities []UserActivityWithRealName
|
|
err := dao.DB.Raw("select u.real_name,ua.* from user_activity ua " +
|
|
" left join users u on u.id=ua.user_id where u.type=2").Find(&userActivities).Error
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
days := map[string]struct{}{}
|
|
teacherDays := map[string]map[string]struct{}{}
|
|
for _, userActivity := range userActivities {
|
|
if _, ok := days[userActivity.Day]; !ok {
|
|
days[userActivity.Day] = struct{}{}
|
|
}
|
|
if teacherDays[userActivity.RealName] == nil {
|
|
teacherDays[userActivity.RealName] = map[string]struct{}{}
|
|
}
|
|
teacherDays[userActivity.RealName][userActivity.Day] = struct{}{}
|
|
}
|
|
var daysArr []string
|
|
for day := range days {
|
|
daysArr = append(daysArr, day)
|
|
}
|
|
sort.Slice(daysArr, func(i, j int) bool {
|
|
return daysArr[i] < daysArr[j]
|
|
})
|
|
f, err := os.OpenFile("out.csv", os.O_WRONLY|os.O_CREATE, 0644)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
io.WriteString(f, "姓名")
|
|
for _, day := range daysArr {
|
|
io.WriteString(f, ","+day)
|
|
}
|
|
io.WriteString(f, "\n")
|
|
for teacherName, teacherMap := range teacherDays {
|
|
io.WriteString(f, teacherName)
|
|
for _, day := range daysArr {
|
|
if _, ok := teacherMap[day]; ok {
|
|
io.WriteString(f, ",1")
|
|
} else {
|
|
io.WriteString(f, ",0")
|
|
}
|
|
}
|
|
io.WriteString(f, "\n")
|
|
}
|
|
}
|