31 lines
789 B
Python
31 lines
789 B
Python
import re
|
|
from functools import lru_cache
|
|
|
|
|
|
TEMPLATE_BASE = """//PREPEND BEGIN
|
|
{}
|
|
//PREPEND END
|
|
|
|
//TEMPLATE BEGIN
|
|
{}
|
|
//TEMPLATE END
|
|
|
|
//APPEND BEGIN
|
|
{}
|
|
//APPEND END"""
|
|
|
|
|
|
@lru_cache(maxsize=100)
|
|
def parse_problem_template(template_str):
|
|
prepend = re.findall(r"//PREPEND BEGIN\n([\s\S]+?)//PREPEND END", template_str)
|
|
template = re.findall(r"//TEMPLATE BEGIN\n([\s\S]+?)//TEMPLATE END", template_str)
|
|
append = re.findall(r"//APPEND BEGIN\n([\s\S]+?)//APPEND END", template_str)
|
|
return {"prepend": prepend[0] if prepend else "",
|
|
"template": template[0] if template else "",
|
|
"append": append[0] if append else ""}
|
|
|
|
|
|
@lru_cache(maxsize=100)
|
|
def build_problem_template(prepend, template, append):
|
|
return TEMPLATE_BASE.format(prepend, template, append)
|