from app.models import Products
from app.util.util import get_or_none


class MerchantProductAccessor:
    @classmethod
    def get_merchant_product(cls, merchant_id, product_id):
        if merchant_id:
            return get_or_none(Products, merchant_id=merchant_id, id=product_id)
        else:
            return get_or_none(Products, id=product_id)

    @classmethod
    def get_merchant_products(cls, merchant_id, center_id=None):
        products = Products.objects.filter(merchant_id=merchant_id, deleted_at=None)

        if center_id:
            products = products.filter(center_id=center_id, deleted_at=None)

        return products