try checker

receiver/challenges/Warmup.py

@@ -0,0 +1,41 @@
+from .Challenge import Challenge
+
+import io
+import requests
+import random
+import subprocess
+
+class Warmup(Challenge):
+    flag_location = 'flags/warmup.txt'
+    history_location = 'history/warmup.txt'
+
+    def distribute(self, flag):
+        try:
+            with open(self.flag_location, 'w') as f:
+                f.write(flag)
+            with open(self.history_location, 'a') as f:
+                f.write(flag + '\n')
+            self.logger.info(f'Flag {flag} written to {self.flag_location}')
+            return True
+        except Exception as e:
+            self.logger.error(f'Could not write flag to {self.flag_location}: {e}')
+            return False
+
+    def check(self):
+        try:
+            url = f'http://localhost:{self.port}/'
+            r = requests.get(url, timeout=5)
+            assert "Simple file viewing application" in r.text.lower(), 'Warmup app not available'
+
+            with open(self.flag_location, 'r') as f:
+                host_flag = f.read().strip()
+            container_flag = subprocess.run([
+                "docker", "exec", "warmup_container", "cat", "/flag.txt"
+            ], capture_output=True, text=True).stdout.strip()
+            assert host_flag == container_flag, 'Flag mismatch between host and container'
+
+            self.logger.info('Check passed for warmup')
+            return True
+        except Exception as e:
+            self.logger.error(f'Could not check warmup: {e}')
+            return False