from marshmallow import Schema, fields, validate


class AccessTokenSchema(Schema):
    access_token = fields.Str()
    token_type = fields.Str()
    expires_in = fields.Int()


class TokenSchema(Schema):
    audience = fields.Str(required=True)
    grant_type = fields.Str(
        validate=validate.OneOf(["client_credentials"]),
        required=True,
    )
    client_id = fields.Str(required=True)
    client_secret = fields.Str(required=True)