mirror of
https://github.com/aljazceru/CTFd.git
synced 2026-02-15 11:14:20 +01:00
Add model changes for Teams.field_entries
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user