andrea/oef
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge branch 'master' into production

Browse source
This commit is contained in:
Andrea Fazzi 2022-05-03 15:16:21 +02:00
commit d63c3b8269
4 changed files with 137 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
VERSION
View file

@ -1 +1,2 @@
0.9.9-137-g4b91066-production
0.9.9-140-ga15cb8b-master

1
orm/participant.go
View file

@ -330,6 +330,7 @@ func (model *Participant) ReadAll(db *Database, args map[string]string, w http.R
}
} else {
if err := db._db.
Preload("User").
Preload("Category").
Preload("School").
Preload("School.Region").

131
scripts/subscribe_participants/main.go
View file

@ -2,11 +2,30 @@ package main
import (
"flag"
"fmt"
"io/ioutil"
"log"
"os"
"strings"
"git.andreafazzi.eu/andrea/oef/client"
"git.andreafazzi.eu/andrea/oef/orm"
"github.com/gocarina/gocsv"
"gopkg.in/yaml.v2"
)
type RegionContest struct {
Id uint
Category string
Regions []string
}
type User struct {
Username string `csv:"username"`
}
var (
regionContests map[string]([]*RegionContest)
)
func alreadySubscribed(participant *orm.Participant, contestId uint) bool {
@ -22,11 +41,15 @@ func main() {
username := flag.String("username", "admin", "Username")
password := flag.String("password", "admin", "Password")
subscribe := flag.Bool("subscribe", true, "Subscribe to the given contest")
region := flag.Bool("region", true, "Region based subscription")
unsubscribeAll := flag.Bool("unsubscribe-all", false, "Unsubscribe all participants from all contests")
juniorContestId := flag.Int("jid", 0, "Contest ID for Junior category")
seniorContestId := flag.Int("sid", 0, "Contest ID for Senior category")
participantsCSV := flag.String("participants-csv", "", "Import participants from CSV file")
// contestId := flag.Int("cid", 0, "Subscribe participants to the given contest ID")
flag.Parse()
client, err := client.Dial(flag.Arg(0), *username, *password)
@ -34,12 +57,34 @@ func main() {
panic(err)
}
participants := make([]*orm.Participant, 0)
err = client.ReadAll(&participants)
allParticipants := make([]*orm.Participant, 0)
log.Println("Read all participants...")
err = client.ReadAll(&allParticipants)
if err != nil {
panic(err)
}
if *participantsCSV != "" {
participants := make([]*orm.Participant, 0)
users := make([]User, 0)
data, err := ioutil.ReadFile(*participantsCSV)
if err != nil {
panic(err)
}
if err := gocsv.Unmarshal(strings.NewReader(string(data)), &users); err != nil {
panic(err)
}
for _, u := range users {
for _, p := range allParticipants {
if u.Username == p.User.Username {
participants = append(participants, p)
}
}
}
log.Println(participants)
}
var updated bool
if *unsubscribeAll {
@ -54,27 +99,47 @@ func main() {
os.Exit(0)
}
if *subscribe {
if *region {
data, err := ioutil.ReadFile("regions_contests.yaml")
if err != nil {
panic(err)
}
err = yaml.Unmarshal(data, &regionContests)
if err != nil {
panic(err)
}
regions_ids := make(map[string](map[string]uint), 0)
for _, contest := range regionContests["contests"] {
for _, region := range contest.Regions {
if regions_ids[region] == nil {
regions_ids[region] = make(map[string]uint)
}
if contest.Category == "Junior" {
regions_ids[region]["Junior"] = contest.Id
} else {
regions_ids[region]["Senior"] = contest.Id
}
}
}
for _, participant := range participants {
for _, contest := range participant.Contests {
participant.ContestIDs = append(participant.ContestIDs, contest.ID)
}
if *juniorContestId > 0 && participant.Category.Name == "Junior" {
if !alreadySubscribed(participant, uint(*juniorContestId)) {
log.Printf("Subscribe %s to contest ID %d", participant, *juniorContestId)
participant.ContestIDs = append(participant.ContestIDs, uint(*juniorContestId))
updated = true
}
}
if *seniorContestId > 0 && participant.Category.Name == "Senior" {
if !alreadySubscribed(participant, uint(*seniorContestId)) {
log.Printf("Subscribe %s to contest ID %d", participant, *seniorContestId)
participant.ContestIDs = append(participant.ContestIDs, uint(*seniorContestId))
updated = true
}
if regions_ids[participant.School.Region.Name] == nil {
panic(fmt.Errorf("Region not found for participant %v", participant))
}
contestId := regions_ids[participant.School.Region.Name][participant.Category.Name]
if !alreadySubscribed(participant, contestId) {
// log.Printf("Subscribe %s to contest ID %d", participant, contestId)
participant.ContestIDs = append(participant.ContestIDs, contestId)
updated = true
}
if updated {
id, err := client.Update(participant)
if err != nil {
@ -84,6 +149,40 @@ func main() {
updated = false
}
}
} else {
if *subscribe {
for _, participant := range participants {
for _, contest := range participant.Contests {
participant.ContestIDs = append(participant.ContestIDs, contest.ID)
}
if *juniorContestId > 0 && participant.Category.Name == "Junior" {
if !alreadySubscribed(participant, uint(*juniorContestId)) {
log.Printf("Subscribe %s to contest ID %d", participant, *juniorContestId)
participant.ContestIDs = append(participant.ContestIDs, uint(*juniorContestId))
updated = true
}
}
if *seniorContestId > 0 && participant.Category.Name == "Senior" {
if !alreadySubscribed(participant, uint(*seniorContestId)) {
log.Printf("Subscribe %s to contest ID %d", participant, *seniorContestId)
participant.ContestIDs = append(participant.ContestIDs, uint(*seniorContestId))
updated = true
}
}
if updated {
id, err := client.Update(participant)
if err != nil {
panic(err)
}
log.Printf("Successfully updated participant ID %d", id)
updated = false
}
}
}
}
if !updated {

19
scripts/subscribe_participants/regions_contests.yaml Normal file
View file

@ -0,0 +1,19 @@
contests:
- id: 22
category: Junior
regions: [Basilicata,Calabria,Lombardia,Piemonte,Valle d'Aosta,Veneto]
- id: 23
category: Senior
regions: [Basilicata,Calabria,Lombardia,Piemonte,Valle d'Aosta,Veneto]
- id: 26
category: Junior
regions: [Campania,Emilia-Romagna,Friuli-Venezia Giulia,Liguria,Sardegna,Toscana,Trentino-Alto Adige]
- id: 27
category: Senior
regions: [Campania,Emilia-Romagna,Friuli-Venezia Giulia,Liguria,Sardegna,Toscana,Trentino-Alto Adige]
- id: 28
category: Junior
regions: [Abruzzo,Lazio,Marche,Molise,Puglia,Sicilia,Umbria]
- id: 25
category: Senior
regions: [Abruzzo,Lazio,Marche,Molise,Puglia,Sicilia,Umbria]