156 lines
4.2 KiB
Python
156 lines
4.2 KiB
Python
from invoke import task
|
|
from pathlib import Path
|
|
from prompt_toolkit import prompt
|
|
from prompt_toolkit.completion import WordCompleter
|
|
from prompt_toolkit.validation import Validator
|
|
from unidecode import unidecode
|
|
import jinja2
|
|
from pytex import texenv
|
|
import datetime
|
|
import locale
|
|
|
|
locale.setlocale(locale.LC_ALL, "fr_FR.UTF-8")
|
|
|
|
TRIBES = [
|
|
{"name": "Terminale Technologique", "directory": "TST"},
|
|
{"name": "Terminale Terminale spé sti2d", "directory": "TST_sti2d"},
|
|
{"name": "Science numérique et technologique", "directory": "SNT"},
|
|
{"name": "Enseignement scientifique", "directory": "EnsSci"},
|
|
{"name": "Maths Complémentaires", "directory": "Complementaire"},
|
|
]
|
|
|
|
AUTHOR = "Benjamin Bertrand"
|
|
|
|
|
|
def is_tribe(tribe):
|
|
return tribe in [t["directory"] for t in TRIBES]
|
|
|
|
|
|
def get_sequences(tribe):
|
|
p = Path(tribe)
|
|
return [x for x in p.iterdir() if (x.is_dir() and x.name[0:2].isdigit())]
|
|
|
|
|
|
def get_tags(tribe=""):
|
|
p = Path(tribe)
|
|
tags = set()
|
|
indexes = p.glob("**/index.rst")
|
|
for index_p in indexes:
|
|
with open(index_p, "r") as index:
|
|
line = index.readline()
|
|
while line:
|
|
if line.startswith(":tags"):
|
|
tags.update(line[7:].replace("\n", "").split(", "))
|
|
break
|
|
line = index.readline()
|
|
return {t.lower() for t in tags}
|
|
|
|
|
|
def next_weekday(date, day=0):
|
|
return date + datetime.timedelta(days=(day - date.weekday() + 7) % 7)
|
|
|
|
|
|
def sequence_prompt(
|
|
author=AUTHOR,
|
|
tribe="",
|
|
title="",
|
|
tags="",
|
|
summary="",
|
|
*args,
|
|
**kwrds,
|
|
):
|
|
sequence = {}
|
|
print("Nouvelle séquence")
|
|
|
|
sequence["author"] = prompt("Auteur: ", default=author)
|
|
|
|
sequence["date"] = datetime.date.today()
|
|
|
|
tribes = WordCompleter([t["directory"] for t in TRIBES])
|
|
sequence["tribe"] = prompt(
|
|
"Classe: ",
|
|
completer=tribes,
|
|
validator=Validator.from_callable(is_tribe),
|
|
default=tribe,
|
|
)
|
|
|
|
sequences = get_sequences(sequence["tribe"])
|
|
if sequences:
|
|
print("Séquences trouvées:\n\t" +
|
|
"\n\t".join(sorted([s.name for s in sequences])))
|
|
sequence["num"] = max([int(seq.name[0:2]) for seq in sequences]) + 1
|
|
else:
|
|
print("Pas de séquence trouvée")
|
|
sequence["num"] = 1
|
|
|
|
sequence["title"] = prompt(
|
|
f"Nom de la séquence (n°{sequence['num']}): ", default=title
|
|
)
|
|
|
|
known_tags = get_tags()
|
|
sequence["tags"] = prompt(
|
|
"Tags (séparés par une virgule): ",
|
|
completer=WordCompleter(known_tags),
|
|
default=tags,
|
|
)
|
|
|
|
sequence["summary"] = prompt("Résumé: ", default=summary)
|
|
return sequence
|
|
|
|
|
|
def import_skel(dest, datas, skel="sequence", skel_path="./tools/skeleton"):
|
|
d = Path(dest)
|
|
d.mkdir()
|
|
|
|
p = Path(skel_path + "/" + skel)
|
|
files = [t.name for t in p.iterdir()]
|
|
|
|
template_loader = jinja2.FileSystemLoader(searchpath=p)
|
|
template_env = jinja2.Environment(loader=template_loader)
|
|
|
|
texenv.loader.loaders.append(template_loader)
|
|
|
|
print("")
|
|
for filename in files:
|
|
if filename.endswith(".tex"):
|
|
template = texenv.get_template(filename)
|
|
else:
|
|
template = template_env.get_template(filename)
|
|
|
|
output_text = template.render(**datas)
|
|
|
|
with open(d / filename, "w") as f:
|
|
f.write(output_text)
|
|
print(f"Importation de {filename} - OK")
|
|
|
|
|
|
@task
|
|
def sequence(c):
|
|
sequence = {}
|
|
validate = False
|
|
while not validate:
|
|
sequence = sequence_prompt(**sequence)
|
|
|
|
dest = unidecode(
|
|
("{tribe}/{num:0>2}_{title}/".format(**sequence)).replace(" ", "_")
|
|
)
|
|
dest = Path(dest)
|
|
|
|
seq_summary = """\
|
|
Séquence: {num:0>2} {title}
|
|
Niveau: {tribe}
|
|
Auteur: {author}
|
|
Date: {date}
|
|
Tags: {tags}
|
|
Sommaire: {summary}
|
|
Dossier: {dest}
|
|
""".format(
|
|
dest=dest, **sequence
|
|
)
|
|
print(seq_summary)
|
|
|
|
validate = prompt("C'est ok? ").lower() in ["o", "k", "oui", "ok"]
|
|
|
|
sequence["title_under"] = "#" * len(sequence["title"])
|
|
import_skel(dest, sequence, "sequence")
|