104 lines
3.7 KiB
Python
104 lines
3.7 KiB
Python
from file_list import Files, FileHandle, FileType
|
|
from pathlib import Path
|
|
from typing import Dict
|
|
from utils import normalize
|
|
from config import NEW_EXERCISE_SHEET_HEADER
|
|
|
|
|
|
class ExerciseWriteUp(FileHandle):
|
|
def __init__(self, root_dir: Path, course):
|
|
self.root_dir = root_dir
|
|
self.course = course
|
|
FileHandle.__init__(self, next(self.root_dir.rglob('*.tex')), FileType.tex)
|
|
|
|
|
|
class Exercise:
|
|
def __init__(self, course, number: int):
|
|
self.course = course
|
|
self.number = number
|
|
self._writeup = None
|
|
self._problem = None
|
|
self._solution = None
|
|
|
|
@property
|
|
def writeup(self):
|
|
if not self._writeup:
|
|
self._writeup = next((w for w in self.course.writeups if w.number == self.number), None)
|
|
return self._writeup
|
|
|
|
@property
|
|
def problem(self):
|
|
if not self._problem:
|
|
self._problem = next((p for p in self.course.problems if p.number == self.number), None)
|
|
return self._problem
|
|
|
|
@property
|
|
def solution(self):
|
|
if not self._solution:
|
|
self._solution = next((s for s in self.course.solutions if s.number == self.number), None)
|
|
return self._solution
|
|
|
|
|
|
class Exercises(list):
|
|
def __init__(self, course):
|
|
self.course = course
|
|
self.info: Dict = course.info['exercises']
|
|
self.root: Path = course.path / self.info['path']
|
|
self.sheet_root = self.root / self.info['sheets'].strip()
|
|
self.solutions_root = self.root / self.info['solutions'].strip()
|
|
self.root.mkdir(parents=True, exist_ok=True)
|
|
self.sheet_root.mkdir(parents=True, exist_ok=True)
|
|
self.solutions_root.mkdir(parents=True, exist_ok=True)
|
|
self._solutions = None
|
|
self._writeups = None
|
|
self._sheets = Files(self.sheet_root)
|
|
self.ignored_folders = [self.sheet_root, self.solutions_root, self.root / 'LatexPackagesBuild', self.root / '.git']
|
|
list.__init__(self, (Exercise(self.course, num) for num in map(lambda s: s.number, self._sheets)))
|
|
|
|
@property
|
|
def sheets(self):
|
|
return self._sheets
|
|
|
|
@property
|
|
def solutions(self):
|
|
if not self._solutions:
|
|
self._solutions = Files(self.solutions_root)
|
|
return self._solutions
|
|
|
|
@property
|
|
def writeups(self):
|
|
if not self._writeups:
|
|
dirs = list(d for d in self.root.iterdir() if d.is_dir() and d not in self.ignored_folders)
|
|
self._writeups = sorted((ExerciseWriteUp(d, self.course) for d in dirs), key=lambda e: e.number)
|
|
return self._writeups
|
|
|
|
@staticmethod
|
|
def __generate_name(name: str):
|
|
return normalize(name.split(' ')[-1])
|
|
|
|
def __generate_names(self):
|
|
names = self.info['name']
|
|
if type(names) == str:
|
|
return self.__generate_name(names)
|
|
elif type(names) == list:
|
|
return '_'.join(map(self.__generate_name, names))
|
|
|
|
def new_writeup(self):
|
|
try:
|
|
new_num = max(self.writeups, key=lambda w: w.number).number + 1
|
|
except ValueError:
|
|
new_num = 1
|
|
new_dir = self.root / 'ub{:02d}'.format(new_num)
|
|
new_dir.mkdir(parents=True, exist_ok=False)
|
|
new_file = new_dir / '{names}_{course}_sheet_{num}.tex'.format(
|
|
names=self.__generate_names(),
|
|
course=normalize(self.course.info['short']),
|
|
num=new_num
|
|
)
|
|
new_file.write_text(NEW_EXERCISE_SHEET_HEADER.format(
|
|
language='ngerman' if self.course.info['language'] == 'german' else 'english',
|
|
author=self.info['name'] if type(self.info['name']) == str else ', '.join(self.info['name']),
|
|
course=self.course.info['title'],
|
|
number=new_num
|
|
))
|
|
return ExerciseWriteUp(new_dir, self.course)
|