Extend back office assets API by user / owner department and user/ ow…

…ner company
allegro · Sep 7, 2020 · 6341d92 · 6341d92
1 parent 97829db
commit 6341d92
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/src/ralph/accounts/api_simple.py b/src/ralph/accounts/api_simple.py
@@ -10,3 +10,12 @@ class Meta:
         fields = ('id', 'url', 'username', 'first_name', 'last_name')
         read_only_fields = fields
         depth = 1
+
+
+class ExtendedSimpleRalphUserSerializer(SimpleRalphUserSerializer):
+    class Meta:
+        model = get_user_model()
+        fields = SimpleRalphUserSerializer.Meta.fields + (
+            'department',
+            'company'
+        )
diff --git a/src/ralph/back_office/api.py b/src/ralph/back_office/api.py
@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-from ralph.accounts.api_simple import SimpleRalphUserSerializer
+from ralph.accounts.api_simple import ExtendedSimpleRalphUserSerializer
 from ralph.api import RalphAPISerializer, RalphAPIViewSet, router
 from ralph.assets.api.serializers import AssetSerializer
 from ralph.assets.api.views import base_object_descendant_prefetch_related
@@ -39,8 +39,8 @@ class Meta(AssetSerializer.Meta):
 
 
 class BackOfficeAssetSerializer(AssetSerializer):
-    user = SimpleRalphUserSerializer()
-    owner = SimpleRalphUserSerializer()
+    user = ExtendedSimpleRalphUserSerializer()
+    owner = ExtendedSimpleRalphUserSerializer()
 
     class Meta(AssetSerializer.Meta):
         model = BackOfficeAsset