mirror of
https://github.com/xlucn/PAT.git
synced 2026-11-02 18:37:42 +08:00
83 lines
2.3 KiB
Python
Executable File
83 lines
2.3 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
|
|
import os, sys, re
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
|
from config import indexes, code_dir, md_dir, html_dir, analysis_dir, configs
|
|
|
|
|
|
def build_file_(category, index):
|
|
"""
|
|
build markdown files.
|
|
"""
|
|
filename = "{}/{}{:04}.md".format(md_dir, category, index)
|
|
|
|
# parse using bs4 and find title and problem content
|
|
html = "{}/{}{}.html".format(html_dir, category, index)
|
|
lines = open(html).read().split('\n')
|
|
|
|
# code related
|
|
code_rel_path = "{}/{}.c".format(code_dirs[category], index)
|
|
repo_url = "https://github.com/OliverLew/PAT/blob/master"
|
|
file_github_path = "{}/{}".format(repo_url, code_rel_path)
|
|
raw_code = open(os.path.join(code_dir, code_rel_path)).read()
|
|
code = raw_code[raw_code.index("#include"):]
|
|
|
|
|
|
print("Building {}".format(filename))
|
|
|
|
# write everything into a file
|
|
with open(filename, 'w') as f:
|
|
f.write("{}\n\n".format(lines[2]))
|
|
|
|
# write problem content
|
|
f.write("## 题目\n\n")
|
|
for line in lines[3:]:
|
|
if configs["quote_text"] is True:
|
|
f.write("> ")
|
|
f.write(line + "\n")
|
|
|
|
# write explanation
|
|
expl = open("{}/{}{}.md".format(analysis_dir, category, index)).read()
|
|
f.write("\n## 思路\n\n")
|
|
f.write(expl)
|
|
|
|
# write code
|
|
f.write("\n## 代码\n\n")
|
|
f.write("[最新代码@github](" + file_github_path + ")，欢迎交流\n")
|
|
f.write("```c\n{}\n```".format(code))
|
|
|
|
|
|
def build_file(c, i):
|
|
try:
|
|
build_file_(c, i)
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
code_dirs = {
|
|
'a': 'PATAdvanced',
|
|
'b': 'PATBasic',
|
|
't': 'PATTop'
|
|
}
|
|
|
|
usage = """to be continued."""
|
|
|
|
if __name__ == "__main__":
|
|
if(not os.path.exists(md_dir)):
|
|
os.mkdir(md_dir)
|
|
if len(sys.argv) == 1:
|
|
for c in list(indexes.keys()):
|
|
for i in indexes[c]:
|
|
build_file(c, i)
|
|
if len(sys.argv) == 2:
|
|
if sys.argv[1] == '-h' or sys.argv[1] == '--help':
|
|
print(usage)
|
|
elif re.match(r"[abt]\d{4}", sys.argv[1]):
|
|
category = sys.argv[1][0]
|
|
index = int(sys.argv[1][1:])
|
|
if index in indexes[category]:
|
|
build_file(category, index)
|
|
else:
|
|
print(usage)
|