aljaz/CTFd
1
0
Fork 0
mirror of https://github.com/aljazceru/CTFd.git synced 2026-02-14 10:44:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add TeamFields and TeamFieldEntries

Browse Source
This commit is contained in:
Kevin Chung
2020-08-18 17:18:31 -04:00
parent fa10060a6e
commit 4950f83922
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
CTFd/models/__init__.py
View File

@@ -815,6 +815,10 @@ class UserFields(Fields):
__mapper_args__ = {"polymorphic_identity": "user"}
class TeamFields(Fields):
__mapper_args__ = {"polymorphic_identity": "team"}
class FieldEntries(db.Model):
__tablename__ = "field_entries"
id = db.Column(db.Integer, primary_key=True)
@@ -841,3 +845,9 @@ class UserFieldEntries(FieldEntries):
__mapper_args__ = {"polymorphic_identity": "user"}
user_id = db.Column(db.Integer, db.ForeignKey("users.id", ondelete="CASCADE"))
user = db.relationship("Users", foreign_keys="UserFieldEntries.user_id")
class TeamFieldEntries(FieldEntries):
__mapper_args__ = {"polymorphic_identity": "team"}
team_id = db.Column(db.Integer, db.ForeignKey("teams.id", ondelete="CASCADE"))
team = db.relationship("Teams", foreign_keys="TeamFieldEntries.team_id")