masschallenge/impact-api

# MIT License
# Copyright (c) 2017 MassChallenge, Inc.

from django.db.models import Q
from rest_framework.response import Response

from accelerator.models import (
    Organization,
    PartnerTeamMember,
    StartupTeamMember,
)
from impact.v1.views.impact_view import ImpactView
from impact.v1.views.utils import (
    coalesce_dictionaries,
    map_data,
)
from impact.v1.helpers import ORGANIZATION_USER_FIELDS


class OrganizationUsersView(ImpactView):
    view_name = "organization_users"
    model = Organization
    list_key = "users"

    @classmethod
    def fields(self):
        return ORGANIZATION_USER_FIELDS

    def get(self, request, pk):
        self.instance = self.model.objects.get(pk=pk)
        all_data = self.startup_data() + self.partner_data()
        return Response({"users": coalesce_dictionaries(all_data)})

    def partner_data(self):
        return map_data(PartnerTeamMember,
                        Q(partner__organization=self.instance),
                        "partner__id",
                        ["team_member_id", "partner_administrator"],
                        ["id", "partner_administrator"])

    def startup_data(self):
        return map_data(StartupTeamMember,
                        Q(startup__organization=self.instance),
                        "startup__id",
                        ["user_id",
                         "startup_administrator",
                         "primary_contact"],
                        ["id", "startup_administrator", "primary_contact"])