Add nested UserSchema and TeamSchema to SubmissionSchema (#2030)

* Add nested UserSchema and TeamSchema to SubmissionSchema to allow easier access to a user/team name without issuing another query/API call
* Closes #2005
This commit is contained in:
Kevin Chung
2021-11-22 23:28:58 -05:00
committed by GitHub
parent 8e96344817
commit 5599e25fc9

View File

@@ -2,11 +2,15 @@ from marshmallow import fields
from CTFd.models import Submissions, ma
from CTFd.schemas.challenges import ChallengeSchema
from CTFd.schemas.teams import TeamSchema
from CTFd.schemas.users import UserSchema
from CTFd.utils import string_types
class SubmissionSchema(ma.ModelSchema):
challenge = fields.Nested(ChallengeSchema, only=["name", "category", "value"])
challenge = fields.Nested(ChallengeSchema, only=["id", "name", "category", "value"])
user = fields.Nested(UserSchema, only=["id", "name"])
team = fields.Nested(TeamSchema, only=["id", "name"])
class Meta:
model = Submissions