diff --git a/backend/marketfeed/migrations/0009_post_stocks.py b/backend/marketfeed/migrations/0009_post_stocks.py new file mode 100644 index 00000000..56cbc7ae --- /dev/null +++ b/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'), + ), + ] diff --git a/backend/marketfeed/serializers.py b/backend/marketfeed/serializers.py index ff767400..c039c511 100644 --- a/backend/marketfeed/serializers.py +++ b/backend/marketfeed/serializers.py @@ -218,6 +218,7 @@ 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) @@ -225,6 +226,7 @@ def create(self, validated_data): 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)