from app.user_employment.company_accessor import CompanyAccessor
from app.util.util import get_or_none


class Company:
    def __init__(self, company):
        self.company = company

    @classmethod
    def get_companies(cls, search_text):
        return CompanyAccessor.get_companies(search_text)

    @classmethod
    def get_or_create(cls, company_name):
        return CompanyAccessor.get_or_create(company_name)

    @classmethod
    def get_all_serialized(cls, companies):
        serialized_companies = []
        companies = companies.values('id', 'company')

        for company in companies:
            serialized_companies.append({
                'id': company['id'],
                'company': company['company']
            })

        return serialized_companies