Adding initial code for the main.go as well as initial code for uploading members

This commit is contained in:
2026-01-19 20:03:40 -05:00
parent 6b17c6fddc
commit 0359efe197
9 changed files with 572 additions and 2 deletions

View File

@@ -0,0 +1,45 @@
package services
import (
"encoding/csv"
"fmt"
"strings"
"github.com/pkg/errors"
)
type Member struct {
Name string
Email string
}
func ParseMembersFromBytes(year int, fileBytes []byte) ([]Member, error) {
reader := csv.NewReader(strings.NewReader(string(fileBytes)))
records, err := reader.ReadAll()
if err != nil {
return nil, errors.Wrap(err, "failed to read CSV from bytes")
}
var members []Member
for i, record := range records {
if i == 0 {
continue // Skip the first line (column headers)
}
if len(record) < 4 {
continue
}
firstName := record[1]
lastName := record[2]
email := record[3]
members = append(members, Member{
Name: fmt.Sprintf("%s %s", firstName, lastName),
Email: email,
})
}
if len(members) == 0 {
members = []Member{}
}
return members, nil
}

View File

@@ -0,0 +1,45 @@
package services
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestParseMembersFromBytes(t *testing.T) {
testCases := []struct {
name string
input string
expected []Member
}{
{
name: "Valid CSV with multiple members",
input: `date,First,Last,Email
2023-01-01,John,Doe,john.doe@example.com
2023-01-02,Jane,Smith,jane.smith@example.com`,
expected: []Member{
{Name: "John Doe", Email: "john.doe@example.com"},
{Name: "Jane Smith", Email: "jane.smith@example.com"},
},
},
{
name: "CSV with missing fields",
input: `date,First,Last
2023-01-01,John,Doe`,
expected: []Member{},
},
{
name: "Empty CSV",
input: ``,
expected: []Member{},
},
}
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
members, err := ParseMembersFromBytes(2023, []byte(tc.input))
assert.NoError(t, err)
assert.Equal(t, tc.expected, members)
})
}
}

View File

@@ -109,7 +109,7 @@ func PreLoadDB() error {
// Insert members
for i := range new_members {
_, err := db_conn.Exec(`INSERT INTO members (email, member_name) VALUES (?, ?)`, new_members[i].email, new_members[i].member_name)
_, err := db_conn.Exec(`INSERT INTO members (email, member_name, school_year) VALUES (?, ?, ?)`, new_members[i].email, new_members[i].member_name, 2023)
if err != nil {
return err
}