from app.models import Company
from app.util import datetime_util


class CompanyAccessor:
    @classmethod
    def get_companies(cls, search_text):
        return Company.objects.filter(company__startswith=search_text)

    @classmethod
    def get_or_create(cls, company_name):
        time_now = datetime_util.get_now_ist_string()

        company = Company.objects.filter(company__iexact=company_name).first()
        if not company:
            return Company.objects.create(company=company_name, created_at=time_now, updated_at=time_now, category='CAT D')

        return company