[Backend] Add EC2 instance ID to challenge (#4093)

apps/challenges/aws_utils.py

@@ -1538,6 +1538,8 @@ def setup_ec2(challenge):
             ],
             UserData=ec2_worker_script,
         )
+        challenge_obj.ec2_instance_id = response['Instances'][0]['InstanceId']
+        challenge_obj.save()
         return response
     except ClientError as e:
         logger.exception(e)

apps/challenges/migrations/0101_challenge_ec2_instance_id.py

@@ -0,0 +1,23 @@
+# Generated by Django 2.2.20 on 2023-08-10 06:02
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('challenges', '0100_challenge_uses_ec2_worker'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='challenge',
+            name='ec2_instance_id',
+            field=models.CharField(blank=True, default='', max_length=200, null=True),
+        ),
+        migrations.AlterField(
+            model_name='challenge',
+            name='uses_ec2_worker',
+            field=models.BooleanField(db_index=True, default=False, verbose_name='Uses EC2 worker instance'),
+        ),
+    ]

apps/challenges/models.py

@@ -86,7 +86,10 @@ def __init__(self, *args, **kwargs):
         default=False, verbose_name="Approved By Admin", db_index=True
     )
     uses_ec2_worker = models.BooleanField(
-        default=False, verbose_name="Uses separate EC2 worker instance", db_index=True
+        default=False, verbose_name="Uses EC2 worker instance", db_index=True
+    )
+    ec2_instance_id = models.CharField(
+        max_length=200, default="", null=True, blank=True
     )
     featured = models.BooleanField(
         default=False, verbose_name="Featured", db_index=True