Add model changes for Teams.field_entries

This commit is contained in:
Kevin Chung
2020-08-20 03:01:11 -04:00
parent 3e534ef9c7
commit 6b0629ed2a

View File

@@ -467,6 +467,10 @@ class Teams(db.Model):
captain_id = db.Column(db.Integer, db.ForeignKey("users.id", ondelete="SET NULL"))
captain = db.relationship("Users", foreign_keys=[captain_id])
field_entries = db.relationship(
"TeamFieldEntries", foreign_keys="TeamFieldEntries.team_id", lazy="joined"
)
created = db.Column(db.DateTime, default=datetime.datetime.utcnow)
def __init__(self, **kwargs):
@@ -478,6 +482,10 @@ class Teams(db.Model):
return hash_password(str(plaintext))
@property
def fields(self):
return self.get_fields(admin=False)
@property
def solves(self):
return self.get_solves(admin=False)
@@ -503,6 +511,12 @@ class Teams(db.Model):
else:
return None
def get_fields(self, admin=False):
if admin:
return self.field_entries
return [entry for entry in self.field_entries if entry.field.public]
def get_solves(self, admin=False):
from CTFd.utils import get_config