261 lines
13 KiB
Python
261 lines
13 KiB
Python
"""
|
||
种子数据初始化模块
|
||
在 main.py 启动时自动调用，向数据库注入测试数据
|
||
幂等设计：已存在数据则跳过，不重复插入
|
||
"""
|
||
from datetime import date
|
||
from sqlalchemy import text
|
||
from databases import engine_all, Base_all, Session
|
||
|
||
|
||
def seed_test_data():
|
||
"""启动时自动注入测试数据（幂等）"""
|
||
db = Session()
|
||
try:
|
||
# 临时关闭外键检查，避免 class_management_model.py 的 FK bug
|
||
db.execute(text("SET FOREIGN_KEY_CHECKS = 0"))
|
||
|
||
# 导入所有模型以获取 metadata（触发 Base_all.metadata.create_all）
|
||
import students.model.students_model
|
||
import Teachers.model.tea_model
|
||
import scores.model.score_model
|
||
import stu_jiuye.model
|
||
|
||
# 创建所有表（临时关闭外键检查）
|
||
db.execute(text("SET FOREIGN_KEY_CHECKS = 0"))
|
||
Base_all.metadata.create_all(bind=engine_all)
|
||
|
||
# 手动建 classes 表（因 FK bug 无法通过 ORM 建表）
|
||
db.execute(text("""
|
||
CREATE TABLE IF NOT EXISTS classes (
|
||
id INT AUTO_INCREMENT PRIMARY KEY,
|
||
num VARCHAR(50) NOT NULL UNIQUE,
|
||
name VARCHAR(50) NOT NULL,
|
||
head_teacher_id INT,
|
||
coach_teacher_id INT,
|
||
tutor_teacher_id INT,
|
||
class_start_time DATE,
|
||
class_end_time DATE,
|
||
is_delete INT DEFAULT 0,
|
||
create_time DATETIME,
|
||
update_time DATETIME,
|
||
FOREIGN KEY (head_teacher_id) REFERENCES teachers(id),
|
||
FOREIGN KEY (coach_teacher_id) REFERENCES teachers(id),
|
||
FOREIGN KEY (tutor_teacher_id) REFERENCES teachers(id)
|
||
)
|
||
"""))
|
||
|
||
# 恢复外键检查
|
||
db.execute(text("SET FOREIGN_KEY_CHECKS = 1"))
|
||
db.commit()
|
||
|
||
# 幂等检查：科目表有数据就跳过全部
|
||
if db.execute(text("SELECT COUNT(*) FROM subject")).scalar() > 0:
|
||
return
|
||
|
||
print(" [种子] 开始注入测试数据...")
|
||
|
||
# ── 1. 科目 ──────────────────────────────────────
|
||
subjects = ["Python程序设计", "数据结构", "数据库原理",
|
||
"人工智能导论", "计算机网络", "操作系统"]
|
||
sub_ids = {}
|
||
for name in subjects:
|
||
result = db.execute(
|
||
text("INSERT INTO subject (name) VALUES (:name)"),
|
||
{"name": name}
|
||
)
|
||
sub_ids[name] = result.lastrowid
|
||
|
||
# ── 2. 教师 ──────────────────────────────────────
|
||
teachers = [
|
||
("张伟", "m", 1, sub_ids["Python程序设计"],
|
||
"北京", "Python", "博士", "清华", date(2018, 3, 1), "6年"),
|
||
("李芳", "f", 1, sub_ids["数据结构"],
|
||
"上海", "算法", "博士", "北大", date(2019, 9, 1), "4年"),
|
||
("王强", "m", 2, sub_ids["数据库原理"],
|
||
"广州", "MySQL", "硕士", "中山大学", date(2020, 2, 1), "3年"),
|
||
("陈静", "f", 2, sub_ids["人工智能导论"],
|
||
"深圳", "机器学习", "博士", "华南理工", date(2017, 7, 1), "7年"),
|
||
("刘洋", "m", 3, sub_ids["计算机网络"],
|
||
"杭州", "网络", "硕士", "浙江大学", date(2021, 1, 1), "2年"),
|
||
("赵敏", "f", 3, sub_ids["操作系统"],
|
||
"南京", "Linux", "博士", "南京大学", date(2019, 3, 1), "5年"),
|
||
]
|
||
teacher_ids = []
|
||
for t in teachers:
|
||
phone = f"1380000{len(teacher_ids):04d}"
|
||
result = db.execute(text("""
|
||
INSERT INTO teachers (name, sex, class_id, phone, subject_id,
|
||
home_place, specialty, education, collage,
|
||
start_time, work_experience, is_delete)
|
||
VALUES (:name, :sex, :class_id, :phone, :subject_id,
|
||
:home_place, :specialty, :education, :collage,
|
||
:start_time, :work_experience, 0)
|
||
"""), {
|
||
"name": t[0], "sex": t[1], "class_id": t[2],
|
||
"phone": phone, "subject_id": t[3],
|
||
"home_place": t[4], "specialty": t[5],
|
||
"education": t[6], "collage": t[7],
|
||
"start_time": t[8], "work_experience": t[9],
|
||
})
|
||
teacher_ids.append(result.lastrowid)
|
||
|
||
# ── 3. 班级 ──────────────────────────────────────
|
||
classes = [
|
||
("CS202401", "计算机科学2401班", teacher_ids[0],
|
||
teacher_ids[1], teacher_ids[2],
|
||
date(2024, 9, 1), date(2028, 6, 30)),
|
||
("CS202402", "计算机科学2402班", teacher_ids[3],
|
||
teacher_ids[4], teacher_ids[5],
|
||
date(2024, 9, 1), date(2028, 6, 30)),
|
||
("AI202501", "人工智能2501班", teacher_ids[0],
|
||
teacher_ids[3], teacher_ids[5],
|
||
date(2025, 9, 1), date(2029, 6, 30)),
|
||
]
|
||
class_ids = []
|
||
for c in classes:
|
||
result = db.execute(text("""
|
||
INSERT INTO classes (num, name, head_teacher_id, coach_teacher_id,
|
||
tutor_teacher_id, class_start_time,
|
||
class_end_time, is_delete)
|
||
VALUES (:num, :name, :head_teacher_id, :coach_teacher_id,
|
||
:tutor_teacher_id, :class_start_time,
|
||
:class_end_time, 0)
|
||
"""), {
|
||
"num": c[0], "name": c[1],
|
||
"head_teacher_id": c[2], "coach_teacher_id": c[3],
|
||
"tutor_teacher_id": c[4],
|
||
"class_start_time": c[5], "class_end_time": c[6],
|
||
})
|
||
class_ids.append(result.lastrowid)
|
||
|
||
# ── 4. 学生 ──────────────────────────────────────
|
||
first_names = ["志强", "丽华", "建国", "美玲", "文博", "晓燕",
|
||
"浩然", "雨桐", "子轩", "思涵",
|
||
"俊杰", "雅琪", "宇航", "欣怡", "天翊",
|
||
"诗涵", "一诺", "浩宇", "梓萱", "博文"]
|
||
home_places = ["北京", "上海", "广州", "深圳", "杭州", "成都", "武汉", "西安"]
|
||
colleges = ["清华大学", "北京大学", "复旦大学", "上海交通大学",
|
||
"浙江大学", "南京大学", "武汉大学", "华中科技大学"]
|
||
specialties = ["计算机科学", "软件工程", "数据科学", "人工智能"]
|
||
|
||
for i in range(20):
|
||
class_idx = i % 3
|
||
enrollment = date(2024, 9, 1) if class_idx != 2 else date(2025, 9, 1)
|
||
graduate = date(2028, 6, 30) if class_idx != 2 else date(2029, 6, 30)
|
||
phone = f"1390000{1000 + i}"
|
||
# 前3个学生年龄设大一些，用于30岁以上统计测试
|
||
age = 30 + i if i < 3 else 18 + (i % 4)
|
||
db.execute(text("""
|
||
INSERT INTO students (num, name, age, sex, home_Place, college,
|
||
specialty, enrollment_time, graduate_time,
|
||
education, class_id, advisor_id, phone, is_deleted)
|
||
VALUES (:num, :name, :age, :sex, :home_Place, :college,
|
||
:specialty, :enrollment_time, :graduate_time,
|
||
'ben', :class_id, :advisor_id, :phone, 0)
|
||
"""), {
|
||
"num": f"2024{1001 + i}",
|
||
"name": first_names[i],
|
||
"age": age,
|
||
"sex": "m" if i % 2 == 0 else "f",
|
||
"home_Place": home_places[i % 8],
|
||
"college": colleges[i % 8],
|
||
"specialty": specialties[i % 4],
|
||
"enrollment_time": enrollment,
|
||
"graduate_time": graduate,
|
||
"class_id": class_ids[class_idx],
|
||
"advisor_id": teacher_ids[i % 6],
|
||
"phone": phone,
|
||
})
|
||
|
||
# ── 5. 成绩 ──────────────────────────────────────
|
||
all_students = db.execute(
|
||
text("SELECT id, class_id FROM students WHERE is_deleted = 0")
|
||
).fetchall()
|
||
all_subjects = db.execute(
|
||
text("SELECT id FROM subject")
|
||
).fetchall()
|
||
|
||
for idx, st in enumerate(all_students):
|
||
for subj_idx, sub in enumerate(all_subjects):
|
||
# 前3名学生设大年龄（用于30岁以上统计）
|
||
# 前2名学生所有科目80分以上（用于优秀成绩统计）
|
||
# 最后2名学生部分科目不及格（用于不及格统计）
|
||
if idx < 2:
|
||
# 优秀学生：所有科目85-95分
|
||
score_val = 85 + (idx * 5 + subj_idx * 3) % 11
|
||
elif idx >= 18:
|
||
# 不及格学生：部分科目50-60分
|
||
if subj_idx < 3:
|
||
score_val = 50 + (idx * 3 + subj_idx) % 11
|
||
else:
|
||
score_val = 75 + (idx * 2 + subj_idx) % 16
|
||
else:
|
||
score_val = 60 + (idx * 7 + subj_idx * 13) % 41
|
||
db.execute(text("""
|
||
INSERT INTO scores (sid, cid, num, score, t_subject, is_deleted)
|
||
VALUES (:sid, :cid, 1, :score, :t_subject, 0)
|
||
"""), {
|
||
"sid": st[0], "cid": st[1],
|
||
"score": score_val, "t_subject": sub[0],
|
||
})
|
||
|
||
# ── 6. 地址 ──────────────────────────────────────
|
||
addresses = ["北京市海淀区", "上海市浦东新区", "深圳市南山区",
|
||
"杭州市西湖区", "广州市天河区"]
|
||
address_ids = []
|
||
for addr in addresses:
|
||
result = db.execute(
|
||
text("INSERT INTO Address (somewhere) VALUES (:where)"),
|
||
{"where": addr}
|
||
)
|
||
address_ids.append(result.lastrowid)
|
||
|
||
# ── 7. 公司 ──────────────────────────────────────
|
||
companies = ["字节跳动", "阿里巴巴", "腾讯", "华为", "百度"]
|
||
company_ids = []
|
||
for idx, comp_name in enumerate(companies):
|
||
result = db.execute(
|
||
text("INSERT INTO Company (employment_company) VALUES (:name)"),
|
||
{"name": comp_name}
|
||
)
|
||
company_ids.append(result.lastrowid)
|
||
|
||
# ── 8. 就业信息 ───────────────────────────────────
|
||
# 添加更多就业记录，确保统计接口有数据
|
||
employment_data = []
|
||
for i in range(min(15, len(all_students))):
|
||
company_idx = i % len(company_ids)
|
||
address_idx = i % len(address_ids)
|
||
salary = 25000 + (i * 2000) % 15000
|
||
# 前10个学生有offer时间，后5个没有（测试不同情况）
|
||
offer_time = date(2028, 4, 15 + i) if i < 10 else None
|
||
employment_data.append((
|
||
all_students[i][0], all_students[i][1], company_ids[company_idx],
|
||
address_ids[address_idx], salary, date(2028, 3, 1), offer_time
|
||
))
|
||
for idx, emp in enumerate(employment_data):
|
||
db.execute(text("""
|
||
INSERT INTO Employments (stuid, class_id, company_id, address_id,
|
||
employment_salary, employment_open_time,
|
||
offer_recived_time, is_deleted)
|
||
VALUES (:stuid, :class_id, :company_id, :addr_id, :salary,
|
||
:open_time, :offer_time, 0)
|
||
"""), {
|
||
"stuid": emp[0], "class_id": emp[1],
|
||
"company_id": emp[2], "addr_id": emp[3],
|
||
"salary": emp[4], "open_time": emp[5], "offer_time": emp[6],
|
||
})
|
||
|
||
db.commit()
|
||
print(f" [种子] 完成！插入 {len(subjects)} 科目、{len(teachers)} 教师、"
|
||
f"{len(classes)} 班级、20 学生、{len(all_students)*len(all_subjects)} 成绩、"
|
||
f"{len(companies)} 公司、{len(employment_data)} 就业")
|
||
|
||
except Exception as e:
|
||
db.rollback()
|
||
print(f" [种子] 失败: {e}")
|
||
raise
|
||
finally:
|
||
db.close()
|