from epub import Epub
from filetools import FileTools
from zktools import ZkTools

if __name__ == '__main__':
    file = FileTools()
    file.mkdir("temp")
    file.mkdir("output")
    zk=ZkTools()
    course = zk.get_courses()
    data={}
    for i in course:
        course_name = i['name']
        data[i['name']]=[]
        questions = zk.get_questions(i['id'])
        for j in questions:
            data[i['name']].append({"name":j['name'], "data":zk.get_content(j['id'])})

    for i in data:
        questions=data[i]
        for question_index in range(0, len(questions)):
            question=questions[question_index]
            epub = Epub("百日题库-%s-%02d-%s" % (i.split(' ')[1], question_index+1, question['name'].replace('/', '_')))
            epub.process(zk.format(question['data']))
            epub.clean()