mirror of
https://github.com/QingdaoU/OnlineJudge.git
synced 2024-12-29 08:32:08 +00:00
27 lines
705 B
Python
27 lines
705 B
Python
import re
|
|
|
|
TEMPLATE_BASE = """//PREPEND BEGIN
|
|
{}
|
|
//PREPEND END
|
|
|
|
//TEMPLATE BEGIN
|
|
{}
|
|
//TEMPLATE END
|
|
|
|
//APPEND BEGIN
|
|
{}
|
|
//APPEND END"""
|
|
|
|
|
|
def parse_problem_template(template_str):
|
|
prepend = re.findall("//PREPEND BEGIN\n([\s\S]+?)//PREPEND END", template_str)
|
|
template = re.findall("//TEMPLATE BEGIN\n([\s\S]+?)//TEMPLATE END", template_str)
|
|
append = re.findall("//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 ""}
|
|
|
|
|
|
def build_problem_template(prepend, template, append):
|
|
return TEMPLATE_BASE.format(prepend, template, append)
|