from app.models import MerchantLocation
from app.util.util import get_or_none


class MerchantCenterAccessor:
    @classmethod
    def get_center(cls, merchant_id, center_id):
        return MerchantLocation.objects.filter(merchant_id=merchant_id, id=center_id).first()

    @classmethod
    def get_merchant_centers(cls, merchant_id):
        return MerchantLocation.objects.filter(merchant_id=merchant_id, deleted_at=None)

    @classmethod
    def get_merchant_center_by_slug(cls, center_slug, merchant_id):
        return get_or_none(MerchantLocation, slug=center_slug, deleted_at=None, merchant_id=merchant_id)

    @classmethod
    def get_merchant_center_by_search_text(cls, search_text, merchant_id):
        return MerchantLocation.objects.filter(name__startswith=search_text, merchant_id=merchant_id)