From 0f951dcf029d4af284467543a3afdf5bf6581a20 Mon Sep 17 00:00:00 2001 From: Matthew Lemon Date: Tue, 23 Apr 2024 11:16:38 +0100 Subject: switched to Django --- internal/models/engagement.go | 54 ------------------------------------------- 1 file changed, 54 deletions(-) delete mode 100644 internal/models/engagement.go (limited to 'internal/models/engagement.go') diff --git a/internal/models/engagement.go b/internal/models/engagement.go deleted file mode 100644 index 87e0786..0000000 --- a/internal/models/engagement.go +++ /dev/null @@ -1,54 +0,0 @@ -package models - -import ( - "database/sql" - "fmt" - "strconv" - "time" -) - -type EngagementStrategy struct { - ID int - ValidFrom time.Time - ValidTo time.Time - Operation Operation -} - -func (es *EngagementStrategy) FormatForTable() string { - startYear := strconv.Itoa(es.ValidFrom.Year()) - endYear := strconv.Itoa(es.ValidTo.Year()) - return fmt.Sprintf("%s-%s", startYear, endYear) -} - -type EngagementStrategyModel struct { - DB *sql.DB -} - -func (m *EngagementStrategyModel) GetForOperation(id int) ([]EngagementStrategy, error) { - stmt := `SELECT id, valid_from, valid_to FROM engagement_strategies -WHERE operation_id = ?` - rows, err := m.DB.Query(stmt, id) - if err != nil { - return nil, err - } - - defer rows.Close() - - var esses []EngagementStrategy - - for rows.Next() { - var es EngagementStrategy - err = rows.Scan(&es.ID, &es.ValidFrom, &es.ValidTo) - if err != nil { - return nil, err - } - - esses = append(esses, es) - } - - if err = rows.Err(); err != nil { - return nil, err - } - - return esses, nil -} -- cgit v1.2.3