From 43f0b7818fa105c18a6f43a851a5c9429f9481ee Mon Sep 17 00:00:00 2001
From: yuetsh <517252939@qq.com>
Date: Sat, 4 Oct 2025 00:32:25 +0800
Subject: [PATCH] add problem prompt

---
 problem/migrations/0002_problem_prompt.py | 18 ++++++++++++++++++
 problem/models.py                         |  1 +
 problem/serializers.py                    |  1 +
 3 files changed, 20 insertions(+)
 create mode 100644 problem/migrations/0002_problem_prompt.py

diff --git a/problem/migrations/0002_problem_prompt.py b/problem/migrations/0002_problem_prompt.py
new file mode 100644
index 0000000..46dcdb1
--- /dev/null
+++ b/problem/migrations/0002_problem_prompt.py
@@ -0,0 +1,18 @@
+# Generated by Django 5.2.3 on 2025-10-03 16:31
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('problem', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='problem',
+            name='prompt',
+            field=models.TextField(null=True),
+        ),
+    ]
diff --git a/problem/models.py b/problem/models.py
index 1859825..6d6aefa 100644
--- a/problem/models.py
+++ b/problem/models.py
@@ -79,6 +79,7 @@ class Problem(models.Model):
     difficulty = models.TextField()
     tags = models.ManyToManyField(ProblemTag)
     source = models.TextField(null=True)
+    prompt = models.TextField(null=True)
     # for OI mode
     total_score = models.IntegerField(default=0)
     submission_number = models.BigIntegerField(default=0)
diff --git a/problem/serializers.py b/problem/serializers.py
index 92aeba8..4747289 100644
--- a/problem/serializers.py
+++ b/problem/serializers.py
@@ -79,6 +79,7 @@ class CreateOrEditProblemSerializer(serializers.Serializer):
     )
     hint = serializers.CharField(allow_blank=True, allow_null=True)
     source = serializers.CharField(max_length=256, allow_blank=True, allow_null=True)
+    prompt = serializers.CharField(allow_blank=True, allow_null=True)
     share_submission = serializers.BooleanField()