"""V1 Vendor schema."""

from marshmallow import Schema, fields

from account.constants.constants import LOOKUP_FETCH_FLAGS


class LookupByUuids(Schema):
    """Schema for requesting entities by uuids.

    Properties:
      -uuids (list): list of uuid ids
    """

    uuids = fields.List(fields.UUID(required=True))


class LookupByUuidsWithFetchFlags(LookupByUuids):
    """Schema for requesting entities by uuids with optional fetch flags.

    Properties:
      -uuids (list): list of uuid ids
      -fetch_flags (list): list of fetch flags
    """

    fetch_flags = fields.List(fields.Enum(enum=LOOKUP_FETCH_FLAGS))


class LookupVendorsByUuids(LookupByUuidsWithFetchFlags):
    """Schema for requesting vendors by uuids.

    Properties:
      -uuids (list): list of uuid ids
      -fetch_flags (list): list of fetch flags
    """

    pass


class LookupSubaccountsByUuids(LookupByUuidsWithFetchFlags):
    """Schema for requesting subaccounts by uuids

    Properties:
     -uuids (list): list of uuid ids
     -fetch_flags (list): list of fetch flags
    """

    pass


class LookupVendorsByVendorIds(Schema):
    """Schema for requesting vendors by vendor ids.

    Properties:
      -vendor_ids (list): list of vendor ids
      -fetch_flags (list): list of fetch flags
    """

    vendor_ids = fields.List(fields.Int(required=True))
    fetch_flags = fields.List(fields.Enum(enum=LOOKUP_FETCH_FLAGS))


class LookupSubaccountsBySubaccountIds(Schema):
    """Schema for requesting subaccounts by subaccount ids.

    Properties:
      -subaccount_ids (list): list of subaccount ids
      -fetch_flags (list): list of fetch flags
    """

    subaccount_ids = fields.List(fields.Int(required=True))
    fetch_flags = fields.List(fields.Enum(enum=LOOKUP_FETCH_FLAGS))


class LookupVendorFeaturesByUuids(LookupByUuids):
    """Schema for requesting vendor features by uuids.

    Properties:
      -uuids (list): list of uuid ids
    """

    pass


class LookupSubaccountFeaturesByUuids(LookupByUuids):
    """Schema for requesting subaccount features by uuids.

    Properties:
      -uuids (list): list of uuid ids
    """

    pass


class LookupCompanyBrandsByUuids(Schema):
    """Schema for requesting company brands by uuids.

    Properties:
      -uuids (list): list of uuid ids
    """

    uuids = fields.List(fields.UUID(required=True))


class LookupCompanyBrandsByIds(Schema):
    """Schema for requesting company brands by ids.

    Properties:
      -company_brand_ids (list): list of ids
    """

    company_brand_ids = fields.List(fields.Int(required=True))