MEI-9862 return users main organizations in user and course api

edx_solutions_api_integration/courses/views.py

@@ -1318,6 +1318,7 @@ def get_serializer_context(self):
             "last_login",
             "attributes",
             "organization_groups",
+            "main_organization",
         ]
 
         active_attributes = []

edx_solutions_api_integration/imports/views.py

@@ -16,6 +16,7 @@
 from edx_solutions_api_integration.permissions import SecureViewSet
 from edx_solutions_api_integration.users.views import _manage_role
 from edx_solutions_organizations.models import Organization
+from edx_solutions_organizations.receivers import user_organization_updated
 from lms.djangoapps.discussion.notification_prefs.views import enable_notifications
 from opaque_keys.edx.keys import CourseKey
 from openedx.core.djangoapps.course_groups.cohorts import (add_cohort,
@@ -186,6 +187,11 @@ def _create_user(self, data, errors, response):
         except Exception as exc:
             self._add_error(errors, str(exc.message), _('Enrolling Participant in Company'), email)
 
+        user_organization_updated.send(
+            sender=__name__, user_id=user.id,
+            organization_id=company.id
+        )
+
     def _enroll_user(self, data, errors, response={}):
         """Enroll user in a course and add him to a cohort."""
         user = data.get('user_object')

edx_solutions_api_integration/users/serializers.py

@@ -4,6 +4,7 @@
 from django.core.exceptions import ObjectDoesNotExist
 from edx_solutions_api_integration.models import APIUser
 from edx_solutions_api_integration.utils import get_profile_image_urls_by_username
+from edx_solutions_organizations.models import Organization
 from edx_solutions_organizations.serializers import BasicOrganizationSerializer
 from rest_framework import serializers
 
@@ -48,6 +49,18 @@ class UserSerializer(DynamicFieldsModelSerializer):
     attributes = serializers.SerializerMethodField('get_organization_attributes')
     course_groups = serializers.SerializerMethodField('get_user_course_groups')
     organization_groups = serializers.SerializerMethodField('get_user_organization_groups')
+    main_organization = serializers.SerializerMethodField('get_user_main_organization')
+
+    def get_user_main_organization(self, user):
+        main_user_organization = None
+        mapped_user_organization =  user.user_organizations.all().filter(is_main_company=True).first()
+        if mapped_user_organization:
+            try:
+                main_user_organization = Organization.objects.get(id=mapped_user_organization.organization_id)
+            except ObjectDoesNotExist:
+                main_user_organization = None
+        main_user_organization = [main_user_organization] if main_user_organization else []
+        return BasicOrganizationSerializer(main_user_organization, many=True, context=self.context).data
 
     def get_user_organization_groups(self, user):
         """
@@ -164,6 +177,7 @@ class Meta:
             "attributes",
             "course_groups",
             "organization_groups",
+            "main_organization",
         )
         read_only_fields = ("id", "email", "username")
 

edx_solutions_api_integration/users/views.py

@@ -1493,8 +1493,18 @@ def get_queryset(self):
         user = get_user_from_request_params(self.request, self.kwargs)
         if not user:
             return []
-
-        return user.organizations.all()
+        main_user_organization = None
+        mapped_user_organization =  user.user_organizations.all().filter(is_main_company=True).first()
+        if mapped_user_organization:
+            try:
+                main_user_organization = Organization.objects.get(id=mapped_user_organization.organization_id)
+            except ObjectDoesNotExist:
+                main_user_organization = None
+        user_organizations_set = user.organizations.all()
+        if main_user_organization:
+            user_organizations_set = user_organizations_set.exclude(id=main_user_organization.id)
+            user_organizations_set = [main_user_organization] + list(user_organizations_set)
+        return user_organizations_set
 
 
 class UsersWorkgroupsList(SecureListAPIView):

setup.py

@@ -4,7 +4,7 @@
 
 setup(
     name='api-integration',
-    version='5.0.5',
+    version='5.0.6',
     description='RESTful api integration for edX platform',
     long_description=open('README.rst').read(),
     author='edX',