"""Vendor schema."""
from constants import general

from marshmallow import Schema
from marshmallow import fields
from marshmallow.validate import OneOf


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.
      -quarterback_label_manager (int): Foreign key to orchadmin_users
          table. Optional.
    """

    email = fields.String(required=False, load_default=general.EMAIL)
    vendor_name = fields.String(
        required=True, data_key='parent_rep_owner_name')
    external_identifier_1 = fields.String(
        required=True, data_key='parent_rep_owner_key')
    is_distributor = fields.String(
        validate=OneOf(['Y', 'N']), required=False,
        load_default=general.IS_DISTRIBUTOR)
    source = fields.String(required=False, load_default=general.SOURCE)
    owner = fields.String(required=False, load_default=general.OWNER)
    company_brand = fields.String(required=False, load_default=general.BRAND)
    service_tier_uuid = fields.String(required=False,
                                      load_default=general.SERVICE_TIER_UUID)
    assigned_to = fields.Integer(
        required=False, load_default=general.ASSIGNED_TO)