Use actual ownerId and activeTaskCount in models

Author: amitkummer

manager/tasks/migrations/0001_initial.py

@@ -1,4 +1,4 @@
-# Generated by Django 3.2.8 on 2021-10-27 15:02
+# Generated by Django 3.2.8 on 2021-10-28 12:34
 
 from django.db import migrations, models
 import django.db.models.deletion
@@ -19,7 +19,6 @@ class Migration(migrations.Migration):
                 ('name', models.CharField(max_length=1000)),
                 ('email', models.EmailField(max_length=254, unique=True)),
                 ('favoriteProgrammingLanguage', models.CharField(max_length=1000)),
-                ('activeTaskCount', models.IntegerField()),
             ],
         ),
         migrations.CreateModel(

manager/tasks/models.py

@@ -10,10 +10,13 @@ class Task(models.Model):
         ('active', 'active'),
         ('done', 'done'),
     ])
-    ownerId = models.ForeignKey('Person', on_delete=models.CASCADE)
+    ownerId = models.ForeignKey('Person', to_field='id', on_delete=models.CASCADE)
 
 class Person(models.Model):
     name = models.CharField(max_length=1000)
     email = models.EmailField(unique=True)
     favoriteProgrammingLanguage = models.CharField(max_length=1000)
-    activeTaskCount = models.IntegerField()
+    
+    @property
+    def activeTaskCount(self):
+        return Task.objects.filter(ownerId=self.id).count()

manager/tasks/serializers.py

@@ -10,4 +10,4 @@ class Meta:
 class PersonSerializer(serializers.ModelSerializer):
     class Meta:
         model = Person
-        fields = '__all__'
+        fields = ['name', 'email', 'favoriteProgrammingLanguage', 'activeTaskCount']