Add loader for Box.displayUnit

back/boxtribute_server/authz.py

@@ -37,6 +37,7 @@
     "standard_product",
     "tag_relation",
     "transaction",
+    "unit",
     "user",
 )
 

back/boxtribute_server/business_logic/warehouse/box/fields.py

@@ -55,6 +55,11 @@ async def resolve_box_location(box_obj, info):
         return
 
 
+@box.field("displayUnit")
+def resolve_box_unit(box_obj, info):
+    return info.context["unit_loader"].load(box_obj.display_unit_id)
+
+
 @box.field("measureValue")
 def resolve_box_measure_value(box_obj, _):
     if box_obj.display_unit_id is None:

back/boxtribute_server/graph_ql/execution.py

@@ -22,6 +22,7 @@
     SizesForSizeRangeLoader,
     TagsForBoxLoader,
     TransferAgreementLoader,
+    UnitLoader,
     UnitsForDimensionLoader,
     UserLoader,
 )
@@ -55,6 +56,7 @@ async def run():
             "tags_for_box_loader": TagsForBoxLoader(),
             "transfer_agreement_loader": TransferAgreementLoader(),
             "units_for_dimension_loader": UnitsForDimensionLoader(),
+            "unit_loader": UnitLoader(),
             "user_loader": UserLoader(),
         }
 

back/boxtribute_server/graph_ql/loaders.py

@@ -89,6 +89,11 @@ def __init__(self):
         super().__init__(Size)
 
 
+class UnitLoader(SimpleDataLoader):
+    def __init__(self):
+        super().__init__(Unit)
+
+
 class OrganisationLoader(SimpleDataLoader):
     def __init__(self):
         super().__init__(Organisation)

back/test/auth.py

@@ -141,6 +141,7 @@ def _create_jwt_payload(
             "transfer_agreement:create",
             "transfer_agreement:edit",
             "transfer_agreement_detail:read",
+            "unit:read",
         ]
     else:
         payload[f"{JWT_CLAIM_PREFIX}/permissions"] = permissions