notifications-admin/app/models/organization.py

from collections import OrderedDict

from werkzeug.utils import cached_property

from app.models import (
    JSONModel,
    ModelList,
    SerialisedModelCollection,
    SortByNameMixin,
)
from app.notify_client.email_branding_client import email_branding_client
from app.notify_client.organizations_api_client import organizations_client


class Organization(JSONModel, SortByNameMixin):

    TYPE_FEDERAL = 'federal'
    TYPE_STATE = 'state'
    TYPE_OTHER = 'other'

    TYPE_LABELS = OrderedDict([
        (TYPE_FEDERAL, 'Federal government'),
        (TYPE_STATE, 'State government'),
        (TYPE_OTHER, 'Other'),
    ])

    ALLOWED_PROPERTIES = {
        'id',
        'name',
        'active',
        'organization_type',
        'email_branding_id',
        'domains',
        'request_to_go_live_notes',
        'count_of_live_services',
        'billing_contact_email_addresses',
        'billing_contact_names',
        'billing_reference',
        'purchase_order_number',
        'notes',
    }

    @classmethod
    def from_id(cls, org_id):
        if not org_id:
            return cls({})
        return cls(organizations_client.get_organization(org_id))

    @classmethod
    def from_domain(cls, domain):
        return cls(organizations_client.get_organization_by_domain(domain))

    @classmethod
    def from_service(cls, service_id):
        return cls(organizations_client.get_service_organization(service_id))

    @classmethod
    def create_from_form(cls, form):
        return cls.create(
            name=form.name.data,
            organization_type=form.organization_type.data,
        )

    @classmethod
    def create(cls, name, organization_type):
        return cls(organizations_client.create_organization(
            name=name,
            organization_type=organization_type,
        ))

    def __init__(self, _dict):

        super().__init__(_dict)

        if self._dict == {}:
            self.name = None
            self.domains = []
            self.organization_type = None
            self.request_to_go_live_notes = None
            self.email_branding_id = None

    @property
    def organization_type_label(self):
        return self.TYPE_LABELS.get(self.organization_type)

    @property
    def billing_details(self):
        billing_details = [
            self.billing_contact_email_addresses,
            self.billing_contact_names,
            self.billing_reference,
            self.purchase_order_number
        ]
        if any(billing_details):
            return billing_details
        else:
            return None

    @cached_property
    def services(self):
        return organizations_client.get_organization_services(self.id)

    @cached_property
    def service_ids(self):
        return [s['id'] for s in self.services]

    @property
    def live_services(self):
        return [s for s in self.services if s['active'] and not s['restricted']]

    @property
    def trial_services(self):
        return [s for s in self.services if not s['active'] or s['restricted']]

    @cached_property
    def invited_users(self):
        from app.models.user import OrganizationInvitedUsers
        return OrganizationInvitedUsers(self.id)

    @cached_property
    def active_users(self):
        from app.models.user import OrganizationUsers
        return OrganizationUsers(self.id)

    @cached_property
    def team_members(self):
        return sorted(
            self.invited_users + self.active_users,
            key=lambda user: user.email_address.lower(),
        )

    @cached_property
    def email_branding(self):
        if self.email_branding_id:
            return email_branding_client.get_email_branding(
                self.email_branding_id
            )['email_branding']

    @property
    def email_branding_name(self):
        if self.email_branding_id:
            return self.email_branding['name']
        return 'GOV.UK'

    def update(self, delete_services_cache=False, **kwargs):
        response = organizations_client.update_organization(
            self.id,
            cached_service_ids=self.service_ids if delete_services_cache else None,
            **kwargs
        )
        self.__init__(response)

    def associate_service(self, service_id):
        organizations_client.update_service_organization(
            service_id,
            self.id
        )

    def services_and_usage(self, financial_year):
        return organizations_client.get_services_and_usage(self.id, financial_year)


class Organizations(SerialisedModelCollection):
    model = Organization


class AllOrganizations(ModelList, Organizations):
    client_method = organizations_client.get_organizations