You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
151 lines
10 KiB
151 lines
10 KiB
package database
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
|
|
"health-ai/internal/model"
|
|
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
// SeedTestUser 创建测试用户
|
|
func SeedTestUser() error {
|
|
var count int64
|
|
DB.Model(&model.User{}).Where("phone = ?", "13800138000").Count(&count)
|
|
if count > 0 {
|
|
log.Println("Test user already exists")
|
|
return nil
|
|
}
|
|
|
|
// 加密密码
|
|
hashedPassword, err := bcrypt.GenerateFromPassword([]byte("123456"), bcrypt.DefaultCost)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
testUser := &model.User{
|
|
Phone: "13800138000",
|
|
PasswordHash: string(hashedPassword),
|
|
Nickname: "测试用户",
|
|
}
|
|
|
|
if err := DB.Create(testUser).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Println("Test user created: phone=13800138000, password=123456")
|
|
return nil
|
|
}
|
|
|
|
// SeedQuestionBank 初始化问卷题库
|
|
func SeedQuestionBank() error {
|
|
// 检查是否已有数据
|
|
var count int64
|
|
DB.Model(&model.QuestionBank{}).Count(&count)
|
|
if count > 0 {
|
|
log.Printf("Question bank already seeded with %d questions", count)
|
|
return nil
|
|
}
|
|
|
|
questions := getQuestions()
|
|
for _, q := range questions {
|
|
if err := DB.Create(&q).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
log.Printf("Question bank seeded with %d questions", len(questions))
|
|
return nil
|
|
}
|
|
|
|
func getQuestions() []model.QuestionBank {
|
|
options, _ := json.Marshal([]string{"没有", "很少", "有时", "经常", "总是"})
|
|
optStr := string(options)
|
|
|
|
return []model.QuestionBank{
|
|
// 平和质 (8题)
|
|
{ConstitutionType: model.ConstitutionPinghe, QuestionText: "您精力充沛吗？", Options: optStr, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionPinghe, QuestionText: "您容易疲乏吗？", Options: optStr, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionPinghe, QuestionText: "您说话声音低弱无力吗？", Options: optStr, OrderNum: 3},
|
|
{ConstitutionType: model.ConstitutionPinghe, QuestionText: "您感到闷闷不乐、情绪低沉吗？", Options: optStr, OrderNum: 4},
|
|
{ConstitutionType: model.ConstitutionPinghe, QuestionText: "您比一般人耐受不了寒冷吗？", Options: optStr, OrderNum: 5},
|
|
{ConstitutionType: model.ConstitutionPinghe, QuestionText: "您能适应外界自然和社会环境的变化吗？", Options: optStr, OrderNum: 6},
|
|
{ConstitutionType: model.ConstitutionPinghe, QuestionText: "您容易失眠吗？", Options: optStr, OrderNum: 7},
|
|
{ConstitutionType: model.ConstitutionPinghe, QuestionText: "您容易忘事吗？", Options: optStr, OrderNum: 8},
|
|
|
|
// 气虚质 (8题)
|
|
{ConstitutionType: model.ConstitutionQixu, QuestionText: "您容易气短（呼吸短促，接不上气）吗？", Options: optStr, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionQixu, QuestionText: "您容易心慌吗？", Options: optStr, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionQixu, QuestionText: "您容易头晕或站起时晕眩吗？", Options: optStr, OrderNum: 3},
|
|
{ConstitutionType: model.ConstitutionQixu, QuestionText: "您比别人容易感冒吗？", Options: optStr, OrderNum: 4},
|
|
{ConstitutionType: model.ConstitutionQixu, QuestionText: "您喜欢安静、懒得说话吗？", Options: optStr, OrderNum: 5},
|
|
{ConstitutionType: model.ConstitutionQixu, QuestionText: "您说话声音低弱无力吗？", Options: optStr, OrderNum: 6},
|
|
{ConstitutionType: model.ConstitutionQixu, QuestionText: "您活动量稍大就容易出虚汗吗？", Options: optStr, OrderNum: 7},
|
|
{ConstitutionType: model.ConstitutionQixu, QuestionText: "您容易疲乏吗？", Options: optStr, OrderNum: 8},
|
|
|
|
// 阳虚质 (7题)
|
|
{ConstitutionType: model.ConstitutionYangxu, QuestionText: "您手脚发凉吗？", Options: optStr, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionYangxu, QuestionText: "您胃脘部、背部或腰膝部怕冷吗？", Options: optStr, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionYangxu, QuestionText: "您穿的衣服总比别人多吗？", Options: optStr, OrderNum: 3},
|
|
{ConstitutionType: model.ConstitutionYangxu, QuestionText: "您比一般人耐受不了寒冷吗？", Options: optStr, OrderNum: 4},
|
|
{ConstitutionType: model.ConstitutionYangxu, QuestionText: "您比别人容易感冒吗？", Options: optStr, OrderNum: 5},
|
|
{ConstitutionType: model.ConstitutionYangxu, QuestionText: "您吃凉东西会感到不舒服或怕吃凉东西吗？", Options: optStr, OrderNum: 6},
|
|
{ConstitutionType: model.ConstitutionYangxu, QuestionText: "您受凉或吃凉的东西后，容易拉肚子吗？", Options: optStr, OrderNum: 7},
|
|
|
|
// 阴虚质 (8题)
|
|
{ConstitutionType: model.ConstitutionYinxu, QuestionText: "您感到手脚心发热吗？", Options: optStr, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionYinxu, QuestionText: "您感觉身体、脸上发热吗？", Options: optStr, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionYinxu, QuestionText: "您皮肤或口唇干吗？", Options: optStr, OrderNum: 3},
|
|
{ConstitutionType: model.ConstitutionYinxu, QuestionText: "您口唇的颜色比一般人红吗？", Options: optStr, OrderNum: 4},
|
|
{ConstitutionType: model.ConstitutionYinxu, QuestionText: "您容易便秘或大便干燥吗？", Options: optStr, OrderNum: 5},
|
|
{ConstitutionType: model.ConstitutionYinxu, QuestionText: "您面部两颧潮红或偏红吗？", Options: optStr, OrderNum: 6},
|
|
{ConstitutionType: model.ConstitutionYinxu, QuestionText: "您感到眼睛干涩吗？", Options: optStr, OrderNum: 7},
|
|
{ConstitutionType: model.ConstitutionYinxu, QuestionText: "您感到口干咽燥、总想喝水吗？", Options: optStr, OrderNum: 8},
|
|
|
|
// 痰湿质 (8题)
|
|
{ConstitutionType: model.ConstitutionTanshi, QuestionText: "您感到胸闷或腹部胀满吗？", Options: optStr, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionTanshi, QuestionText: "您感到身体沉重不轻松或不爽快吗？", Options: optStr, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionTanshi, QuestionText: "您腹部肥满松软吗？", Options: optStr, OrderNum: 3},
|
|
{ConstitutionType: model.ConstitutionTanshi, QuestionText: "您额头部位油脂分泌多吗？", Options: optStr, OrderNum: 4},
|
|
{ConstitutionType: model.ConstitutionTanshi, QuestionText: "您上眼睑比别人肿吗？", Options: optStr, OrderNum: 5},
|
|
{ConstitutionType: model.ConstitutionTanshi, QuestionText: "您嘴里有黏黏的感觉吗？", Options: optStr, OrderNum: 6},
|
|
{ConstitutionType: model.ConstitutionTanshi, QuestionText: "您平时痰多吗？", Options: optStr, OrderNum: 7},
|
|
{ConstitutionType: model.ConstitutionTanshi, QuestionText: "您舌苔厚腻或有舌苔厚厚的感觉吗？", Options: optStr, OrderNum: 8},
|
|
|
|
// 湿热质 (7题)
|
|
{ConstitutionType: model.ConstitutionShire, QuestionText: "您面部或鼻部有油腻感或油光发亮吗？", Options: optStr, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionShire, QuestionText: "您脸上容易生痤疮或皮肤容易生疮疖吗？", Options: optStr, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionShire, QuestionText: "您感到口苦或嘴里有异味吗？", Options: optStr, OrderNum: 3},
|
|
{ConstitutionType: model.ConstitutionShire, QuestionText: "您大便黏滞不爽、有解不尽的感觉吗？", Options: optStr, OrderNum: 4},
|
|
{ConstitutionType: model.ConstitutionShire, QuestionText: "您小便时尿道有发热感、尿色浓吗？", Options: optStr, OrderNum: 5},
|
|
{ConstitutionType: model.ConstitutionShire, QuestionText: "您带下色黄（白带颜色发黄）吗？（限女性回答）", Options: optStr, OrderNum: 6},
|
|
{ConstitutionType: model.ConstitutionShire, QuestionText: "您的阴囊部位潮湿吗？（限男性回答）", Options: optStr, OrderNum: 7},
|
|
|
|
// 血瘀质 (7题)
|
|
{ConstitutionType: model.ConstitutionXueyu, QuestionText: "您皮肤在不知不觉中会出现青紫瘀斑吗？", Options: optStr, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionXueyu, QuestionText: "您两颧部有细微红丝吗？", Options: optStr, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionXueyu, QuestionText: "您身体上有哪里疼痛吗？", Options: optStr, OrderNum: 3},
|
|
{ConstitutionType: model.ConstitutionXueyu, QuestionText: "您面色晦暗或容易出现褐斑吗？", Options: optStr, OrderNum: 4},
|
|
{ConstitutionType: model.ConstitutionXueyu, QuestionText: "您容易有黑眼圈吗？", Options: optStr, OrderNum: 5},
|
|
{ConstitutionType: model.ConstitutionXueyu, QuestionText: "您容易忘事吗？", Options: optStr, OrderNum: 6},
|
|
{ConstitutionType: model.ConstitutionXueyu, QuestionText: "您口唇颜色偏暗吗？", Options: optStr, OrderNum: 7},
|
|
|
|
// 气郁质 (7题)
|
|
{ConstitutionType: model.ConstitutionQiyu, QuestionText: "您感到闷闷不乐、情绪低沉吗？", Options: optStr, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionQiyu, QuestionText: "您精神紧张、焦虑不安吗？", Options: optStr, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionQiyu, QuestionText: "您多愁善感、感情脆弱吗？", Options: optStr, OrderNum: 3},
|
|
{ConstitutionType: model.ConstitutionQiyu, QuestionText: "您容易感到害怕或受到惊吓吗？", Options: optStr, OrderNum: 4},
|
|
{ConstitutionType: model.ConstitutionQiyu, QuestionText: "您胁肋部或乳房胀痛吗？", Options: optStr, OrderNum: 5},
|
|
{ConstitutionType: model.ConstitutionQiyu, QuestionText: "您无缘无故叹气吗？", Options: optStr, OrderNum: 6},
|
|
{ConstitutionType: model.ConstitutionQiyu, QuestionText: "您咽喉部有异物感吗？", Options: optStr, OrderNum: 7},
|
|
|
|
// 特禀质 (7题)
|
|
{ConstitutionType: model.ConstitutionTebing, QuestionText: "您没有感冒时也会打喷嚏吗？", Options: optStr, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionTebing, QuestionText: "您没有感冒时也会鼻塞、流鼻涕吗？", Options: optStr, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionTebing, QuestionText: "您有因季节变化、温度变化或异味引起的咳嗽吗？", Options: optStr, OrderNum: 3},
|
|
{ConstitutionType: model.ConstitutionTebing, QuestionText: "您容易过敏吗？", Options: optStr, OrderNum: 4},
|
|
{ConstitutionType: model.ConstitutionTebing, QuestionText: "您皮肤容易起荨麻疹吗？", Options: optStr, OrderNum: 5},
|
|
{ConstitutionType: model.ConstitutionTebing, QuestionText: "您皮肤一抓就红，并出现抓痕吗？", Options: optStr, OrderNum: 6},
|
|
{ConstitutionType: model.ConstitutionTebing, QuestionText: "您皮肤或身上容易出现紫红色瘀点、瘀斑吗？", Options: optStr, OrderNum: 7},
|
|
}
|
|
}
|
|
|