#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ 批量创建学生信息脚本 为每个班级创建30个学生的测试数据 """ import requests import json import random import time from datetime import datetime, timedelta import logging # 配置日志 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s') logger = logging.getLogger(__name__) class SimpleDataGenerator: """简单的数据生成器，替代faker""" def __init__(self): # 姓氏 self.surnames = ['李', '王', '张', '刘', '陈', '杨', '黄', '赵', '周', '吴', '徐', '孙', '朱', '马', '胡', '郭', '林', '何', '高', '梁'] # 男性名字 self.male_names = ['伟', '强', '磊', '军', '勇', '涛', '明', '超', '辉', '华', '建', '文', '健', '云', '峰', '浩', '鹏', '飞', '宇', '杰'] # 女性名字 self.female_names = ['丽', '娜', '敏', '静', '佳', '雅', '莉', '萍', '红', '梅', '芳', '燕', '玲', '霞', '琳', '晶', '欣', '宁', '蕾', '洁'] # 省份 self.provinces = ['北京市', '上海市', '天津市', '重庆市', '河北省', '山西省', '辽宁省', '吉林省', '黑龙江省', '江苏省', '浙江省', '安徽省', '福建省', '江西省', '山东省', '河南省', '湖北省', '湖南省', '广东省', '海南省', '四川省', '贵州省', '云南省', '陕西省', '甘肃省', '青海省', '台湾省', '内蒙古自治区', '广西壮族自治区', '西藏自治区', '宁夏回族自治区', '新疆维吾尔自治区'] def generate_name(self, gender): """生成姓名""" surname = random.choice(self.surnames) if gender == '0': # 男性 given_name = random.choice(self.male_names) else: # 女性 given_name = random.choice(self.female_names) # 有时候生成双字名 if random.random() < 0.3: if gender == '0': given_name += random.choice(self.male_names) else: given_name += random.choice(self.female_names) return surname + given_name def generate_phone(self): """生成手机号""" prefixes = ['130', '131', '132', '133', '134', '135', '136', '137', '138', '139', '150', '151', '152', '153', '155', '156', '157', '158', '159', '180', '181', '182', '183', '184', '185', '186', '187', '188', '189'] prefix = random.choice(prefixes) suffix = ''.join([str(random.randint(0, 9)) for _ in range(8)]) return prefix + suffix def generate_id_card(self, birth_date): """生成身份证号""" # 地区代码（简化版） area_codes = ['110101', '110102', '110105', '110106', '110107', '110108', '310101', '310104', '310105', '310106', '310107', '310109', '440101', '440103', '440104', '440105', '440106', '440111'] area_code = random.choice(area_codes) # 出生日期（YYYYMMDD格式） birth_str = birth_date.replace('-', '') # 顺序码（3位）+ 校验码（1位） sequence = str(random.randint(100, 999)) check_digit = str(random.randint(0, 9)) return area_code + birth_str + sequence + check_digit def generate_address(self): """生成地址""" province = random.choice(self.provinces) cities = ['市', '县'] city = f"某{random.choice(cities)}" district = f"某区某街道{random.randint(1, 999)}号" return province + city + district def generate_province(self): """生成省份""" return random.choice(self.provinces) class StudentBatchCreator: def __init__(self, base_url="http://localhost:8080", token=None): """ 初始化学生批量创建器 Args: base_url: 后端API基础URL token: 认证token """ self.base_url = base_url self.token = token self.session = requests.Session() self.data_gen = SimpleDataGenerator() # 设置请求头 if token: self.session.headers.update({ 'Authorization': f'Bearer {token}', 'Content-Type': 'application/json' }) def set_token(self, token): """设置认证token""" self.token = token self.session.headers.update({ 'Authorization': f'Bearer {token}' }) def get_class_list(self): """获取班级列表""" try: url = f"{self.base_url}/classMsg/classMsg/list" params = { 'pageNum': 1, 'pageSize': 1000 # 获取所有班级 } response = self.session.get(url, params=params) response.raise_for_status() data = response.json() if data.get('code') == 200: classes = data.get('rows', []) logger.info(f"获取到 {len(classes)} 个班级") return classes else: logger.error(f"获取班级列表失败: {data.get('msg')}") return [] except Exception as e: logger.error(f"获取班级列表异常: {str(e)}") return [] def generate_birth_date(self): """生成出生日期（18-22岁）""" today = datetime.now() min_age = 18 max_age = 22 # 计算出生年份范围 birth_year = today.year - random.randint(min_age, max_age) birth_month = random.randint(1, 12) birth_day = random.randint(1, 28) # 使用28避免月份天数问题 return f"{birth_year:04d}-{birth_month:02d}-{birth_day:02d}" def generate_student_data(self, class_info, student_index): """ 生成单个学生的测试数据 Args: class_info: 班级信息 student_index: 学生编号（1-30） Returns: dict: 学生数据 """ # 生成基础信息 gender = random.choice(['0', '1']) # 0-男，1-女 name = self.data_gen.generate_name(gender) # 生成学号：班级代码 + 学生编号 class_code = class_info.get('classCode', class_info.get('class_code', '')) if not class_code: # 如果没有班级代码，使用班级ID class_code = f"CLS{class_info.get('id', '')}" student_no = f"{class_code}{student_index:02d}" # 生成出生日期 birth_date = self.generate_birth_date() # 生成身份证号 id_card = self.data_gen.generate_id_card(birth_date) # 生成手机号 phone = self.data_gen.generate_phone() # 民族选择 nations = ['01', '02', '03', '04', '05'] # 汉族、蒙古族、回族、藏族、维吾尔族等 nationality = random.choice(nations) # 政治面貌选择 political_statuses = ['01', '02', '03', '13'] # 群众、团员、党员等 political_status = random.choice(political_statuses) # 构建学生数据 student_data = { # 基础信息 "studentNo": student_no, "idCard": id_card, "name": name, "gender": gender, "birthDate": birth_date, "nationality": nationality, "politicalStatus": political_status, "phone": phone, "email": f"{student_no}@student.edu.cn", # 学籍信息（从班级信息继承） "classId": class_info.get('id'), "className": class_info.get('className', class_info.get('class_name', '')), "majorId": class_info.get('majorId', class_info.get('major_id')), "majorName": class_info.get('majorName', class_info.get('major_name', '')), "collegeId": class_info.get('collegeId', class_info.get('college_id')), "collegeName": class_info.get('collegeName', class_info.get('college_name', '')), "grade": class_info.get('grade', ''), "enrollmentYear": class_info.get('enrollmentYear', class_info.get('enrollment_year')), "academicLevel": class_info.get('classType', class_info.get('class_type', 1)), "classTeacherId": class_info.get('classTeacherId', class_info.get('class_teacher_id')), "classTeacherName": class_info.get('classTeacherName', class_info.get('class_teacher_name', '')), "schoolSystem": class_info.get('schoolSystem', class_info.get('school_system', 3)), # 默认状态 "studentStatus": "1", # 在籍 "trainingType": "全日制", "admissionCategory": "统招", "householdRegisterType": "非农业", "educationLevel": "2", # 高中 "maritalStatus": "1", # 未婚 # 地址信息 "nativePlace": self.data_gen.generate_province(), "sourceRegion": self.data_gen.generate_province(), "homeAddress": self.data_gen.generate_address(), "currentAddress": "学校宿舍", # 家庭信息 "fatherName": self.data_gen.generate_name('0'), "fatherPhone": self.data_gen.generate_phone(), "motherName": self.data_gen.generate_name('1'), "motherPhone": self.data_gen.generate_phone(), # 健康信息 "height": random.randint(155, 185), "weight": random.randint(45, 80), "bloodType": random.choice(['A', 'B', 'AB', 'O']), "healthStatus": "良好", # 布尔字段 "isBoarder": 1, # 住校 "isPoorStudent": random.randint(0, 1), "isMinority": 1 if nationality != '01' else 0, # 非汉族为少数民族 "isDisabled": 0, "isOrphan": 0, "isLeftBehindChild": 0, "isSingleChild": random.randint(0, 1), "isLowIncome": 0, "hasLoan": random.randint(0, 1), # 其他字段 "remark": f"批量生成的测试学生数据 - {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}" } return student_data def create_student(self, student_data): """ 创建单个学生 Args: student_data: 学生数据 Returns: tuple: (success, result) """ try: url = f"{self.base_url}/studentMsg/studentMsg" response = self.session.post(url, json=student_data) response.raise_for_status() result = response.json() if result.get('code') == 200: logger.info(f"成功创建学生: {student_data['name']} ({student_data['studentNo']})") return True, result else: logger.error(f"创建学生失败: {student_data['name']} - {result.get('msg')}") return False, result except Exception as e: logger.error(f"创建学生异常: {student_data['name']} - {str(e)}") return False, str(e) def batch_create_students_for_class(self, class_info, student_count=30): """ 为指定班级批量创建学生 Args: class_info: 班级信息 student_count: 要创建的学生数量 Returns: dict: 创建结果统计 """ class_name = class_info.get('className', class_info.get('class_name', f"班级{class_info.get('id')}")) logger.info(f"开始为班级 [{class_name}] 创建 {student_count} 个学生...") success_count = 0 failed_count = 0 failed_students = [] for i in range(1, student_count + 1): try: # 生成学生数据 student_data = self.generate_student_data(class_info, i) # 创建学生 success, result = self.create_student(student_data) if success: success_count += 1 else: failed_count += 1 failed_students.append({ 'student_no': student_data['studentNo'], 'name': student_data['name'], 'error': result }) # 添加短暂延迟，避免请求过于频繁 time.sleep(0.1) except Exception as e: failed_count += 1 logger.error(f"处理第 {i} 个学生时发生异常: {str(e)}") failed_students.append({ 'student_no': f"未知_{i}", 'name': '未知', 'error': str(e) }) result = { 'class_name': class_name, 'total': student_count, 'success': success_count, 'failed': failed_count, 'failed_students': failed_students } logger.info(f"班级 [{class_name}] 学生创建完成: 成功 {success_count}，失败 {failed_count}") return result def batch_create_students_for_all_classes(self, student_count_per_class=30): """ 为所有班级批量创建学生 Args: student_count_per_class: 每个班级创建的学生数量 Returns: list: 所有班级的创建结果 """ # 获取班级列表 classes = self.get_class_list() if not classes: logger.error("没有获取到班级信息，无法创建学生") return [] logger.info(f"开始为 {len(classes)} 个班级批量创建学生，每个班级 {student_count_per_class} 个学生") all_results = [] total_success = 0 total_failed = 0 for class_info in classes: try: result = self.batch_create_students_for_class(class_info, student_count_per_class) all_results.append(result) total_success += result['success'] total_failed += result['failed'] # 班级间添加延迟 time.sleep(0.5) except Exception as e: logger.error(f"处理班级 {class_info} 时发生异常: {str(e)}") all_results.append({ 'class_name': f"班级{class_info.get('id', '未知')}", 'total': student_count_per_class, 'success': 0, 'failed': student_count_per_class, 'error': str(e) }) total_failed += student_count_per_class # 输出总结 logger.info("="*50) logger.info("批量创建学生完成！") logger.info(f"总班级数: {len(classes)}") logger.info(f"总学生数: {len(classes) * student_count_per_class}") logger.info(f"成功创建: {total_success}") logger.info(f"创建失败: {total_failed}") logger.info("="*50) return all_results def main(): """主函数""" print("学生批量创建工具") print("="*50) # 配置参数 base_url = input("请输入后端API地址 (默认: http://localhost:8080): ").strip() if not base_url: base_url = "http://localhost:8080" token = input("请输入认证Token (可为空): ").strip() student_count = input("请输入每个班级创建的学生数量 (默认: 30): ").strip() if not student_count: student_count = 30 else: try: student_count = int(student_count) except ValueError: print("输入的学生数量无效，使用默认值 30") student_count = 30 # 创建批量创建器 creator = StudentBatchCreator(base_url, token) # 确认操作 print(f"\n配置信息:") print(f"API地址: {base_url}") print(f"每个班级学生数: {student_count}") confirm = input("\n确认开始批量创建学生？(y/N): ").strip().lower() if confirm not in ['y', 'yes']: print("操作已取消") return # 开始批量创建 start_time = time.time() results = creator.batch_create_students_for_all_classes(student_count) end_time = time.time() # 保存结果到文件 timestamp = datetime.now().strftime('%Y%m%d_%H%M%S') result_file = f"student_creation_result_{timestamp}.json" with open(result_file, 'w', encoding='utf-8') as f: json.dump(results, f, ensure_ascii=False, indent=2) print(f"\n创建结果已保存到: {result_file}") print(f"总耗时: {end_time - start_time:.2f} 秒") if __name__ == "__main__": main()