[Backend] Add custom EC2 storage to challenges (#4103)

Cloud-CV · Aug 10, 2023 · 35dc987 · 35dc987
1 parent d9dd84c
commit 35dc987
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/apps/challenges/aws_utils.py b/apps/challenges/aws_utils.py
@@ -1513,7 +1513,7 @@ def setup_ec2(challenge):
             'DeviceName': '/dev/sda1',
             'Ebs': {
                 'DeleteOnTermination': True,
-                'VolumeSize': 8,  # TODO: Make this customizable
+                'VolumeSize': challenge_obj.ec2_storage,  # TODO: Make this customizable
                 'VolumeType': 'gp2'
             }
         },

diff --git a/apps/challenges/models.py b/apps/challenges/models.py
@@ -91,6 +91,9 @@ def __init__(self, *args, **kwargs):
     ec2_instance_id = models.CharField(
         max_length=200, default="", null=True, blank=True
     )
+    ec2_storage = models.PositiveIntegerField(
+        default=8, verbose_name="EC2 storage (GB)"
+    )
     featured = models.BooleanField(
         default=False, verbose_name="Featured", db_index=True
     )

diff --git a/apps/challenges/serializers.py b/apps/challenges/serializers.py
@@ -84,6 +84,7 @@ class Meta:
             "cpu_only_jobs",
             "job_cpu_cores",
             "job_memory",
+            "ec2_storage",
         )
 
 
@@ -295,6 +296,7 @@ class Meta:
             "cpu_only_jobs",
             "job_cpu_cores",
             "job_memory",
+            "ec2_storage",
         )