"""Marshmallow Schema for input validation."""

from marshmallow import Schema, fields


class DeleteRelationship(Schema):
    """Class for deleting relationship."""

    from_node = fields.Dict(required=True)
    to_node = fields.Dict(required=True)
    relationship = fields.Dict(required=True)
    vendor_id = fields.Int(required=True)
    subaccount_id = fields.Int(required=True)