镜像自地址
https://github.com/a13zhy/a13rollcall.git
已同步 2026-10-21 03:41:51 +08:00
433 行
16 KiB
Python
433 行
16 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
A13课堂点名系统 - 分组管理模块
|
||
支持创建、编辑、删除学生分组，按分组抽取，随机分组等功能
|
||
"""
|
||
|
||
import json
|
||
import os
|
||
import random
|
||
from typing import List, Dict, Optional, Tuple
|
||
|
||
try:
|
||
from PyQt5.QtWidgets import QDialog
|
||
except ImportError:
|
||
QDialog = object
|
||
|
||
|
||
class GroupManager:
|
||
"""
|
||
分组管理器
|
||
管理学生分组，支持持久化存储
|
||
"""
|
||
|
||
def __init__(self, config_manager=None, base_dir=None):
|
||
self.config_manager = config_manager
|
||
self.base_dir = base_dir or os.path.dirname(os.path.abspath(__file__))
|
||
self.groups_file = os.path.join(self.base_dir, "groups.json")
|
||
self.groups: Dict[str, List[str]] = {}
|
||
self.current_group: Optional[str] = None
|
||
self._load_groups()
|
||
|
||
def _load_groups(self):
|
||
"""从文件加载分组"""
|
||
try:
|
||
if os.path.exists(self.groups_file):
|
||
with open(self.groups_file, "r", encoding="utf-8") as f:
|
||
data = json.load(f)
|
||
self.groups = data.get("groups", {})
|
||
self.current_group = data.get("current_group")
|
||
except Exception as e:
|
||
print(f"加载分组失败: {e}")
|
||
self.groups = {}
|
||
|
||
def _save_groups(self):
|
||
"""保存分组到文件"""
|
||
try:
|
||
data = {
|
||
"groups": self.groups,
|
||
"current_group": self.current_group
|
||
}
|
||
with open(self.groups_file, "w", encoding="utf-8") as f:
|
||
json.dump(data, f, ensure_ascii=False, indent=2)
|
||
except Exception as e:
|
||
print(f"保存分组失败: {e}")
|
||
|
||
def create_group(self, group_name: str, students: List[str] = None) -> bool:
|
||
"""
|
||
创建新分组
|
||
返回是否成功
|
||
"""
|
||
if not group_name or not group_name.strip():
|
||
return False
|
||
group_name = group_name.strip()
|
||
if group_name in self.groups:
|
||
return False
|
||
self.groups[group_name] = students or []
|
||
self._save_groups()
|
||
return True
|
||
|
||
def delete_group(self, group_name: str) -> bool:
|
||
"""删除分组"""
|
||
if group_name in self.groups:
|
||
del self.groups[group_name]
|
||
if self.current_group == group_name:
|
||
self.current_group = None
|
||
self._save_groups()
|
||
return True
|
||
return False
|
||
|
||
def rename_group(self, old_name: str, new_name: str) -> bool:
|
||
"""重命名分组"""
|
||
if old_name not in self.groups or not new_name.strip():
|
||
return False
|
||
new_name = new_name.strip()
|
||
if new_name in self.groups:
|
||
return False
|
||
self.groups[new_name] = self.groups.pop(old_name)
|
||
if self.current_group == old_name:
|
||
self.current_group = new_name
|
||
self._save_groups()
|
||
return True
|
||
|
||
def add_student_to_group(self, group_name: str, student: str) -> bool:
|
||
"""添加学生到分组"""
|
||
if group_name not in self.groups:
|
||
return False
|
||
if student not in self.groups[group_name]:
|
||
self.groups[group_name].append(student)
|
||
self._save_groups()
|
||
return True
|
||
|
||
def remove_student_from_group(self, group_name: str, student: str) -> bool:
|
||
"""从分组移除学生"""
|
||
if group_name not in self.groups:
|
||
return False
|
||
if student in self.groups[group_name]:
|
||
self.groups[group_name].remove(student)
|
||
self._save_groups()
|
||
return True
|
||
|
||
def get_group_students(self, group_name: str) -> List[str]:
|
||
"""获取分组中的学生列表"""
|
||
return self.groups.get(group_name, [])
|
||
|
||
def get_all_groups(self) -> List[str]:
|
||
"""获取所有分组名称"""
|
||
return list(self.groups.keys())
|
||
|
||
def set_current_group(self, group_name: Optional[str]):
|
||
"""设置当前选中的分组"""
|
||
if group_name is None or group_name in self.groups:
|
||
self.current_group = group_name
|
||
self._save_groups()
|
||
|
||
def get_current_group(self) -> Optional[str]:
|
||
"""获取当前选中的分组"""
|
||
return self.current_group
|
||
|
||
def get_current_students(self, all_students: List[str]) -> List[str]:
|
||
"""
|
||
获取当前应该抽取的学生列表
|
||
如果选中了分组，返回分组中的学生（且在总名单中）
|
||
否则返回所有学生
|
||
"""
|
||
if self.current_group and self.current_group in self.groups:
|
||
group_students = self.groups[self.current_group]
|
||
return [s for s in group_students if s in all_students]
|
||
return all_students
|
||
|
||
def random_grouping(self, students: List[str], group_count: int) -> Dict[str, List[str]]:
|
||
"""
|
||
随机分组
|
||
将学生随机分成指定数量的组
|
||
"""
|
||
if group_count <= 0 or not students:
|
||
return {}
|
||
|
||
shuffled = students.copy()
|
||
random.shuffle(shuffled)
|
||
|
||
groups = {}
|
||
base_size = len(shuffled) // group_count
|
||
remainder = len(shuffled) % group_count
|
||
|
||
idx = 0
|
||
for i in range(group_count):
|
||
size = base_size + (1 if i < remainder else 0)
|
||
group_name = f"第{i+1}组"
|
||
groups[group_name] = shuffled[idx:idx+size]
|
||
idx += size
|
||
|
||
return groups
|
||
|
||
def save_random_groups(self, groups: Dict[str, List[str]]) -> bool:
|
||
"""保存随机生成的分组"""
|
||
try:
|
||
for name, students in groups.items():
|
||
self.groups[name] = students
|
||
self._save_groups()
|
||
return True
|
||
except Exception:
|
||
return False
|
||
|
||
def clear_all_groups(self):
|
||
"""清空所有分组"""
|
||
self.groups = {}
|
||
self.current_group = None
|
||
self._save_groups()
|
||
|
||
def get_group_count(self) -> int:
|
||
"""获取分组数量"""
|
||
return len(self.groups)
|
||
|
||
def get_student_groups(self, student: str) -> List[str]:
|
||
"""获取学生所在的所有分组"""
|
||
return [name for name, students in self.groups.items() if student in students]
|
||
|
||
def import_groups_from_file(self, file_path: str) -> bool:
|
||
"""从文件导入分组"""
|
||
try:
|
||
if not os.path.exists(file_path):
|
||
return False
|
||
with open(file_path, "r", encoding="utf-8") as f:
|
||
data = json.load(f)
|
||
if "groups" in data:
|
||
self.groups.update(data["groups"])
|
||
self._save_groups()
|
||
return True
|
||
except Exception as e:
|
||
print(f"导入分组失败: {e}")
|
||
return False
|
||
|
||
def export_groups_to_file(self, file_path: str) -> bool:
|
||
"""导出分组到文件"""
|
||
try:
|
||
data = {
|
||
"groups": self.groups,
|
||
"export_time": __import__("datetime").datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||
}
|
||
with open(file_path, "w", encoding="utf-8") as f:
|
||
json.dump(data, f, ensure_ascii=False, indent=2)
|
||
return True
|
||
except Exception as e:
|
||
print(f"导出分组失败: {e}")
|
||
return False
|
||
|
||
|
||
class GroupDialog(QDialog):
|
||
"""
|
||
分组管理对话框
|
||
可视化管理学生分组
|
||
"""
|
||
|
||
def __init__(self, group_manager: GroupManager, all_students: List[str], parent=None):
|
||
super().__init__(parent)
|
||
self.group_manager = group_manager
|
||
self.all_students = all_students
|
||
self.setWindowTitle("分组管理")
|
||
self.setMinimumSize(700, 500)
|
||
self.resize(750, 550)
|
||
self._setup_ui()
|
||
self._refresh_group_list()
|
||
|
||
def _setup_ui(self):
|
||
from PyQt5.QtWidgets import (
|
||
QVBoxLayout, QHBoxLayout, QListWidget, QListWidgetItem,
|
||
QPushButton, QLabel, QLineEdit, QInputDialog, QMessageBox,
|
||
QFrame, QSpinBox
|
||
)
|
||
|
||
layout = QHBoxLayout(self)
|
||
layout.setContentsMargins(20, 20, 20, 20)
|
||
layout.setSpacing(16)
|
||
|
||
left = QFrame()
|
||
left_layout = QVBoxLayout(left)
|
||
left_layout.setContentsMargins(0, 0, 0, 0)
|
||
left_layout.setSpacing(8)
|
||
|
||
left_header = QHBoxLayout()
|
||
title = QLabel("分组列表")
|
||
title.setStyleSheet("font-size: 14px; font-weight: 700; color: #0067c0;")
|
||
left_header.addWidget(title)
|
||
left_header.addStretch()
|
||
left_layout.addLayout(left_header)
|
||
|
||
self.group_list = QListWidget()
|
||
self.group_list.itemClicked.connect(self._on_group_selected)
|
||
left_layout.addWidget(self.group_list, 1)
|
||
|
||
btn_layout = QVBoxLayout()
|
||
btn_layout.setSpacing(6)
|
||
add_btn = QPushButton("➕ 新建分组")
|
||
add_btn.clicked.connect(self._create_group)
|
||
btn_layout.addWidget(add_btn)
|
||
rename_btn = QPushButton("✏️ 重命名")
|
||
rename_btn.clicked.connect(self._rename_group)
|
||
btn_layout.addWidget(rename_btn)
|
||
delete_btn = QPushButton("🗑️ 删除分组")
|
||
delete_btn.clicked.connect(self._delete_group)
|
||
btn_layout.addWidget(delete_btn)
|
||
random_btn = QPushButton("🎲 随机分组")
|
||
random_btn.clicked.connect(self._random_grouping)
|
||
btn_layout.addWidget(random_btn)
|
||
left_layout.addLayout(btn_layout)
|
||
|
||
layout.addWidget(left, 1)
|
||
|
||
right = QFrame()
|
||
right_layout = QVBoxLayout(right)
|
||
right_layout.setContentsMargins(0, 0, 0, 0)
|
||
right_layout.setSpacing(8)
|
||
|
||
right_header = QHBoxLayout()
|
||
self.group_title = QLabel("选择一个分组查看学生")
|
||
self.group_title.setStyleSheet("font-size: 14px; font-weight: 700; color: #0067c0;")
|
||
right_header.addWidget(self.group_title)
|
||
right_header.addStretch()
|
||
right_layout.addLayout(right_header)
|
||
|
||
self.student_list = QListWidget()
|
||
right_layout.addWidget(self.student_list, 1)
|
||
|
||
student_btn_layout = QHBoxLayout()
|
||
add_student_btn = QPushButton("添加学生")
|
||
add_student_btn.clicked.connect(self._add_student)
|
||
student_btn_layout.addWidget(add_student_btn)
|
||
remove_student_btn = QPushButton("移除学生")
|
||
remove_student_btn.clicked.connect(self._remove_student)
|
||
student_btn_layout.addWidget(remove_student_btn)
|
||
right_layout.addLayout(student_btn_layout)
|
||
|
||
layout.addWidget(right, 1)
|
||
|
||
bottom_btn_layout = QHBoxLayout()
|
||
bottom_btn_layout.addStretch()
|
||
ok_btn = QPushButton("确定")
|
||
ok_btn.setObjectName("primary")
|
||
ok_btn.clicked.connect(self.accept)
|
||
bottom_btn_layout.addWidget(ok_btn)
|
||
cancel_btn = QPushButton("取消")
|
||
cancel_btn.clicked.connect(self.reject)
|
||
bottom_btn_layout.addWidget(cancel_btn)
|
||
|
||
main_layout = QVBoxLayout()
|
||
main_layout.addLayout(layout, 1)
|
||
main_layout.addLayout(bottom_btn_layout)
|
||
self.setLayout(main_layout)
|
||
|
||
def _refresh_group_list(self):
|
||
self.group_list.clear()
|
||
all_item = QListWidgetItem("📋 全部学生")
|
||
all_item.setData(0, "__all__")
|
||
self.group_list.addItem(all_item)
|
||
for name in self.group_manager.get_all_groups():
|
||
count = len(self.group_manager.get_group_students(name))
|
||
item = QListWidgetItem(f"📁 {name} ({count}人)")
|
||
item.setData(0, name)
|
||
self.group_list.addItem(item)
|
||
|
||
def _on_group_selected(self, item):
|
||
group_name = item.data(0)
|
||
self.student_list.clear()
|
||
if group_name == "__all__":
|
||
self.group_title.setText("全部学生")
|
||
for student in self.all_students:
|
||
self.student_list.addItem(QListWidgetItem(student))
|
||
else:
|
||
self.group_title.setText(f"分组：{group_name}")
|
||
students = self.group_manager.get_group_students(group_name)
|
||
for student in students:
|
||
self.student_list.addItem(QListWidgetItem(student))
|
||
|
||
def _create_group(self):
|
||
from PyQt5.QtWidgets import QInputDialog, QMessageBox
|
||
name, ok = QInputDialog.getText(self, "新建分组", "请输入分组名称：")
|
||
if ok and name.strip():
|
||
if self.group_manager.create_group(name.strip()):
|
||
self._refresh_group_list()
|
||
else:
|
||
QMessageBox.warning(self, "创建失败", "分组名称已存在或无效！")
|
||
|
||
def _rename_group(self):
|
||
from PyQt5.QtWidgets import QInputDialog, QMessageBox
|
||
item = self.group_list.currentItem()
|
||
if not item or item.data(0) == "__all__":
|
||
QMessageBox.information(self, "提示", "请先选择一个分组！")
|
||
return
|
||
old_name = item.data(0)
|
||
new_name, ok = QInputDialog.getText(self, "重命名分组", "请输入新名称：", text=old_name)
|
||
if ok and new_name.strip():
|
||
if self.group_manager.rename_group(old_name, new_name.strip()):
|
||
self._refresh_group_list()
|
||
else:
|
||
QMessageBox.warning(self, "重命名失败", "新名称已存在或无效！")
|
||
|
||
def _delete_group(self):
|
||
from PyQt5.QtWidgets import QMessageBox
|
||
item = self.group_list.currentItem()
|
||
if not item or item.data(0) == "__all__":
|
||
QMessageBox.information(self, "提示", "请先选择一个分组！")
|
||
return
|
||
group_name = item.data(0)
|
||
reply = QMessageBox.question(self, "确认删除", f"确定要删除分组「{group_name}」吗？",
|
||
QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
|
||
if reply == QMessageBox.Yes:
|
||
self.group_manager.delete_group(group_name)
|
||
self._refresh_group_list()
|
||
self.student_list.clear()
|
||
self.group_title.setText("选择一个分组查看学生")
|
||
|
||
def _random_grouping(self):
|
||
from PyQt5.QtWidgets import QInputDialog, QMessageBox
|
||
if not self.all_students:
|
||
QMessageBox.information(self, "提示", "没有学生可以分组！")
|
||
return
|
||
count, ok = QInputDialog.getInt(self, "随机分组", "请输入分组数量：", 2, 2, 20, 1)
|
||
if ok:
|
||
groups = self.group_manager.random_grouping(self.all_students, count)
|
||
reply = QMessageBox.question(self, "确认保存",
|
||
f"已随机分成 {len(groups)} 组，是否保存这些分组？\n（将覆盖同名分组）",
|
||
QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
|
||
if reply == QMessageBox.Yes:
|
||
self.group_manager.save_random_groups(groups)
|
||
self._refresh_group_list()
|
||
|
||
def _add_student(self):
|
||
from PyQt5.QtWidgets import QInputDialog, QMessageBox
|
||
item = self.group_list.currentItem()
|
||
if not item or item.data(0) == "__all__":
|
||
QMessageBox.information(self, "提示", "请先选择一个分组！")
|
||
return
|
||
group_name = item.data(0)
|
||
available = [s for s in self.all_students if s not in self.group_manager.get_group_students(group_name)]
|
||
if not available:
|
||
QMessageBox.information(self, "提示", "所有学生都已在此分组中！")
|
||
return
|
||
student, ok = QInputDialog.getItem(self, "添加学生", "选择要添加的学生：", available, 0, False)
|
||
if ok and student:
|
||
self.group_manager.add_student_to_group(group_name, student)
|
||
self._on_group_selected(item)
|
||
self._refresh_group_list()
|
||
|
||
def _remove_student(self):
|
||
from PyQt5.QtWidgets import QMessageBox
|
||
item = self.group_list.currentItem()
|
||
if not item or item.data(0) == "__all__":
|
||
return
|
||
group_name = item.data(0)
|
||
student_item = self.student_list.currentItem()
|
||
if not student_item:
|
||
QMessageBox.information(self, "提示", "请先选择要移除的学生！")
|
||
return
|
||
student = student_item.text()
|
||
self.group_manager.remove_student_from_group(group_name, student)
|
||
self._on_group_selected(item)
|
||
self._refresh_group_list()
|
||
|
||
|
||
def create_group_manager(config_manager=None, base_dir=None) -> GroupManager:
|
||
"""创建分组管理器的工厂函数"""
|
||
return GroupManager(config_manager, base_dir)
|