l9_stud_bot/modules/ssau_parser/database.go

package ssau_parser

import (
	"log"
	"strings"

	"git.l9labs.ru/anufriev.g.a/l9_stud_bot/modules/database"
	"xorm.io/xorm"
)

func UpdateSchedule(db *xorm.Engine, sh WeekShedule) error {
	err := checkGroupOrTeacher(db, sh)
	if err != nil {
		return err
	}

	first_new := sh.Uncovered[0]
	_, week := first_new.Begin.ISOWeek()
	var old []database.Lesson
	db.Where("WEEK(`Begin`) = ?", week).Asc("Begin").Find(&old)
	Compare(sh.Uncovered, old)
	return nil
}

func UploadShedule(db *xorm.Engine, sh WeekShedule) error {
	err := checkGroupOrTeacher(db, sh)
	if err != nil {
		return err
	}

	var pairs []database.Lesson
	for _, line := range sh.Lessons {
		for _, lesson := range line {
			var pair database.Lesson
			for _, subLesson := range lesson.Lessons {
				pair = database.Lesson{
					Begin:     lesson.Begin,
					End:       lesson.End,
					Type:      subLesson.Type,
					Name:      subLesson.Name,
					TeacherId: subLesson.TeacherId[0],
				}

				exists, err := isTeacherExists(db, subLesson.TeacherId[0])
				if err != nil {
					return err
				}

				if !exists && subLesson.TeacherId[0] != 0 {
					uri := GenerateUri(subLesson.TeacherId[0], true)
					doc, err := DownloadShedule(uri, sh.Week)
					if err != nil {
						return err
					}
					var gr WeekShedule
					gr.IsGroup = false
					gr.SheduleId = subLesson.TeacherId[0]
					GetSheduleInfo(doc.Doc, &gr)
					checkGroupOrTeacher(db, gr)
				}

				for _, groupId := range subLesson.GroupId {
					pair.GroupId = groupId

					exists, err := isGroupExists(db, groupId)
					if err != nil {
						return err
					}

					if !exists {
						uri := GenerateUri(groupId, false)
						doc, err := DownloadShedule(uri, sh.Week)
						if err != nil {
							return err
						}
						var gr WeekShedule
						gr.IsGroup = true
						gr.SheduleId = groupId
						GetSheduleInfo(doc.Doc, &gr)
						checkGroupOrTeacher(db, gr)
					}

					var existsLessons []database.Lesson
					err = db.Find(&existsLessons, pair)
					if err != nil {
						return err
					}
					if len(existsLessons) == 0 {
						pair.NumInShedule = lesson.NumInShedule
						pair.Place = subLesson.Place
						pair.Comment = subLesson.Comment
						pair.SubGroup = int64(subLesson.SubGroup[0])
						pairs = append(pairs, pair)
					}
				}
			}
		}
	}
	if len(pairs) > 0 {
		_, err := db.Insert(&pairs)
		return err
	}
	return nil
}

func isGroupExists(db *xorm.Engine, groupId int64) (bool, error) {
	var exists []database.Group
	err := db.Find(&exists, database.Group{GroupId: groupId})
	if err != nil {
		return false, err
	}

	return len(exists) == 1, nil
}

func isTeacherExists(db *xorm.Engine, teacherId int64) (bool, error) {
	var exists []database.Teacher
	err := db.Find(&exists, database.Teacher{TeacherId: teacherId})
	if err != nil {
		return false, err
	}

	return len(exists) == 1, nil
}

// Проверка наличия группы или преподавателя в БД и добавление при необходимости
// TODO: Добавить проверку изменений в полях данных
func checkGroupOrTeacher(db *xorm.Engine, sh WeekShedule) error {
	if sh.IsGroup {
		exists, err := isGroupExists(db, sh.SheduleId)
		if err != nil {
			return err
		}

		if !exists {
			group := database.Group{
				GroupId:   sh.SheduleId,
				GroupName: sh.FullName,
				SpecName:  sh.SpecName,
			}
			db.InsertOne(group)
		}
	} else {
		exists, err := isTeacherExists(db, sh.SheduleId)
		if err != nil {
			return err
		}

		if !exists {
			name := strings.Split(sh.FullName, " ")
			log.Println(name)
			teacher := database.Teacher{
				TeacherId: sh.SheduleId,
				FirstName: name[0],
				LastName:  strings.Join(name[1:], " "),
				SpecName:  sh.SpecName,
			}
			db.InsertOne(teacher)
		}

	}
	return nil
}