Parsons converted to PTX

Author: bnmnetp

runestone/parsons/parsons.py

@@ -14,7 +14,9 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 #
 __author__ = "isaiahmayerchak"
-
+from textwrap import dedent
+import pdb
+import random
 from docutils import nodes
 from docutils.parsers.rst import directives
 from runestone.mchoice import Assessment
@@ -53,13 +55,87 @@ class ParsonsNode(nodes.General, nodes.Element, RunestoneIdNode):
     pass
 
 
+def wrap_with_tag(text: str, tag: str, **kwargs) -> str:
+    start = f"<{tag}"
+    for k, v in kwargs.items():
+        start += f" {k}='{v}'"
+    start += ">"
+    end = f"</{tag}>\n"
+    return start + text + end
+
+
 def visit_parsons_xml(self, node):
-    res = TEMPLATE_START % node["runestone_options"]
+    if node["runestone_options"]["numbered"]:
+        node["runestone_options"]["numbered"] = "numbered='yes'"
+
+    if node["runestone_options"]["adaptive"]:
+        node["runestone_options"]["adaptive"] = "adaptive='yes'"
+
+    if not node["runestone_options"]["noindent"]:
+        node["runestone_options"]["noindent"] = "indent='yes'"
+
+    if node["runestone_options"]["language"]:
+        node["runestone_options"]["language"] = node["runestone_options"]["language"].replace(
+            "data-", "")
+    else:
+        node["runestone_options"]["language"] = "language='python'"
+
+    res = "<exercise xml:id='{divid}' {numbered} {adaptive} {noindent} {language}>".format(
+        **node["runestone_options"])
+    res += "<statement>\n"
     self.output.append(res)
 
 
+def process_one_block(block, choice=False, correct=False, natural=False):
+    code_list = block.split("\n")
+    if choice:
+        attrs = "correct='yes'" if correct else ""
+        res = f"<choice {attrs}>"
+    else:
+        res = ""
+    for line in code_list:
+        if line:
+            line = line.replace("#paired", "")
+            line = line.replace("#distractor", "")
+            if natural:
+                res += wrap_with_tag(line, "p")
+            else:
+                res += wrap_with_tag(line, "cline")
+    if choice:
+        res += "</choice>"
+    return res
+
+
 def depart_parsons_xml(self, node):
-    res = TEMPLATE_END % node["runestone_options"]
+    res = "</statement>\n<blocks>"
+    if "natural" in node["runestone_options"]["language"]:
+        natural = True
+    else:
+        natural = False
+    # first split into blocks based on --- then process each block...
+    block_list = dedent(node["runestone_options"]["code"]).split("---")
+    order = list(range(1, len(block_list) + 1))
+    random.shuffle(order)
+    block = block_list.pop(0)
+    while block:
+        attr = f"order='{order.pop()}' "
+        attr += "correct='no'" if "#distractor" in block else ""
+        res += f"<block {attr}>\n"
+        # check to see if the next block is a distractor
+        if block_list and "#paired" in block_list[0]:
+            choice = True
+            res += process_one_block(block, choice, True, natural)
+            res += process_one_block(block_list.pop(0), choice, False, natural)
+        else:
+            choice = False
+            res += process_one_block(block, choice, True, natural)
+        res += "</block>"
+        if block_list:
+            block = block_list.pop(0)
+        else:
+            block = None
+
+    res += "</blocks></exercise>"
     self.output.append(res)