create rancher incentive plan structure - add pos device main sheba

apps/product/pos/api/v1/serializers/quota_distribution_serializers.py

@@ -80,6 +80,11 @@ class QuotaDistributionSerializer(serializers.ModelSerializer):
         """ Custom output of serializer """
 
         representation = super().to_representation(instance)
+
+        organization = self.context['organization']
+        rancher = self.context['rancher']
+        device = self.context['device']
+
         if instance.quota:
             representation['quota'] = {
                 'quota_identity': instance.quota.quota_id,
@@ -87,7 +92,7 @@ class QuotaDistributionSerializer(serializers.ModelSerializer):
                 'quota_livestock_allocations': quota_live_stock_allocation_info(
                     instance.quota
                 ),
-                'quota_incentive_plans': quota_incentive_plans_info(instance.quota),
+                'quota_incentive_plans': quota_incentive_plans_info(instance.quota, rancher),
                 'quota_sale_license': instance.quota.sale_license,
                 'has_sale_license': instance.quota.is_in_valid_time()
             }
@@ -101,13 +106,14 @@ class QuotaDistributionSerializer(serializers.ModelSerializer):
             'name': instance.quota.product.name,
             'id': instance.quota.product.id,
             'free_sale_for_all': sale_limitation if sale_limitation else False,
-            'free_sale_for_this_rancher': self.context['rancher'].ignore_purchase_limit
+            'free_sale_for_this_rancher': rancher.ignore_purchase_limit
         }
 
         representation['pricing'] = {  # noqa
+            'main_account_sheba': organization.bank_information.first().sheba,
             'pricing_attributes': quota_attribute_value(instance.quota),
             'sharing': pos_organizations_sharing_information(
-                self.context['device'],
+                device,
                 instance.quota,
                 distribution=instance
             ),
@@ -127,12 +133,12 @@ class QuotaDistributionSerializer(serializers.ModelSerializer):
 
         if 'rancher' in self.context.keys():
             # rancher herd & live stock statistics
-            livestock_counts_list, livestock_counts_dict = get_rancher_statistics(self.context['rancher'])
+            livestock_counts_list, livestock_counts_dict = get_rancher_statistics(rancher)
             representation['rancher_statistics'] = livestock_counts_list
 
             # rancher live stock statistics by quota distributions
             representation['rancher_quota_weight_statistics'] = rancher_quota_weight(
-                self.context['rancher'], distribution=instance
+                rancher, distribution=instance
             )
 
         if instance.assigned_organization:

apps/product/pos/api/v1/viewsets/quota_distribution_api.py

@@ -106,7 +106,11 @@ class QuotaDistributionViewSet(viewsets.ModelViewSet, DynamicSearchMixin, POSDev
         # paginate & response
         page = self.paginate_queryset(available_distributions)  # noqa
         if page is not None:
-            serializer = self.get_serializer(page, many=True, context={'rancher': rancher.first(), 'device': device})
+            serializer = self.get_serializer(page, many=True, context={
+                'rancher': rancher.first(),
+                'device': device,
+                'organization': organization
+            })
             # set custom message for paginator
             if not rancher:
                 self.paginator.set_message("دامدار با کد ملی مد نظر یافت نشد")  # noqa