{devbugfix} counts ams list

app/applicationlayer/ams/asset_group/serializers.py

@@ -7,8 +7,17 @@ class AssetGroupSerializer(serializers.ModelSerializer):
 
     def to_representation(self, instance):
         ret = super().to_representation(instance)
-        ret['asset_type_count'] = 0
-        ret['asset_count'] = 0
+
+        asset_stocks = models.AMSAssetStock.objects.filter(asset_group=instance.code)
+
+        asset_counter = 0
+        asset_type_counter = 0
+        for asset_stock in asset_stocks:
+            asset_counter = asset_counter + models.AMSAsset.objects.filter(code=asset_stock.asset.code).count()
+            asset_type_counter = asset_type_counter + 1
+        
+        ret['asset_type_count'] = asset_type_counter
+        ret['asset_count'] = asset_counter
         ret['asset_stock_count'] = instance.ams_asset_to_assetgroup.count()
         
         return ret

app/applicationlayer/ams/asset_type/serializers.py

@@ -7,8 +7,15 @@ class AssetTypeSerializer(serializers.ModelSerializer):
 
     def to_representation(self, instance):
         ret = super().to_representation(instance)
-        ret['asset_count'] = instance.ams_asset_type.count()
-        ret['asset_stock_count'] = 0
+
+        assets = instance.ams_asset_type.select_related()
+        ams_stock_counter = 0
+
+        for asset in assets:
+            ams_stock_counter = ams_stock_counter + models.AMSAsset.objects.get(code=asset.code).ams_asset_to_assetdetail.count()
+        
+        ret['asset_count'] = assets.count()
+        ret['asset_stock_count'] = ams_stock_counter
         
         return ret
 

app/entities/models.py

@@ -1358,6 +1358,7 @@ class AMSAsset(models.Model):
         to_field='code',
         related_name='ams_asset_type'
     )
+    # asset_type = models.ManyToManyField(AMSAssetType)
     description = models.CharField(
         blank=True,
         null=True,