dev

utils/control.py

@@ -22,6 +22,7 @@ class ControlUtil:
             DockerUtils.remove_container(container)
             DBContainer.remove_container_record(user_id)
             return False, msg
+        DBContainer.create_container_log(container)
         return True, 'Container created'
 
     @staticmethod

utils/db.py

@@ -2,7 +2,7 @@ import datetime
 
 from CTFd.models import db
 from CTFd.utils import get_config
-from ..models import WhaleContainer, WhaleRedirectTemplate
+from ..models import WhaleContainer,WhaleContainerLog, WhaleRedirectTemplate
 
 
 class DBContainer:
@@ -14,6 +14,12 @@ class DBContainer:
 
         return container
 
+    @staticmethod
+    def create_container_log(container):
+        log = WhaleContainerLog(container=container)
+        db.session.add(log)
+        db.session.commit()
+
     @staticmethod
     def get_current_containers(user_id):
         q = db.session.query(WhaleContainer)