219 lines
6.9 KiB
Python
Executable file
219 lines
6.9 KiB
Python
Executable file
import hashlib
|
|
import logging
|
|
import re
|
|
|
|
import requests
|
|
from django.conf import settings
|
|
from django.core.cache import caches
|
|
from django.utils.html import format_html
|
|
from django.utils.safestring import mark_safe
|
|
from mistune import escape
|
|
|
|
from judge.utils.file_cache import HashFileCache
|
|
from judge.utils.unicode import utf8bytes, utf8text
|
|
|
|
logger = logging.getLogger("judge.mathoid")
|
|
reescape = re.compile(r"(?<!\\)(?:\\{2})*[$]")
|
|
|
|
REPLACES = [
|
|
("\u2264", r"\le"),
|
|
("\u2265", r"\ge"),
|
|
("\u2026", "..."),
|
|
("\u2212", "-"),
|
|
("≤", r"\le"),
|
|
("≥", r"\ge"),
|
|
("<", "<"),
|
|
(">", ">"),
|
|
("&", "&"),
|
|
("−", "-"),
|
|
("≤", r"\le"),
|
|
("≥", r"\ge"),
|
|
("…", "..."),
|
|
(r"\lt", "<"),
|
|
(r"\gt", ">"),
|
|
]
|
|
|
|
|
|
def format_math(math):
|
|
for a, b in REPLACES:
|
|
math = math.replace(a, b)
|
|
return math
|
|
|
|
|
|
class MathoidMathParser(object):
|
|
types = ("svg", "mml", "tex", "jax")
|
|
|
|
def __init__(self, type):
|
|
self.type = type
|
|
|
|
self.mathoid_url = settings.MATHOID_URL
|
|
self.cache = HashFileCache(
|
|
settings.MATHOID_CACHE_ROOT,
|
|
settings.MATHOID_CACHE_URL,
|
|
settings.MATHOID_GZIP,
|
|
)
|
|
|
|
mml_cache = settings.MATHOID_MML_CACHE
|
|
self.mml_cache = mml_cache and caches[mml_cache]
|
|
self.css_cache = caches[settings.MATHOID_CSS_CACHE]
|
|
|
|
self.mml_cache_ttl = settings.MATHOID_MML_CACHE_TTL
|
|
|
|
def query_mathoid(self, formula, hash):
|
|
self.cache.create(hash)
|
|
|
|
try:
|
|
response = requests.post(
|
|
self.mathoid_url,
|
|
data={
|
|
"q": reescape.sub(lambda m: "\\" + m.group(0), formula).encode(
|
|
"utf-8"
|
|
),
|
|
"type": "tex"
|
|
if formula.startswith(r"\displaystyle")
|
|
else "inline-tex",
|
|
},
|
|
)
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
except requests.ConnectionError:
|
|
logger.exception("Failed to connect to mathoid for: %s", formula)
|
|
return
|
|
except requests.HTTPError as e:
|
|
logger.error("Mathoid failed to render: %s\n%s", formula, e.response.text)
|
|
return
|
|
except Exception:
|
|
logger.exception("Failed to connect to mathoid for: %s", formula)
|
|
return
|
|
|
|
if not data["success"]:
|
|
logger.error("Mathoid failure for: %s\n%s", formula, data)
|
|
return
|
|
|
|
if any(i not in data for i in ("mml", "png", "svg", "mathoidStyle")):
|
|
logger.error(
|
|
"Mathoid did not return required information (mml, png, svg, mathoidStyle needed):\n%s",
|
|
data,
|
|
)
|
|
return
|
|
|
|
css = data["mathoidStyle"]
|
|
mml = data["mml"]
|
|
result = {
|
|
"css": css,
|
|
"mml": mml,
|
|
"png": self.cache.cache_data(hash, "png", bytearray(data["png"]["data"])),
|
|
"svg": self.cache.cache_data(hash, "svg", data["svg"].encode("utf-8")),
|
|
}
|
|
self.cache.cache_data(hash, "mml", mml.encode("utf-8"), url=False, gzip=False)
|
|
self.cache.cache_data(hash, "css", css.encode("utf-8"), url=False, gzip=False)
|
|
return result
|
|
|
|
def query_cache(self, hash):
|
|
result = {
|
|
"svg": self.cache.get_url(hash, "svg"),
|
|
"png": self.cache.get_url(hash, "png"),
|
|
}
|
|
|
|
key = "mathoid:css:" + hash
|
|
css = result["css"] = self.css_cache.get(key)
|
|
if css is None:
|
|
css = result["css"] = self.cache.read_data(hash, "css").decode("utf-8")
|
|
self.css_cache.set(key, css, self.mml_cache_ttl)
|
|
|
|
mml = None
|
|
if self.mml_cache:
|
|
mml = result["mml"] = self.mml_cache.get("mathoid:mml:" + hash)
|
|
if mml is None:
|
|
mml = result["mml"] = self.cache.read_data(hash, "mml").decode("utf-8")
|
|
if self.mml_cache:
|
|
self.mml_cache.set("mathoid:mml:" + hash, mml, self.mml_cache_ttl)
|
|
return result
|
|
|
|
def get_result(self, formula):
|
|
if self.type == "tex":
|
|
return
|
|
|
|
hash = hashlib.sha1(utf8bytes(formula)).hexdigest()
|
|
formula = utf8text(formula)
|
|
if self.cache.has_file(hash, "css"):
|
|
result = self.query_cache(hash)
|
|
else:
|
|
result = self.query_mathoid(formula, hash)
|
|
|
|
if not result:
|
|
return None
|
|
|
|
result["tex"] = formula
|
|
result["display"] = formula.startswith(r"\displaystyle")
|
|
return {
|
|
"mml": self.output_mml,
|
|
"msp": self.output_msp,
|
|
"svg": self.output_svg,
|
|
"jax": self.output_jax,
|
|
"png": self.output_png,
|
|
"raw": lambda x: x,
|
|
}[self.type](result)
|
|
|
|
def output_mml(self, result):
|
|
return result["mml"]
|
|
|
|
def output_msp(self, result):
|
|
# 100% MediaWiki compatibility.
|
|
return format_html(
|
|
'<span class="{5}-math">'
|
|
'<span class="mwe-math-mathml-{5} mwe-math-mathml-a11y"'
|
|
' style="display: none;">{0}</span>'
|
|
'<img src="{1}" class="mwe-math-fallback-image-{5}"'
|
|
" onerror=\"this.src='{2}';this.onerror=null\""
|
|
' aria-hidden="true" style="{3}" alt="{4}"></span>',
|
|
mark_safe(result["mml"]),
|
|
result["svg"],
|
|
result["png"],
|
|
result["css"],
|
|
result["tex"],
|
|
["inline", "display"][result["display"]],
|
|
)
|
|
|
|
def output_jax(self, result):
|
|
return format_html(
|
|
'<span class="{4}">'
|
|
'''<img class="tex-image" src="{0}" style="{2}" alt="{3}"'''
|
|
""" onerror="this.src='{1}';this.onerror=null">"""
|
|
"""<span class="tex-text" style="display:none">{5}{3}{5}</span>"""
|
|
"</span>",
|
|
result["svg"],
|
|
result["png"],
|
|
result["css"],
|
|
result["tex"],
|
|
["inline-math", "display-math"][result["display"]],
|
|
["~", "$$"][result["display"]],
|
|
)
|
|
|
|
def output_svg(self, result):
|
|
return format_html(
|
|
'<img class="{4}" src="{0}" style="{2}" alt="{3}" '
|
|
"""onerror="this.src='{1}';this.onerror=null">""",
|
|
result["svg"],
|
|
result["png"],
|
|
result["css"],
|
|
result["tex"],
|
|
["inline-math", "display-math"][result["display"]],
|
|
)
|
|
|
|
def output_png(self, result):
|
|
return format_html(
|
|
'<img class="{3}" src="{0}" style="{1}" alt="{2}">',
|
|
result["png"],
|
|
result["css"],
|
|
result["tex"],
|
|
["inline-math", "display-math"][result["display"]],
|
|
)
|
|
|
|
def display_math(self, math):
|
|
math = format_math(math)
|
|
return self.get_result(r"\displaystyle " + math) or r"\[%s\]" % escape(math)
|
|
|
|
def inline_math(self, math):
|
|
math = format_math(math)
|
|
return self.get_result(math) or r"\(%s\)" % escape(math)
|