import random
import pptx
from pptx.util import Pt
import re
from typing import Iterable

# Chinese colon and English colon is different but both should be supported with love!
choice_question_regex = re.compile(
    r'(?P<problem_1>\S*?)[\(（](?P<answer>[a-dA-D])[\)）](?P<problem_2>\S*?)\s*[aA]\s+(?P<choice_a>\S+?)\s+[bB]\s*(?P<choice_b>\S+?)\s+[cC]\s*(?P<choice_c>\S+?)\s+[dD]\s*(?P<choice_d>\S+?)\s+', re.MULTILINE)

choice_text = open('./choose.txt', 'r', encoding='utf-8').read()

questions = []

for match in choice_question_regex.finditer(choice_text):
    questions.append({
        'type': 'c',
        'problem': match.group('problem_1') + "（ ）" + match.group('problem_2'),
        'answer': match.group('answer').lower(),
        'choices': {
            'a': match.group('choice_a'),
            'b': match.group('choice_b'),
            'c': match.group('choice_c'),
            'd': match.group('choice_d')
        }
    })

judge_question_regex = re.compile(
    r'(?P<problem>\S+?)\s*[\(（](?P<answer>[a-bA-B])[\)）]', re.MULTILINE)

judge_text = open('./judge.txt', 'r', encoding='utf-8').read()


for match in judge_question_regex.finditer(judge_text):
    questions.append({
        'type': 'j',
        'problem': match.group('problem'),
        'answer': match.group('answer').lower()
    })


random.shuffle(questions)

presentation: pptx.presentation.Presentation = pptx.Presentation()

buttons: Iterable[pptx.shapes.autoshape.Shape] = []

for question in questions:
    slide: pptx.slide.Slide = presentation.slides.add_slide(
        presentation.slide_layouts[6])

    title = slide.shapes.add_textbox(Pt(20), Pt(40), Pt(680), Pt(50))
    problem = slide.shapes.add_textbox(Pt(20), Pt(120), Pt(680), Pt(350))
    answer = slide.shapes.add_textbox(Pt(560), Pt(450), Pt(40), Pt(40))

    if question['type'] == 'c':
        title.text = '选择题'
        problem.text = question['problem'] + '\n' + '\n'.join(
            [choice.upper() + '. ' + question['choices'][choice]for choice in 'abcd'])
        answer.text = question['answer'].upper()
    else:
        title.text = '判断题'
        problem.text = question['problem']
        answer.text = '√' if question['answer'] == 'a' else '×'

    title_text_frame: pptx.text.text.TextFrame = title.text_frame
    para: pptx.text.text._Paragraph
    for para in title_text_frame.paragraphs:
        para.font.size = Pt(42)
        para.font.color.theme_color = pptx.enum.dml.MSO_THEME_COLOR.TEXT_2

    problem_text_frame: pptx.text.text.TextFrame = problem.text_frame
    problem_text_frame.margin_top = Pt(10)
    para: pptx.text.text._Paragraph
    for para in problem_text_frame.paragraphs:
        para.space_after = Pt(6)
        para.font.size = Pt(30)
    problem_text_frame.word_wrap = True

    answer_text_frame: pptx.text.text.TextFrame = answer.text_frame
    para: pptx.text.text._Paragraph
    for para in answer_text_frame.paragraphs:
        para.font.size = Pt(40)
        para.font.color.rgb = pptx.dml.color.RGBColor(255, 0, 0)

    buttons.append(slide.shapes.add_shape(pptx.enum.shapes.MSO_SHAPE.OVAL, Pt(
        660), Pt(460), Pt(40), Pt(40)))

slide: pptx.slide.Slide = presentation.slides.add_slide(
    presentation.slide_layouts[6])

for button in buttons:
    button.click_action.target_slide = slide

presentation.save('test.pptx')