Adding initial code for the main.go as well as initial code for uploading members
This commit is contained in:
45
server/services/putmembers.go
Normal file
45
server/services/putmembers.go
Normal 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
|
||||
}
|
||||
45
server/services/putmembers_test.go
Normal file
45
server/services/putmembers_test.go
Normal 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)
|
||||
})
|
||||
}
|
||||
}
|
||||
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user