cristiancmoises
/
gophish
mirror of https://github.com/gophish/gophish
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Can now add files to a previously created template that didn't have files to begin with 

Fixes #8
pull/24/head
unknown 2015-02-13 20:17:25 -06:00
parent 5d1774996f
commit cb081f4a17
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
models/template.go
View File

@ -46,13 +46,13 @@ func GetTemplates(uid int64) ([]Template, error) {
} }
for i, _ := range ts { for i, _ := range ts {
err = db.Where("template_id=?", ts[i].Id).Find(&ts[i].Attachments).Error err = db.Where("template_id=?", ts[i].Id).Find(&ts[i].Attachments).Error
if err == nil && len(ts[i].Attachments) == 0 {
ts[i].Attachments = make([]Attachment, 0)
}
if err != nil && err != gorm.RecordNotFound { if err != nil && err != gorm.RecordNotFound {
Logger.Println(err) Logger.Println(err)
return ts, err return ts, err
} }
if err == gorm.RecordNotFound {
err = nil
}
} }
return ts, err return ts, err
} }
@ -70,8 +70,8 @@ func GetTemplate(id int64, uid int64) (Template, error) {
Logger.Println(err) Logger.Println(err)
return t, err return t, err
} }
if err == gorm.RecordNotFound { if err == nil && len(t.Attachments) == 0 {
err = nil t.Attachments = make([]Attachment, 0)
} }
return t, err return t, err
} }