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.models import Submissions, ma
from CTFd.schemas.challenges import ChallengeSchema from CTFd.schemas.challenges import ChallengeSchema
from CTFd.schemas.teams import TeamSchema
from CTFd.schemas.users import UserSchema
from CTFd.utils import string_types from CTFd.utils import string_types
class SubmissionSchema(ma.ModelSchema): 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: class Meta:
model = Submissions model = Submissions