Common

models.py

@@ -103,3 +103,31 @@ class WhaleContainer(db.Model):
     def __repr__(self):
         return "<WhaleContainer ID:{0} {1} {2} {3} {4}>".format(self.id, self.user_id, self.challenge_id,
                                                                 self.start_time, self.renew_count)
+
+class WhaleContainerLog(db.Model):
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    user_id = db.Column(None, db.ForeignKey("users.id"))
+    challenge_id = db.Column(None, db.ForeignKey("challenges.id"))
+    start_time = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
+    uuid = db.Column(db.String(256))
+    port = db.Column(db.Integer, nullable=True, default=0)
+    flag = db.Column(db.String(128), nullable=False)
+
+    # Relationships
+    user = db.relationship(
+        "Users", foreign_keys="WhaleContainerLog.user_id", lazy="select")
+    challenge = db.relationship(
+        "DynamicDockerChallenge", foreign_keys="WhaleContainerLog.challenge_id", lazy="select"
+    )
+
+    def __init__(self, container):
+        self.user_id = container.user_id
+        self.challenge_id = container.challenge_id
+        self.start_time = container.start_time
+        self.uuid = container.uuid
+        self.port = container.port
+        self.flag = container.flag
+
+    def __repr__(self):
+        return "<WhaleContainer ID:{0} {1} {2} {3} {4} {5}>".format(self.id, self.user_id, self.challenge_id,
+                                                                self.start_time, self.uuid, self.port)