class Settings(ReprMixin, db.Model):
__tablename__ = "settings"
__repr_fields__ = ("ctf_name", "organization_name")
id = db.Column(db.Integer, primary_key=True)
dummy = db.Column(db.Boolean, nullable=False, default=True)
ctf_name = db.Column(db.String(64), nullable=False, default="RootTheBox CTF")
organization_name = db.Column(
db.String(80), nullable=False, default="codeiiest-dev"
)
from_date = db.Column(db.Date, nullable=True, default=date.today())
from_time = db.Column(db.Time, nullable=True, default=time())
to_date = db.Column(
db.Date, nullable=False, default=date.today() + timedelta(days=2)
)
to_time = db.Column(db.Time, nullable=False, default=time())
@staticmethod
@cache.cached(timeout=3600 * 6, key_prefix="settings")
def get_settings():
return Settings.query.get(1)
@hybrid_property
def running_time_from(self):
return datetime.combine(self.from_date, self.from_time)
@hybrid_property
def running_time_to(self):
return datetime.combine(self.to_date, self.to_time)
class Machine(TimeMixin, ReprMixin, db.Model):
__tablename__ = "machine"
__repr_fields__ = (
"name",
"os",
)
id = db.Column(db.Integer, primary_key=True, index=True)
name = db.Column(db.String(64), nullable=False, unique=True)
user_hash = db.Column(db.String(32), nullable=False)
root_hash = db.Column(db.String(32), nullable=False)
user_points = db.Column(db.Integer, default=0)
root_points = db.Column(db.Integer, default=0)
os = db.Column(db.String, nullable=False, default="linux")
ip = db.Column(db.String(64), nullable=False)
difficulty = db.Column(db.String, nullable=False, default="Easy")
@staticmethod
@cache.memoize(timeout=3600)
def avg_rating(id):
avg_rating = (
UserMachine.query.with_entities(func.avg(UserMachine.rating))
.filter(UserMachine.machine_id == id, UserMachine.rating != 0)
.scalar()
)
return round(avg_rating, 1) if avg_rating else 0
@staticmethod
@cache.cached(timeout=3600 * 3, key_prefix="machines")
def get_all():
return Machine.query.all()
class Challenge(TimeMixin, ReprMixin, db.Model):
__tablename__ = "challenge"
__repr_fields__ = ("title", "category")
id = db.Column(db.Integer, primary_key=True, index=True)
title = db.Column(db.String(64), nullable=False, unique=True)
description = db.Column(db.TEXT, nullable=True)
flag = db.Column(db.TEXT, nullable=False)
points = db.Column(db.Integer, nullable=False, default=0)
url = db.Column(db.TEXT, nullable=True)
difficulty = db.Column(db.String, nullable=True)
category_id = db.Column(db.Integer, db.ForeignKey("category.id"), nullable=False)
category = db.relationship("Category", backref=db.backref("challenges", lazy=True))
tags = db.relationship(
"Tag",
secondary=tags,
lazy="subquery",
backref=db.backref("challenges", lazy="noload"),
)
@staticmethod
@cache.memoize(timeout=3600)
def avg_rating(id):
avg_rating = (
UserChallenge.query.with_entities(func.avg(UserChallenge.rating))
.filter(UserChallenge.challenge_id == id, UserChallenge.rating != 0)
.scalar()
)
return round(avg_rating, 1) if avg_rating else 0
class Machine(TimeMixin, ReprMixin, db.Model):
__tablename__ = "machine"
__repr_fields__ = (
"name",
"os",
)
id = db.Column(db.Integer, primary_key=True, index=True)
name = db.Column(db.String(64), nullable=False, unique=True)
user_hash = db.Column(db.String(32), nullable=False)
root_hash = db.Column(db.String(32), nullable=False)
user_points = db.Column(db.Integer, default=0)
root_points = db.Column(db.Integer, default=0)
os = db.Column(db.String, nullable=False, default="linux")
ip = db.Column(db.String(64), nullable=False)
difficulty = db.Column(db.String, nullable=False, default="Easy")
@staticmethod
@cache.cached(timeout=3600 * 3, key_prefix="machines")
def get_all():
return Machine.query.all()
class Category(ReprMixin, db.Model):
__tablename__ = "category"
__repr_fields__ = ("name", )
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(32), nullable=False)
@staticmethod
@cache.cached(timeout=3600 * 3, key_prefix="challenges")
def get_challenges():
categories = (Category.query.options(joinedload("challenges")).filter(
Category.challenges).all())
return categories
class Challenge(TimeMixin, ReprMixin, db.Model):
__tablename__ = "challenge"
__repr_fields__ = ("title", "category")
id = db.Column(db.Integer, primary_key=True, index=True)
title = db.Column(db.String(64), nullable=False, unique=True)
description = db.Column(db.TEXT, nullable=True)
flag = db.Column(db.TEXT, nullable=False)
points = db.Column(db.Integer, nullable=False, default=0)
url = db.Column(db.TEXT, nullable=True)
difficulty = db.Column(db.String, nullable=True)
category_id = db.Column(db.Integer,
db.ForeignKey("category.id"),
nullable=False)
category = db.relationship("Category",
backref=db.backref("challenges", lazy=True))
tags = db.relationship(
"Tag",
secondary=tags,
lazy="subquery",
backref=db.backref("challenges", lazy="noload"),
)
class Tag(ReprMixin, db.Model):
__tablename__ = "tag"
__repr_fields__ = ("label", )
id = db.Column(db.Integer, primary_key=True)
label = db.Column(db.String(32), nullable=False)
color = db.Column(db.String(16), nullable=False)
class Notification(TimeMixin, ReprMixin, db.Model):
__tablename__ = "notification"
__repr_fields__ = ("title",)
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(64), nullable=False)
body = db.Column(db.TEXT(), nullable=False)