Move ANSI HTML-ifying to site; DMOJ/judge-server#478

judge/jinja2/reference.py

@@ -2,6 +2,7 @@
 from collections import defaultdict
 from urllib.parse import urljoin
 
+from ansi2html import Ansi2HTMLConverter
 from django.contrib.auth.models import AbstractUser
 from django.urls import reverse
 from lxml.html import Element
@@ -178,3 +179,8 @@ def join(first, second, *rest):
     if not rest:
         return urljoin(first, second)
     return urljoin(urljoin(first, second), *rest)
+
+
+@registry.filter(name='ansi2html')
+def ansi2html(s):
+    return Ansi2HTMLConverter(inline=True).convert(s, full=False)

judge/migrations/0091_compiler_message_ansi2html.py

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+import lxml.html as lh
+from django.db import migrations
+from lxml.html.clean import clean_html
+
+
+def strip_error_html(apps, schema_editor):
+    Submission = apps.get_model('judge', 'Submission')
+    for sub in Submission.objects.filter(error__isnull=False).iterator():
+        if sub.error:
+            sub.error = clean_html(lh.fromstring(sub.error)).text_content()
+            sub.save(update_fields=['error'])
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('judge', '0090_fix_contest_visibility'),
+    ]
+
+    operations = [
+        migrations.RunPython(strip_error_html, migrations.RunPython.noop, atomic=True),
+    ]

requirements.txt

@@ -28,3 +28,4 @@ jsonfield
 pymoss
 packaging
 celery
+ansi2html

templates/submission/status-testcases.html

@@ -12,11 +12,11 @@ <h4>{{ _('We are waiting for a suitable judge to process your submission...') }}
         <h4>{{ _('Your submission is being processed...') }}</h4>
     {% elif submission.status == 'CE' %}
         <h3>{{ _('Compilation Error') }}</h3>
-        <pre>{{ submission.error|safe }}</pre>
+        <pre>{{ submission.error|ansi2html }}</pre>
     {% else %}
         {% if submission.error %}
             <h3>{{ _('Compilation Warnings') }}</h3>
-            <pre>{{ submission.error|safe }}</pre>
+            <pre>{{ submission.error|ansi2html }}</pre>
             <hr class="half-hr"><br>
         {% endif %}
         {% if is_pretest %}