101 lines
3.6 KiB
Python
101 lines
3.6 KiB
Python
#!/usr/bin/python3
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
from lectures import Lectures, number2filename
|
|
from config import DEFAULT_MASTER_FILE_NAME, LECTURE_START_MARKER, LECTURE_END_MARKER, DEFAULT_IMPORT_INDENTATION
|
|
from parse_counters import parse_counters, dict2setcounters
|
|
|
|
|
|
class Notes:
|
|
def __init__(self, course):
|
|
self.course = course
|
|
if 'notes' in course.info:
|
|
self.info = course.info['notes']
|
|
else:
|
|
self.info = []
|
|
if 'path' in self.info:
|
|
self.root = course.path / self.info['path']
|
|
self.root.mkdir(parents=True, exist_ok=True)
|
|
else:
|
|
self.root = course.path
|
|
if 'master_file' in self.info:
|
|
self.master_file = self.root / self.info['master_file']
|
|
else:
|
|
self.master_file = self.root / DEFAULT_MASTER_FILE_NAME
|
|
if 'full_file' in self.info:
|
|
self.full_file: Path = self.root / self.info['full_file']
|
|
else:
|
|
self.full_file: Path = None
|
|
self._lectures = None
|
|
|
|
@staticmethod
|
|
def get_header_footer(filepath):
|
|
part = 0
|
|
header = ''
|
|
footer = ''
|
|
with filepath.open() as f:
|
|
for line in f:
|
|
# order of if-statements is important here!
|
|
if LECTURE_END_MARKER in line:
|
|
part = 2
|
|
|
|
if part == 0:
|
|
header += line
|
|
if part == 2:
|
|
footer += line
|
|
|
|
if LECTURE_START_MARKER in line:
|
|
part = 1
|
|
return header, footer
|
|
|
|
def new_lecture(self):
|
|
lec = self.lectures.new_lecture()
|
|
if lec.number == 1:
|
|
self.update_lectures_in_master([1])
|
|
else:
|
|
self.update_lectures_in_master([lec.number - 1, lec.number])
|
|
self.update_lectures_in_full(self.lectures.parse_range_string('all'))
|
|
return lec
|
|
|
|
def input_lecture_command(self, num: int):
|
|
if self.lectures.root.relative_to(self.root) == Path('.'):
|
|
input_command = r'\input{'
|
|
else:
|
|
input_command = r'\import{' + str(self.lectures.root.relative_to(self.root)) + '/}{'
|
|
return ' ' * DEFAULT_IMPORT_INDENTATION + input_command + number2filename(num) + '}\n'
|
|
|
|
def set_counters(self, lecture_list, lec, setcounters=False):
|
|
if not setcounters:
|
|
return ''
|
|
if lec - 1 not in lecture_list and self.full_file:
|
|
return dict2setcounters(parse_counters(self.full_file.with_suffix('.counters'), {'lecture': lec}))
|
|
return ''
|
|
|
|
def update_lectures_in_file(self, filename, lecture_list, setcounters=False):
|
|
header, footer = self.get_header_footer(filename)
|
|
body = ''.join([self.set_counters(lecture_list, num, setcounters) + self.input_lecture_command(num)
|
|
for num in lecture_list])
|
|
filename.write_text(header + body + footer)
|
|
|
|
def update_lectures_in_master(self, lecture_list):
|
|
self.update_lectures_in_file(self.master_file, lecture_list, True)
|
|
|
|
def update_lectures_in_full(self, lecture_list):
|
|
if self.full_file:
|
|
self.update_lectures_in_file(self.full_file, lecture_list)
|
|
|
|
def compile_master(self):
|
|
result = subprocess.run(
|
|
['latexmk', '-f', '-interaction=nonstopmode', str(self.master_file)],
|
|
stdout=subprocess.DEVNULL,
|
|
stderr=subprocess.DEVNULL,
|
|
cwd=str(self.root)
|
|
)
|
|
return result.returncode
|
|
|
|
@property
|
|
def lectures(self):
|
|
if not self._lectures:
|
|
self._lectures = Lectures(self)
|
|
return self._lectures
|