chgksuite 0.25.1__py3-none-any.whl → 0.26.0__py3-none-any.whl

chgksuite/composer/docx.py

@@ -1,4 +1,5 @@
 import os
+import re
 import shutil
 import sys
 import tempfile
@@ -12,8 +13,14 @@ from docx.oxml.ns import qn
 from docx.shared import Inches
 from docx.shared import Pt as DocxPt
 
-from chgksuite.common import log_wrap, replace_escaped
-from chgksuite.composer.composer_common import BaseExporter, backtick_replace, parseimg
+import chgksuite.typotools as typotools
+from chgksuite.common import DummyLogger, get_chgksuite_dir, log_wrap, replace_escaped
+from chgksuite.composer.composer_common import (
+    BaseExporter,
+    _parse_4s_elem,
+    backtick_replace,
+    parseimg,
+)
 
 WHITEN = {
     "handout": False,
@@ -59,6 +66,430 @@ def replace_font_in_docx(template_path, new_font):
     return temp_template
 
 
+def replace_no_break_standalone(s, replace_spaces=True, replace_hyphens=True):
+    """Standalone version of _replace_no_break"""
+    return typotools.replace_no_break(s, spaces=replace_spaces, hyphens=replace_hyphens)
+
+
+def get_label_standalone(
+    question, field, labels, language="ru", only_question_number=False, number=None
+):
+    """Standalone version of get_label"""
+    if field == "question" and only_question_number:
+        return str(question.get("number") or number)
+    if field in ("question", "tour"):
+        lbl = (question.get("overrides") or {}).get(field) or labels["question_labels"][
+            field
+        ]
+        num = question.get("number") or number
+        if language in ("uz", "uz_cyr"):
+            return f"{num} – {lbl}"
+        elif language == "kz":
+            return f"{num}-{lbl}"
+        else:
+            return f"{lbl} {num}"
+    if field in (question.get("overrides") or {}):
+        return question["overrides"][field]
+    if field == "source" and isinstance(question.get("source" or ""), list):
+        return labels["question_labels"]["sources"]
+    return labels["question_labels"][field]
+
+
+def remove_square_brackets_standalone(s, labels):
+    """Standalone version of remove_square_brackets"""
+    hs = labels["question_labels"]["handout_short"]
+    s = s.replace("\\[", "LEFTSQUAREBRACKET")
+    s = s.replace("\\]", "RIGHTSQUAREBRACKET")
+    s = re.sub(f"\\[{hs}(.+?)\\]", "{" + hs + "\\1}", s, flags=re.DOTALL)
+    i = 0
+    while "[" in s and "]" in s and i < 10:
+        s = re.sub(" *\\[.+?\\]", "", s, flags=re.DOTALL)
+        s = s.strip()
+        i += 1
+    if i == 10:
+        sys.stderr.write(
+            f"Error replacing square brackets on question: {s}, retries exceeded\n"
+        )
+    s = re.sub("\\{" + hs + "(.+?)\\}", "[" + hs + "\\1]", s, flags=re.DOTALL)
+    s = s.replace("LEFTSQUAREBRACKET", "[")
+    s = s.replace("RIGHTSQUAREBRACKET", "]")
+    return s
+
+
+def add_hyperlink_to_docx(doc, paragraph, text, url):
+    """Standalone version of add_hyperlink"""
+    run = paragraph.add_run(text)
+    run.style = doc.styles["Hyperlink"]
+    part = paragraph.part
+    r_id = part.relate_to(
+        url, docx.opc.constants.RELATIONSHIP_TYPE.HYPERLINK, is_external=True
+    )
+    hyperlink = docx.oxml.shared.OxmlElement("w:hyperlink")
+    hyperlink.set(docx.oxml.shared.qn("r:id"), r_id)
+    hyperlink.append(run._r)
+    paragraph._p.append(hyperlink)
+    return hyperlink
+
+
+def format_docx_element(
+    doc,
+    el,
+    para,
+    whiten,
+    spoilers="none",
+    logger=None,
+    labels=None,
+    language="ru",
+    remove_accents=False,
+    remove_brackets=False,
+    replace_no_break_spaces=False,
+    **kwargs,
+):
+    """
+    Standalone version of docx_format that can be used outside DocxExporter.
+
+    Args:
+        doc: docx Document object
+        el: Element to format
+        para: Paragraph object to add content to
+        whiten: Whether to apply whitening
+        spoilers: Spoiler handling mode ("none", "whiten", "dots", "pagebreak")
+        logger: Logger instance
+        labels: Labels dictionary
+        language: Language code
+        remove_accents: Whether to remove accents
+        remove_brackets: Whether to remove square brackets
+        replace_no_break_spaces: Whether to replace non-breaking spaces
+        **kwargs: Additional arguments (tmp_dir, targetdir, etc.)
+    """
+    if logger is None:
+        logger = DummyLogger()
+
+    if isinstance(el, list):
+        if len(el) > 1 and isinstance(el[1], list):
+            format_docx_element(
+                doc,
+                el[0],
+                para,
+                whiten,
+                spoilers,
+                logger,
+                labels,
+                language,
+                remove_accents,
+                remove_brackets,
+                replace_no_break_spaces,
+                **kwargs,
+            )
+            licount = 0
+            for li in el[1]:
+                licount += 1
+                para.add_run("\n{}. ".format(licount))
+                format_docx_element(
+                    doc,
+                    li,
+                    para,
+                    whiten,
+                    spoilers,
+                    logger,
+                    labels,
+                    language,
+                    remove_accents,
+                    remove_brackets,
+                    replace_no_break_spaces,
+                    **kwargs,
+                )
+        else:
+            licount = 0
+            for li in el:
+                licount += 1
+                para.add_run("\n{}. ".format(licount))
+                format_docx_element(
+                    doc,
+                    li,
+                    para,
+                    whiten,
+                    spoilers,
+                    logger,
+                    labels,
+                    language,
+                    remove_accents,
+                    remove_brackets,
+                    replace_no_break_spaces,
+                    **kwargs,
+                )
+
+    if isinstance(el, str):
+        logger.debug("parsing element {}:".format(log_wrap(el)))
+
+        if remove_accents:
+            el = el.replace("\u0301", "")
+        if remove_brackets and labels:
+            el = remove_square_brackets_standalone(el, labels)
+        else:
+            el = replace_escaped(el)
+
+        el = backtick_replace(el)
+
+        for run in _parse_4s_elem(el, logger=logger):
+            if run[0] == "pagebreak":
+                if spoilers == "dots":
+                    for _ in range(30):
+                        para = doc.add_paragraph()
+                        para.add_run(".")
+                    para = doc.add_paragraph()
+                else:
+                    para = doc.add_page_break()
+            elif run[0] == "linebreak":
+                para.add_run("\n")
+            elif run[0] == "screen":
+                if remove_accents or remove_brackets:
+                    text = run[1]["for_screen"]
+                else:
+                    text = run[1]["for_print"]
+                if replace_no_break_spaces:
+                    text = replace_no_break_standalone(text)
+                r = para.add_run(text)
+            elif run[0] == "hyperlink" and not (whiten and spoilers == "whiten"):
+                r = add_hyperlink_to_docx(doc, para, run[1], run[1])
+            elif run[0] == "img":
+                if run[1].endswith(".shtml"):
+                    r = para.add_run("(ТУТ БЫЛА ССЫЛКА НА ПРОТУХШУЮ КАРТИНКУ)\n")
+                    continue
+                parsed_image = parseimg(
+                    run[1],
+                    dimensions="inches",
+                    tmp_dir=kwargs.get("tmp_dir"),
+                    targetdir=kwargs.get("targetdir"),
+                )
+                imgfile = parsed_image["imgfile"]
+                width = parsed_image["width"]
+                height = parsed_image["height"]
+                inline = parsed_image["inline"]
+                if inline:
+                    r = para.add_run("")
+                else:
+                    r = para.add_run("\n")
+
+                try:
+                    if inline:
+                        r.add_picture(imgfile, height=Inches(1.0 / 6))
+                    else:
+                        r.add_picture(
+                            imgfile, width=Inches(width), height=Inches(height)
+                        )
+                except UnrecognizedImageError:
+                    sys.stderr.write(
+                        f"python-docx can't recognize header for {imgfile}\n"
+                    )
+                if not inline:
+                    r = para.add_run("\n")
+                continue
+            else:
+                text = run[1]
+                if replace_no_break_spaces:
+                    text = replace_no_break_standalone(text)
+                r = para.add_run(text)
+                if "italic" in run[0]:
+                    r.italic = True
+                if "bold" in run[0]:
+                    r.bold = True
+                if "underline" in run[0]:
+                    r.underline = True
+                if run[0] == "strike":
+                    r.font.strike = True
+                if run[0] == "sc":
+                    r.small_caps = True
+                if whiten and spoilers == "whiten":
+                    r.style = "Whitened"
+
+
+def add_question_to_docx(
+    doc,
+    question_data,
+    labels,
+    qcount=None,
+    skip_qcount=False,
+    screen_mode=False,
+    external_para=None,
+    noparagraph=False,
+    noanswers=False,
+    spoilers="none",
+    language="ru",
+    only_question_number=False,
+    add_question_label=True,
+    logger=None,
+    **kwargs,
+):
+    """
+    Standalone function to add a question to a docx document.
+
+    Args:
+        doc: docx Document object
+        question_data: Dictionary containing question data
+        labels: Labels dictionary
+        qcount: Current question count (will be incremented if not skip_qcount)
+        skip_qcount: Whether to skip incrementing question count
+        screen_mode: Whether to use screen mode formatting
+        external_para: External paragraph to use instead of creating new ones
+        noparagraph: Whether to skip paragraph breaks
+        noanswers: Whether to skip adding answers
+        spoilers: Spoiler handling mode ("none", "whiten", "dots", "pagebreak")
+        language: Language code
+        only_question_number: Whether to show only question numbers
+        logger: Logger instance
+        **kwargs: Additional arguments passed to format_docx_element
+
+    Returns:
+        Updated question count
+    """
+    if not kwargs.get("tmp_dir"):
+        kwargs["tmp_dir"] = tempfile.mkdtemp()
+    if not kwargs.get("targetdir"):
+        kwargs["targetdir"] = os.getcwd()
+    if logger is None:
+        logger = DummyLogger()
+
+    q = question_data
+    if external_para is None:
+        p = doc.add_paragraph()
+    else:
+        p = external_para
+    if add_question_label:
+        p.paragraph_format.space_before = DocxPt(18)
+    p.paragraph_format.keep_together = True
+
+    # Handle question numbering
+    if qcount is None:
+        qcount = 1
+    if "number" not in q and not skip_qcount:
+        qcount += 1
+    if "setcounter" in q:
+        qcount = int(q["setcounter"])
+
+    # Add question label
+    if add_question_label:
+        question_label = get_label_standalone(
+            q,
+            "question",
+            labels,
+            language,
+            only_question_number,
+            number=qcount if "number" not in q else q["number"],
+        )
+        p.add_run(f"{question_label}. ").bold = True
+
+    # Add handout if present
+    if "handout" in q:
+        handout_label = get_label_standalone(q, "handout", labels, language)
+        p.add_run(f"\n[{handout_label}: ")
+        format_docx_element(
+            doc,
+            q["handout"],
+            p,
+            WHITEN["handout"],
+            spoilers,
+            logger,
+            labels,
+            language,
+            remove_accents=screen_mode,
+            remove_brackets=screen_mode,
+            **kwargs,
+        )
+        p.add_run("\n]")
+
+    if not noparagraph:
+        p.add_run("\n")
+
+    # Add question text
+    format_docx_element(
+        doc,
+        q["question"],
+        p,
+        False,
+        spoilers,
+        logger,
+        labels,
+        language,
+        remove_accents=screen_mode,
+        remove_brackets=screen_mode,
+        replace_no_break_spaces=True,
+        **kwargs,
+    )
+
+    # Add answers and other fields if not disabled
+    if not noanswers:
+        if spoilers == "pagebreak":
+            p = doc.add_page_break()
+        elif spoilers == "dots":
+            for _ in range(30):
+                if external_para is None:
+                    p = doc.add_paragraph()
+                else:
+                    p.add_run("\n")
+                p.add_run(".")
+            if external_para is None:
+                p = doc.add_paragraph()
+            else:
+                p.add_run("\n")
+        else:
+            if external_para is None:
+                p = doc.add_paragraph()
+            else:
+                p.add_run("\n")
+
+        p.paragraph_format.keep_together = True
+        p.paragraph_format.space_before = DocxPt(6)
+
+        # Add answer
+        answer_label = get_label_standalone(q, "answer", labels, language)
+        p.add_run(f"{answer_label}: ").bold = True
+        format_docx_element(
+            doc,
+            q["answer"],
+            p,
+            True,
+            spoilers,
+            logger,
+            labels,
+            language,
+            remove_accents=screen_mode,
+            replace_no_break_spaces=True,
+            **kwargs,
+        )
+
+        # Add other fields
+        for field in ["zachet", "nezachet", "comment", "source", "author"]:
+            if field in q:
+                if field == "source":
+                    if external_para is None:
+                        p = doc.add_paragraph()
+                        p.paragraph_format.keep_together = True
+                    else:
+                        p.add_run("\n")
+                else:
+                    p.add_run("\n")
+
+                field_label = get_label_standalone(q, field, labels, language)
+                p.add_run(f"{field_label}: ").bold = True
+                format_docx_element(
+                    doc,
+                    q[field],
+                    p,
+                    WHITEN[field],
+                    spoilers,
+                    logger,
+                    labels,
+                    language,
+                    remove_accents=screen_mode,
+                    remove_brackets=screen_mode,
+                    replace_no_break_spaces=field != "source",
+                    **kwargs,
+                )
+
+    return qcount
+
+
 class DocxExporter(BaseExporter):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
@@ -76,230 +507,53 @@ class DocxExporter(BaseExporter):
 
     def _docx_format(self, *args, **kwargs):
         kwargs.update(self.dir_kwargs)
-        return self.docx_format(*args, **kwargs)
+        return format_docx_element(
+            self.doc,
+            *args,
+            spoilers=self.args.spoilers,
+            logger=self.logger,
+            labels=self.labels,
+            language=self.args.language,
+            **kwargs,
+        )
 
     def docx_format(self, el, para, whiten, **kwargs):
-        if isinstance(el, list):
-            if len(el) > 1 and isinstance(el[1], list):
-                self.docx_format(el[0], para, whiten, **kwargs)
-                licount = 0
-                for li in el[1]:
-                    licount += 1
-
-                    para.add_run("\n{}. ".format(licount))
-                    self.docx_format(li, para, whiten, **kwargs)
-            else:
-                licount = 0
-                for li in el:
-                    licount += 1
-
-                    para.add_run("\n{}. ".format(licount))
-                    self.docx_format(li, para, whiten, **kwargs)
-
-        if isinstance(el, str):
-            self.logger.debug("parsing element {}:".format(log_wrap(el)))
-
-            if kwargs.get("remove_accents"):
-                el = el.replace("\u0301", "")
-            if kwargs.get("remove_brackets"):
-                el = self.remove_square_brackets(el)
-            else:
-                el = replace_escaped(el)
-
-            el = backtick_replace(el)
-
-            for run in self.parse_4s_elem(el):
-                if run[0] == "pagebreak":
-                    if self.args.spoilers == "dots":
-                        for _ in range(30):
-                            para = self.doc.add_paragraph()
-                            para.add_run(".")
-                        para = self.doc.add_paragraph()
-                    else:
-                        para = self.doc.add_page_break()
-                elif run[0] == "linebreak":
-                    para.add_run("\n")
-                elif run[0] == "screen":
-                    if kwargs.get("remove_accents") or kwargs.get("remove_brackets"):
-                        text = run[1]["for_screen"]
-                    else:
-                        text = run[1]["for_print"]
-                    if kwargs.get("replace_no_break_spaces"):
-                        text = self._replace_no_break(text)
-                    r = para.add_run(text)
-                elif run[0] == "hyperlink" and not (
-                    whiten and self.args.spoilers == "whiten"
-                ):
-                    r = self.add_hyperlink(para, run[1], run[1])
-                elif run[0] == "img":
-                    if run[1].endswith(".shtml"):
-                        r = para.add_run(
-                            "(ТУТ БЫЛА ССЫЛКА НА ПРОТУХШУЮ КАРТИНКУ)\n"
-                        )  # TODO: добавить возможность пропускать кривые картинки опцией
-                        continue
-                    parsed_image = parseimg(
-                        run[1],
-                        dimensions="inches",
-                        tmp_dir=kwargs.get("tmp_dir"),
-                        targetdir=kwargs.get("targetdir"),
-                    )
-                    imgfile = parsed_image["imgfile"]
-                    width = parsed_image["width"]
-                    height = parsed_image["height"]
-                    inline = parsed_image["inline"]
-                    if inline:
-                        r = para.add_run("")
-                    else:
-                        r = para.add_run("\n")
-
-                    try:
-                        if inline:
-                            r.add_picture(
-                                imgfile,
-                                height=Inches(
-                                    1.0 / 6
-                                ),  # Height is based on docx template
-                            )
-                        else:
-                            r.add_picture(
-                                imgfile, width=Inches(width), height=Inches(height)
-                            )
-                    except UnrecognizedImageError:
-                        sys.stderr.write(
-                            f"python-docx can't recognize header for {imgfile}\n"
-                        )
-                    if not inline:
-                        r = para.add_run("\n")
-                    continue
-                else:
-                    text = run[1]
-                    if kwargs.get("replace_no_break_spaces"):
-                        text = self._replace_no_break(text)
-                    r = para.add_run(text)
-                    if "italic" in run[0]:
-                        r.italic = True
-                    if "bold" in run[0]:
-                        r.bold = True
-                    if "underline" in run[0]:
-                        r.underline = True
-                    if run[0] == "strike":
-                        r.font.strike = True
-                    if run[0] == "sc":
-                        r.small_caps = True
-                    if whiten and self.args.spoilers == "whiten":
-                        r.style = "Whitened"
+        # Redirect to standalone function
+        return format_docx_element(
+            self.doc,
+            el,
+            para,
+            whiten,
+            spoilers=self.args.spoilers,
+            logger=self.logger,
+            labels=self.labels,
+            language=self.args.language,
+            **kwargs,
+        )
 
     def add_hyperlink(self, paragraph, text, url):
-        # adapted from https://github.com/python-openxml/python-docx/issues/610
-        doc = self.doc
-        run = paragraph.add_run(text)
-        run.style = doc.styles["Hyperlink"]
-        part = paragraph.part
-        r_id = part.relate_to(
-            url, docx.opc.constants.RELATIONSHIP_TYPE.HYPERLINK, is_external=True
-        )
-        hyperlink = docx.oxml.shared.OxmlElement("w:hyperlink")
-        hyperlink.set(docx.oxml.shared.qn("r:id"), r_id)
-        hyperlink.append(run._r)
-        paragraph._p.append(hyperlink)
-        return hyperlink
+        return add_hyperlink_to_docx(self.doc, paragraph, text, url)
 
     def add_question(
         self, element, skip_qcount=False, screen_mode=False, external_para=None
     ):
-        q = element[1]
-        if external_para is None:
-            p = self.doc.add_paragraph()
-        else:
-            p = external_para
-        p.paragraph_format.space_before = DocxPt(18)
-        p.paragraph_format.keep_together = True
-        if "number" not in q and not skip_qcount:
-            self.qcount += 1
-        if "setcounter" in q:
-            self.qcount = int(q["setcounter"])
-        p.add_run(
-            "{question}. ".format(
-                question=self.get_label(
-                    q,
-                    "question",
-                    number=self.qcount if "number" not in q else q["number"],
-                )
-            )
-        ).bold = True
-
-        if "handout" in q:
-            p.add_run("\n[{handout}: ".format(handout=self.get_label(q, "handout")))
-            self._docx_format(
-                q["handout"],
-                p,
-                WHITEN["handout"],
-                remove_accents=screen_mode,
-                remove_brackets=screen_mode,
-            )
-            p.add_run("\n]")
-        if not self.args.noparagraph:
-            p.add_run("\n")
-
-        self._docx_format(
-            q["question"],
-            p,
-            False,
-            remove_accents=screen_mode,
-            remove_brackets=screen_mode,
-            replace_no_break_spaces=True,
+        self.qcount = add_question_to_docx(
+            self.doc,
+            element[1],
+            self.labels,
+            self.qcount,
+            skip_qcount,
+            screen_mode,
+            external_para,
+            self.args.noparagraph,
+            self.args.noanswers,
+            self.args.spoilers,
+            self.args.language,
+            self.args.only_question_number,
+            self.logger,
+            **self.dir_kwargs,
         )
 
-        if not self.args.noanswers:
-            if self.args.spoilers == "pagebreak":
-                p = self.doc.add_page_break()
-            elif self.args.spoilers == "dots":
-                for _ in range(30):
-                    if external_para is None:
-                        p = self.doc.add_paragraph()
-                    else:
-                        p.add_run("\n")
-                    p.add_run(".")
-                if external_para is None:
-                    p = self.doc.add_paragraph()
-                else:
-                    p.add_run("\n")
-            else:
-                if external_para is None:
-                    p = self.doc.add_paragraph()
-                else:
-                    p.add_run("\n")
-            p.paragraph_format.keep_together = True
-            p.paragraph_format.space_before = DocxPt(6)
-            p.add_run(f"{self.get_label(q, 'answer')}: ").bold = True
-            self._docx_format(
-                q["answer"],
-                p,
-                True,
-                remove_accents=screen_mode,
-                replace_no_break_spaces=True,
-            )
-
-            for field in ["zachet", "nezachet", "comment", "source", "author"]:
-                if field in q:
-                    if field == "source":
-                        if external_para is None:
-                            p = self.doc.add_paragraph()
-                            p.paragraph_format.keep_together = True
-                        else:
-                            p.add_run("\n")
-                    else:
-                        p.add_run("\n")
-                    p.add_run(f"{self.get_label(q, field)}: ").bold = True
-                    self._docx_format(
-                        q[field],
-                        p,
-                        WHITEN[field],
-                        remove_accents=screen_mode,
-                        remove_brackets=screen_mode,
-                        replace_no_break_spaces=field != "source",
-                    )
-
     def _add_question_columns(self, element):
         table = self.doc.add_table(rows=1, cols=2)
         table.autofit = True
@@ -332,82 +586,6 @@ class DocxExporter(BaseExporter):
 
         self.doc.add_paragraph()
 
-    def _add_question_content(self, q, p, skip_qcount=False, screen_mode=False):
-        """Helper method to add question content to a paragraph"""
-        if "number" not in q and not skip_qcount:
-            self.qcount += 1
-        if "setcounter" in q:
-            self.qcount = int(q["setcounter"])
-        p.add_run(
-            "{question}. ".format(
-                question=self.get_label(
-                    q,
-                    "question",
-                    number=self.qcount if "number" not in q else q["number"],
-                )
-            )
-        ).bold = True
-
-        if "handout" in q:
-            p.add_run("\n[{handout}: ".format(handout=self.get_label(q, "handout")))
-            self._docx_format(
-                q["handout"],
-                p,
-                WHITEN["handout"],
-                remove_accents=screen_mode,
-                remove_brackets=screen_mode,
-            )
-            p.add_run("\n]")
-        if not self.args.noparagraph:
-            p.add_run("\n")
-
-        self._docx_format(
-            q["question"],
-            p,
-            False,
-            remove_accents=screen_mode,
-            remove_brackets=screen_mode,
-            replace_no_break_spaces=True,
-        )
-
-        if not self.args.noanswers:
-            if self.args.spoilers == "pagebreak":
-                p = self.doc.add_page_break()
-            elif self.args.spoilers == "dots":
-                for _ in range(30):
-                    p = self.doc.add_paragraph()
-                    p.add_run(".")
-                p = self.doc.add_paragraph()
-            else:
-                p = self.doc.add_paragraph()
-            p.paragraph_format.keep_together = True
-            p.paragraph_format.space_before = DocxPt(6)
-            p.add_run(f"{self.get_label(q, 'answer')}: ").bold = True
-            self._docx_format(
-                q["answer"],
-                p,
-                True,
-                remove_accents=screen_mode,
-                replace_no_break_spaces=True,
-            )
-
-            for field in ["zachet", "nezachet", "comment", "source", "author"]:
-                if field in q:
-                    if field == "source":
-                        p = self.doc.add_paragraph()
-                        p.paragraph_format.keep_together = True
-                    else:
-                        p.add_run("\n")
-                    p.add_run(f"{self.get_label(q, field)}: ").bold = True
-                    self._docx_format(
-                        q[field],
-                        p,
-                        WHITEN[field],
-                        remove_accents=screen_mode,
-                        remove_brackets=screen_mode,
-                        replace_no_break_spaces=field != "source",
-                    )
-
     def export(self, outfilename):
         self.logger.debug(self.args.docx_template)
         self.doc = Document(self.args.docx_template)
@@ -473,3 +651,53 @@ class DocxExporter(BaseExporter):
 
         self.doc.save(outfilename)
         self.logger.info("Output: {}".format(outfilename))
+
+
+# Example usage of the extracted DOCX functions:
+"""
+from docx import Document
+import toml
+from chgksuite.composer.docx import add_question_to_docx, format_docx_element
+
+# Load labels
+with open("labels.toml", encoding="utf8") as f:
+    labels = toml.load(f)
+
+# Create a new document
+doc = Document()
+
+# Example question data
+question_data = {
+    "question": "What is the capital of France?",
+    "answer": "Paris",
+    "comment": "This is a basic geography question",
+    "source": "World Geography Book"
+}
+
+# Add question to document
+qcount = add_question_to_docx(
+    doc=doc,
+    question_data=question_data,
+    labels=labels,
+    qcount=0,  # Starting question count
+    noanswers=False,  # Include answers
+    spoilers="none",  # No spoiler handling
+    language="en",
+    only_question_number=False
+)
+
+# Or use the lower-level formatting function directly
+paragraph = doc.add_paragraph()
+format_docx_element(
+    doc=doc,
+    el="This is **bold text** and _italic text_",
+    para=paragraph,
+    whiten=False,
+    spoilers="none",
+    labels=labels,
+    language="en"
+)
+
+# Save the document
+doc.save("example_output.docx")
+"""