from apps.companies.exceptions import NoActiveCompanyError

class ActiveCompanyMixin:
    def get_active_company(self):
        company = self.request.user.active_company
        if not company:
            raise NoActiveCompanyError()
        return company