Merge pull request #518 from bounswe/fix/BE-stock-serializer-fix

Fix serializer issue with post stocks

backend/marketfeed/migrations/0009_post_stocks.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2 on 2024-12-15 20:09
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('marketfeed', '0008_post_disliked_by_alter_post_liked_by'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='post',
+            name='stocks',
+            field=models.ManyToManyField(blank=True, to='marketfeed.stock', verbose_name='associated stocks'),
+        ),
+    ]

backend/marketfeed/serializers.py

@@ -218,13 +218,15 @@ def create(self, validated_data):
         disliked_by = validated_data.pop('disliked_by', [])
         tags = validated_data.pop('tags', [])
         portfolios = validated_data.pop('portfolios', [])
+        stocks = validated_data.pop('stocks', [])
 
         post = Post.objects.create(**validated_data)
 
         post.liked_by.set(liked_by)
         post.disliked_by.set(disliked_by)
         post.tags.set(tags)
         post.portfolios.set(portfolios)
+        post.stocks.set(stocks)
 
         return post
 
@@ -305,7 +307,6 @@ def __init__(self, *args, **kwargs):
             self.fields['stocks'].required = False
 
 
-
 class MinimalAnnotationSerializer(serializers.Serializer):
     post_id = serializers.IntegerField()
     user_id = serializers.PrimaryKeyRelatedField(read_only=True)