Add language template

judge/admin/problem.py

@@ -9,12 +9,15 @@ from django.forms import ModelForm
 from django.urls import reverse_lazy
 from django.utils.html import format_html
 from django.utils.translation import gettext, gettext_lazy as _, ungettext
+from django_ace import AceWidget
+
 from reversion.admin import VersionAdmin
 from reversion_compare.admin import CompareVersionAdmin
 
 
 from judge.models import (
     LanguageLimit,
+    LanguageTemplate,
     Problem,
     ProblemClarification,
     ProblemTranslation,
@@ -100,6 +103,20 @@ class LanguageLimitInline(admin.TabularInline):
     form = LanguageLimitInlineForm
 
 
+class LanguageTemplateInlineForm(ModelForm):
+    class Meta:
+        widgets = {
+            "language": AdminSelect2Widget,
+            "source": AceWidget(width="600px", height="200px", toolbar=False),
+        }
+
+
+class LanguageTemplateInline(admin.TabularInline):
+    model = LanguageTemplate
+    fields = ("language", "source")
+    form = LanguageTemplateInlineForm
+
+
 class ProblemClarificationForm(ModelForm):
     class Meta:
         if HeavyPreviewPageDownWidget is not None:
@@ -208,6 +225,7 @@ class ProblemAdmin(CompareVersionAdmin):
     )
     inlines = [
         LanguageLimitInline,
+        LanguageTemplateInline,
         ProblemClarificationInline,
         ProblemSolutionInline,
         ProblemTranslationInline,

judge/migrations/0126_languagetemplate.py

@@ -0,0 +1,54 @@
+# Generated by Django 2.2.25 on 2022-06-12 06:59
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("judge", "0125_auto_20220602_1216"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="LanguageTemplate",
+            fields=[
+                (
+                    "id",
+                    models.AutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                (
+                    "source",
+                    models.TextField(max_length=65536, verbose_name="source code"),
+                ),
+                (
+                    "language",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        to="judge.Language",
+                        verbose_name="language",
+                    ),
+                ),
+                (
+                    "problem",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="language_templates",
+                        to="judge.Problem",
+                        verbose_name="problem",
+                    ),
+                ),
+            ],
+            options={
+                "verbose_name": "language-specific template",
+                "verbose_name_plural": "language-specific templates",
+                "unique_together": {("problem", "language")},
+            },
+        ),
+    ]

judge/models/__init__.py

@@ -20,6 +20,7 @@ from judge.models.interface import BlogPost, MiscConfig, NavigationBar, validate
 from judge.models.message import PrivateMessage, PrivateMessageThread
 from judge.models.problem import (
     LanguageLimit,
+    LanguageTemplate,
     License,
     Problem,
     ProblemClarification,

judge/models/problem.py

@@ -639,6 +639,24 @@ class LanguageLimit(models.Model):
         verbose_name_plural = _("language-specific resource limits")
 
 
+class LanguageTemplate(models.Model):
+    problem = models.ForeignKey(
+        Problem,
+        verbose_name=_("problem"),
+        related_name="language_templates",
+        on_delete=CASCADE,
+    )
+    language = models.ForeignKey(
+        Language, verbose_name=_("language"), on_delete=CASCADE
+    )
+    source = models.TextField(verbose_name=_("source code"), max_length=65536)
+
+    class Meta:
+        unique_together = ("problem", "language")
+        verbose_name = _("language-specific template")
+        verbose_name_plural = _("language-specific templates")
+
+
 class Solution(models.Model):
     problem = models.OneToOneField(
         Problem,

judge/views/problem.py

@@ -53,6 +53,7 @@ from judge.models import (
     Organization,
     VolunteerProblemVote,
     Profile,
+    LanguageTemplate,
 )
 from judge.pdf_problems import DefaultPdfMaker, HAS_PDF
 from judge.utils.diggpaginator import DiggPaginator
@@ -895,10 +896,21 @@ class ProblemFeed(ProblemList):
 class LanguageTemplateAjax(View):
     def get(self, request, *args, **kwargs):
         try:
-            language = get_object_or_404(Language, id=int(request.GET.get("id", 0)))
+            problem = request.GET.get("problem", None)
+            lang_id = int(request.GET.get("id", 0))
+            res = None
+            if problem:
+                try:
+                    res = LanguageTemplate.objects.get(
+                        language__id=lang_id, problem__id=problem
+                    ).source
+                except ObjectDoesNotExist:
+                    pass
+            if not res:
+                res = get_object_or_404(Language, id=lang_id).template
         except ValueError:
             raise Http404()
-        return HttpResponse(language.template, content_type="text/plain")
+        return HttpResponse(res, content_type="text/plain")
 
 
 class RandomProblem(ProblemList):
@@ -1107,6 +1119,7 @@ def problem_submit(request, problem, submission=None):
             "submissions_left": submissions_left,
             "ACE_URL": settings.ACE_URL,
             "default_lang": default_lang,
+            "problem_id": problem.id,
         },
     )