Implement flag hashing in Challenge model and remove duplicate port field

Signed-off-by: Soumyaditya Batabyal <soumyaditya2021@gmail.com>

Author: Soumyaditya25

challenge/management/commands/custom_command.py

@@ -0,0 +1,40 @@
+import json
+import os
+from django.core.management.base import BaseCommand
+from challenge.models import Challenge
+
+class Command(BaseCommand):
+    help = "Populate challenge data from challenge/challenge.json"
+
+    def handle(self, *args, **options):
+        file_path = os.path.join('challenge', 'challenge.json')
+
+        try:
+            with open(file_path, 'r') as json_file:
+                challenges_data = json.load(json_file)
+        except FileNotFoundError:
+            self.stderr.write(self.style.ERROR(f"JSON file not found: {file_path}"))
+            return
+        except json.JSONDecodeError as e:
+            self.stderr.write(self.style.ERROR(f"Error decoding JSON: {e}"))
+            return
+
+        for item in challenges_data:
+            challenge, created = Challenge.objects.get_or_create(
+                name=item.get("name"),
+                defaults={
+                    "description": item.get("description", ""),
+                    "docker_image": item.get("docker_image", ""),
+                    "docker_port": item.get("docker_port", 0),
+                    "start_port": item.get("start_port", 0),
+                    "end_port": item.get("end_port", 0),
+                    "flag": item.get("flag", ""),
+                    "point": item.get("point", 0)
+                }
+            )
+            if created:
+                self.stdout.write(self.style.SUCCESS(f"Challenge '{challenge.name}' created."))
+            else:
+                self.stdout.write(f"Challenge '{challenge.name}' already exists.")
+
+        self.stdout.write(self.style.SUCCESS("Challenge data has been populated successfully."))

challenge/models.py

@@ -1,7 +1,8 @@
 from django.db import models
 from django.contrib.auth.models import User
+import hashlib # Import hashlib
+from django.core.exceptions import ValidationError  # Import ValidationError
 
-# Create your models here
 class Challenge(models.Model):
     id = models.AutoField(primary_key=True, unique=True)
     name = models.CharField(max_length=100, unique=True)
@@ -18,11 +19,13 @@ class Challenge(models.Model):
     def __str__(self):
         return self.name
 
-    # overwriting default save  method
+    # Overriding default save method
     def save(self, *args, **kwargs):
         if self.start_port > self.end_port:
-            raise Exception("Start port should be less than end port")
-        # Here flag need to be hashed
+            raise ValidationError("Start port should be less than end port") # Raise ValidationError if start_port is greater than end_port
+        if self.flag:
+            if not self.flag.startswith("hashed_"):
+                self.flag = "hashed_" + hashlib.sha256(self.flag.encode('utf-8')).hexdigest()
         super(Challenge, self).save(*args, **kwargs)
 
 class UserChallenge(models.Model):
@@ -40,7 +43,6 @@ class UserChallenge(models.Model):
     is_live = models.BooleanField(default=False)
     no_of_attempt = models.IntegerField(default=0)
     is_solved = models.BooleanField(default=False)
-    port = models.IntegerField()
 
     def __str__(self):
-        return f"{self.user.username} - {self.challenge.name}"
+        return f"{self.user.username} - {self.challenge.name}"