"""Vendor schema."""

from marshmallow import Schema
from marshmallow import fields


class CreateVendorSchema(Schema):
    """Create Vendor schema.

    Properties:
      -email (str): the email for this Vendor
      -vendor_name (str): full name of the Vendor
      -owner (str): owner of this Vendor
      -source (str): source of creation
      -company_brand (str): brand for this Vendor
      -service_tier_uuid (str): uuid of service tier for this vendor.
      -assigned_to (int): Foreign key to orchadmin_users table. Optional.
      -assigned_reviewer (int): ID of the orchadmin user who is reviewer.
      -quarterback_label_manager (int): Foreign key to orchadmin_users
          table. Optional.
    """

    email = fields.String(required=True)
    vendor_name = fields.String(required=True)
    source = fields.String(data_key='source', required=True)
    owner = fields.String(required=False)
    company_brand = fields.String(
        data_key='brand', required=True)
    service_tier_uuid = fields.String(required=True)
    migrated_to_abacus = fields.Boolean(required=False, missing=True)
    assigned_to = fields.Integer(required=False)
    assigned_reviewer = fields.Integer(required=False)
    quarterback_label_manager = fields.Integer(required=False)