"""Logic for Suppliers.

Provides logic for getting supplier information.
"""

from owsresponse import response

from account.logic import vendor
from account.models import supplier as supplier_model


def get_suppliers(vendor_id):
    """Get supplier list.

    Args:
        vendor_id (int): unique identifier of the vendor.

    Returns:
        response.Response: success if vendor valid.
    """
    if vendor.is_vendor(vendor_id):
        return supplier_model.get_default(vendor_id)

    return response.create_not_found_response()


def set_suppliers(vendor_id, store_ids):
    """Set supplier list.

    Args:
        vendor_id (int): unique identifier of the vendor.
        store_ids (list): list of store ids to be stored.

    Returns:
        response.Response: success if vendor valid.
    """
    if vendor.is_vendor(vendor_id):
        return supplier_model.set_default(vendor_id, store_ids)

    return response.create_not_found_response()