130 lines
3.4 KiB
Python
130 lines
3.4 KiB
Python
from typing import Optional
|
|
|
|
from openpyxl.reader.excel import load_workbook
|
|
from openpyxl.workbook.workbook import Workbook
|
|
from openpyxl.worksheet.worksheet import Worksheet
|
|
|
|
|
|
class PickerStudent:
|
|
total_time: int = 0
|
|
|
|
def __init__(self, name: str, position: int, scores: list[int]):
|
|
self._name = name
|
|
self._position = position
|
|
self._scores = scores
|
|
self._modify = False
|
|
self._scores = []
|
|
self._scores.extend(scores)
|
|
self._scores_pointer = -1
|
|
|
|
self.init_score()
|
|
|
|
def init_score(self):
|
|
self._scores += [0] * (self.total_time - len(self._scores))
|
|
|
|
for idx, val in enumerate(self._scores):
|
|
if val is None:
|
|
self._scores[idx] = 0
|
|
if self._scores[idx] == 0:
|
|
self._scores_pointer = idx
|
|
break
|
|
|
|
def append_score(self, score: int) -> int:
|
|
if self._scores_pointer > self.total_time or self._scores_pointer < 0:
|
|
raise IndexError
|
|
self._scores[self._scores_pointer] = score
|
|
self._scores_pointer += 1
|
|
self.modified()
|
|
return self._scores_pointer - 1
|
|
|
|
def change_score(self, new_score: int, index: int):
|
|
if index < 0 or index > self.total_time:
|
|
raise IndexError
|
|
self._scores[index] = new_score
|
|
self.modified()
|
|
|
|
@classmethod
|
|
def set_total_time(cls, total_time: int):
|
|
cls.total_time = total_time
|
|
|
|
@property
|
|
def position(self) -> int:
|
|
return self._position
|
|
|
|
@property
|
|
def scores(self) -> list[int]:
|
|
return self._scores
|
|
|
|
@property
|
|
def modify(self) -> bool:
|
|
return self._modify
|
|
|
|
def saved(self):
|
|
self._modify = False
|
|
|
|
def modified(self):
|
|
self._modify = True
|
|
|
|
def __str__(self):
|
|
return f"PickerStudent {self._name} at {self.position}, with scores: {','.join(str(x) for x in self._scores if x)}"
|
|
|
|
def __repr__(self):
|
|
return self.__str__()
|
|
|
|
|
|
class PickerExcel:
|
|
wb: Optional[Workbook]
|
|
ws: Optional[Worksheet]
|
|
path: str = ''
|
|
max_time_position = 'L1'
|
|
|
|
def __init__(self, path: str):
|
|
self.open(path)
|
|
|
|
@classmethod
|
|
def open(cls, path: str):
|
|
cls.path = path
|
|
cls.wb = load_workbook(path)
|
|
cls.ws = cls.wb.active
|
|
PickerStudent.set_total_time(cls.ws[cls.max_time_position].value)
|
|
|
|
@classmethod
|
|
def read_student(cls, path: Optional[str] = None) -> list[PickerStudent]:
|
|
if path:
|
|
cls.open(path)
|
|
|
|
if cls.wb and cls.ws:
|
|
ret = []
|
|
for row in cls.ws.iter_rows(min_row=4, max_col=4 + cls.ws[cls.max_time_position].value, values_only=True):
|
|
if (name := row[2]) is None:
|
|
break
|
|
ret.append(PickerStudent(name, int(row[0]) + 3, row[4:]))
|
|
return ret
|
|
else:
|
|
raise Exception('No Workbook or Worksheet')
|
|
|
|
@classmethod
|
|
def write_back(cls, student: PickerStudent):
|
|
start_col = 5 # E列
|
|
row = student.position
|
|
|
|
for i, val in enumerate(student.scores):
|
|
cls.ws.cell(row=row, column=start_col + i, value=val if val != 0 else '')
|
|
|
|
student.saved()
|
|
|
|
@classmethod
|
|
def write_all(cls, students: list[PickerStudent]):
|
|
for student in students:
|
|
if student.modify:
|
|
cls.write_back(student)
|
|
|
|
@classmethod
|
|
def save(cls):
|
|
if cls.ws:
|
|
cls.wb.save(cls.path)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
...
|