# problems_dict = {
#     "第一讲": "1:20",
#     "第二讲": "50:60,70:80"
# } #题号字典, 每一项的key表示标题, 内容字符串表示题号
# notetitle = "测试批量讲义" #讲义标题, 也是文件标题
# answered = True #是否展示答案, True表示展示答案, False表示不展示答案

from database_tools import *


configjson = load_dict("文本文件/config.json")["批量讲义生成.py"]

problems_dict = configjson["标题与题号"]
notetitle = configjson["pdf标题"]
answered = configjson["提供答案"]


prodictpath = "../题库0.3/Problems.json"
pro_dict = load_dict(prodictpath)

outputdir = "临时文件" #输出文件的目录
outputfilepath = os.path.join(outputdir,notetitle+".tex")
print("输出文件目录: %s\n输出文件名: %s"%(os.path.join(os.getcwd(),outputdir),notetitle+".tex"))

latex_raw = ReadTextFile("模板文件/讲义模板.txt")
latex_raw = latex_raw.replace(r"学号\blank{50} \ 姓名\blank{80}","上海市控江中学")

if sys.platform != "win32": #非win系统用默认字体
    latex_raw = re.sub(r"fontset[\s]*=[\s]*none","fontset = fandol",latex_raw)
    latex_raw = re.sub(r"\\setCJKmainfont",r"% \\setCJKmainfont",latex_raw)

bodystring = "\\tableofcontents\n\n\\newpage\n\n"

for p in problems_dict: 
    bodystring += GenerateStudentBodyString([problems_dict[p]],[p],pro_dict,True,answered,False) + "\n\n\\newpage\n\n" #生成.tex中的内容主体字符串, 用于替换模板中的相应部分

latex_data = StringSubstitute(r"<<[\s\S]*?待替换[\s\S]*?>>",latex_raw,(notetitle,bodystring)) #替换标题和bodystring
SaveTextFile(latex_data,outputfilepath) #保存.tex文件

if XeLaTeXCompile(outputdir,notetitle+".tex"):
    print("编译成功")
else:
    print("编译失败")