from apps.core.models import CompanyInfo


class CompanyInfoRepository:

    @staticmethod
    def get_company_info(company_name):
        return CompanyInfo.objects.filter(company_name=company_name).first()

    @staticmethod
    def create_company_info(**data):
        return CompanyInfo.objects.create(**data)