Add add_test_data

add_test_data

@@ -0,0 +1,39 @@
+import sqlite3
+
+def add_test_students():
+    """添加测试学生数据"""
+    conn = sqlite3.connect('database.db')
+    cursor = conn.cursor()
+    
+    # 测试学生数据
+    test_students = [
+        ('2024001', '谢坷菡', '计算机科学'),
+        ('2024002', '苏诗颖', '软件工程'),
+        ('2024003', '陈梓旸', '人工智能'),
+        ('2024004', '姚思源', '数据科学'),
+        ('2024005', '颜妍妍', '网络安全'),
+        ('2024006', '尹常在', '物联网'),
+        ('2024007', '陈亦旸', '大数据'),
+        ('2024008', '陈紧新', '云计算')
+    ]
+    
+    # 清空现有数据（可选）
+    cursor.execute('DELETE FROM students')
+    cursor.execute('DELETE FROM call_records')
+    
+    # 插入测试数据
+    for student_id, name, major in test_students:
+        cursor.execute('''
+            INSERT INTO students (student_id, name, major) 
+            VALUES (?, ?, ?)
+        ''', (student_id, name, major))
+    
+    conn.commit()
+    conn.close()
+    print("测试学生数据添加成功！")
+    print("添加了8个测试学生：")
+    for i, (student_id, name, major) in enumerate(test_students, 1):
+        print(f"{i}. {name} (学号: {student_id}, 专业: {major})")
+
+if __name__ == '__main__':
+    add_test_students()