ob-dj-store 0.0.12.1__tar.gz → 0.0.12.2__tar.gz

{ob-dj-store-0.0.12.1 → ob-dj-store-0.0.12.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-dj-store
-Version: 0.0.12.1
+Version: 0.0.12.2
 Summary: OBytes django application for managing ecommerce stores.
 Home-page: https://www.obytes.com/
 Author: OBytes

{ob-dj-store-0.0.12.1 → ob-dj-store-0.0.12.2}/ob_dj_store/apis/stores/rest/serializers/serializers.py

@@ -970,15 +970,15 @@ class GenericSerializer(serializers.Serializer):
         )
 
 
-class FavoriteExtraSerializer(serializers.ModelSerializer):
-    def get_favorite_extras_models():
-        extras_list = []
+def get_favorite_extras_models():
+    extras_list = []
+    for key, value in settings.FAVORITE_TYPES.items():
+        if "extras" in value:
+            extras_list.extend(value["extras"].keys())
+    return extras_list
 
-        for key, value in settings.FAVORITE_TYPES.items():
-            if "extras" in value:
-                extras_list.extend(value["extras"].keys())
-        return extras_list
 
+class FavoriteExtraSerializer(serializers.ModelSerializer):
     content_object = GenericSerializer(read_only=True)
     object_id = serializers.IntegerField(min_value=1, write_only=True)
     object_type = serializers.ChoiceField(
@@ -994,6 +994,16 @@ class FavoriteExtraSerializer(serializers.ModelSerializer):
             "object_type",
         )
 
+    def to_representation(self, instance):
+        data = super().to_representation(instance)
+        extras = get_favorite_extras_models()
+        model_name = instance.content_type.model
+        for extra in extras:
+            if model_name == extra.lower():
+                data["object_type"] = extra
+                break
+        return data
+
 
 class FavoriteSerializer(serializers.ModelSerializer):
     content_object = GenericSerializer(read_only=True)
@@ -1017,6 +1027,15 @@ class FavoriteSerializer(serializers.ModelSerializer):
             "name": {"required": True},
         }
 
+    def to_representation(self, instance):
+        data = super().to_representation(instance)
+        model_name = instance.content_type.model
+        for favorite in store_settings.FAVORITE_TYPES.keys():
+            if model_name == favorite.lower():
+                data["object_type"] = favorite
+                break
+        return data
+
     def _lookup_validation(self, data):
         content_type = ContentType.objects.get_for_model(data["content_object"])
         queryset = Favorite.objects.filter(

{ob-dj-store-0.0.12.1 → ob-dj-store-0.0.12.2}/ob_dj_store.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-dj-store
-Version: 0.0.12.1
+Version: 0.0.12.2
 Summary: OBytes django application for managing ecommerce stores.
 Home-page: https://www.obytes.com/
 Author: OBytes