From 4950f8392241b048f6d9b0c4afa5d828fe79aa78 Mon Sep 17 00:00:00 2001 From: Kevin Chung Date: Tue, 18 Aug 2020 17:18:31 -0400 Subject: [PATCH] Add TeamFields and TeamFieldEntries --- CTFd/models/__init__.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/CTFd/models/__init__.py b/CTFd/models/__init__.py index a849bb24..e48d5004 100644 --- a/CTFd/models/__init__.py +++ b/CTFd/models/__init__.py @@ -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") \ No newline at end of file