CHANGE: added report section
This commit is contained in:
@@ -1,27 +1,45 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"log"
|
||||
"time"
|
||||
)
|
||||
|
||||
type WorkWeek struct {
|
||||
Id int
|
||||
WorkDays []WorkDay
|
||||
User User
|
||||
WeekStart time.Time
|
||||
WorkHours time.Duration
|
||||
}
|
||||
|
||||
func (w *WorkWeek) GetWeek(user User, tsMonday time.Time) WorkWeek {
|
||||
func (w *WorkWeek) GetWeek(user User, tsMonday time.Time, populateDays bool) WorkWeek {
|
||||
var week WorkWeek
|
||||
week.WorkDays = (*WorkDay).GetWorkDays(nil, user.CardUID, tsMonday, tsMonday.Add(7*24*time.Hour))
|
||||
if populateDays {
|
||||
week.WorkDays = (*WorkDay).GetWorkDays(nil, user.CardUID, tsMonday, tsMonday.Add(7*24*time.Hour))
|
||||
week.WorkHours = aggregateWorkTime(week.WorkDays)
|
||||
}
|
||||
week.User = user
|
||||
week.WeekStart = tsMonday
|
||||
return week
|
||||
}
|
||||
|
||||
func (w *WorkWeek) GetWorkHourString() string {
|
||||
return formatDuration(w.WorkHours)
|
||||
}
|
||||
|
||||
func aggregateWorkTime(days []WorkDay) time.Duration {
|
||||
var workTime time.Duration
|
||||
for _, day := range days {
|
||||
workTime += day.workTime
|
||||
}
|
||||
return workTime
|
||||
}
|
||||
|
||||
func (w *WorkWeek) GetSendWeeks(user User) []WorkWeek {
|
||||
var weeks []WorkWeek
|
||||
qStr, err := DB.Prepare(`SELECT woche_start::DATE FROM buchung_wochen WHERE bestaetigt = FALSE AND personal_nummer = $1;`)
|
||||
qStr, err := DB.Prepare(`SELECT id, woche_start::DATE FROM wochen_report WHERE bestaetigt = FALSE AND personal_nummer = $1;`)
|
||||
if err != nil {
|
||||
log.Println("Error preparing SQL statement", err)
|
||||
return weeks
|
||||
@@ -37,11 +55,12 @@ func (w *WorkWeek) GetSendWeeks(user User) []WorkWeek {
|
||||
for rows.Next() {
|
||||
var week WorkWeek
|
||||
week.User = user
|
||||
if err := rows.Scan(&week.WeekStart); err != nil {
|
||||
if err := rows.Scan(&week.Id, &week.WeekStart); err != nil {
|
||||
log.Println("Error scanning row!", err)
|
||||
return weeks
|
||||
}
|
||||
week.WorkDays = (*WorkDay).GetWorkDays(nil, user.CardUID, week.WeekStart, week.WeekStart.Add(7*24*time.Hour))
|
||||
week.WorkHours = aggregateWorkTime(week.WorkDays)
|
||||
weeks = append(weeks, week)
|
||||
}
|
||||
if err = rows.Err(); err != nil {
|
||||
@@ -50,3 +69,39 @@ func (w *WorkWeek) GetSendWeeks(user User) []WorkWeek {
|
||||
|
||||
return weeks
|
||||
}
|
||||
|
||||
var ErrRunningWeek = errors.New("Week is in running week")
|
||||
|
||||
// creates a new entry in the woche_report table with the given workweek
|
||||
func (w *WorkWeek) Send() error {
|
||||
if time.Since(w.WeekStart) < 5*24*time.Hour {
|
||||
log.Println("Cannot send week, because it's the running week!")
|
||||
return ErrRunningWeek
|
||||
}
|
||||
qStr, err := DB.Prepare(`INSERT INTO wochen_report (personal_nummer, woche_start) VALUES ($1, $2);`)
|
||||
if err != nil {
|
||||
log.Println("Error preparing SQL statement", err)
|
||||
return err
|
||||
}
|
||||
_, err = qStr.Exec(w.User.PersonalNummer, w.WeekStart)
|
||||
if err != nil {
|
||||
log.Println("Error executing query!", err)
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
|
||||
}
|
||||
|
||||
func (w *WorkWeek) Accept() error {
|
||||
qStr, err := DB.Prepare(`UPDATE "wochen_report" SET bestaetigt = TRUE WHERE personal_nummer = $1 AND woche_start = $2;`)
|
||||
if err != nil {
|
||||
log.Println("Error preparing SQL statement", err)
|
||||
return err
|
||||
}
|
||||
_, err = qStr.Exec(w.User.PersonalNummer, w.WeekStart)
|
||||
if err != nil {
|
||||
log.Println("Error executing query!", err)
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user