fixed projects not creating when not specifying leader/industry

Author: VeryBigSad

projects/migrations/0008_alter_project_description_and_more.py

@@ -0,0 +1,67 @@
+# Generated by Django 4.1.3 on 2022-11-18 11:45
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+
+def set_userachievement_user_to_not_null(apps, schema_editor):
+    Project = apps.get_model('projects', 'Project')
+    Project.objects.filter(leader=None).delete()
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ("industries", "0001_initial"),
+        ("projects", "0007_remove_collaborator_unique_collaorator_and_more"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="project",
+            name="description",
+            field=models.TextField(blank=True, null=True),
+        ),
+        migrations.AlterField(
+            model_name="project",
+            name="image_address",
+            field=models.URLField(blank=True, null=True),
+        ),
+        migrations.AlterField(
+            model_name="project",
+            name="industry",
+            field=models.ForeignKey(
+                blank=True,
+                null=True,
+                on_delete=django.db.models.deletion.SET_NULL,
+                related_name="projects",
+                to="industries.industry",
+            ),
+        ),
+        migrations.RunPython(set_userachievement_user_to_not_null),
+        migrations.AlterField(
+            model_name="project",
+            name="leader",
+            field=models.ForeignKey(
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name="leaders_projects",
+                to=settings.AUTH_USER_MODEL,
+            ),
+        ),
+        migrations.AlterField(
+            model_name="project",
+            name="presentation_address",
+            field=models.URLField(blank=True, null=True),
+        ),
+        migrations.AlterField(
+            model_name="project",
+            name="region",
+            field=models.CharField(blank=True, max_length=256, null=True),
+        ),
+        migrations.AlterField(
+            model_name="project",
+            name="short_description",
+            field=models.TextField(blank=True, null=True),
+        ),
+    ]

projects/models.py

@@ -31,24 +31,24 @@ class Project(models.Model):
     """
 
     name = models.CharField(max_length=256, null=True, blank=True)
-    description = models.TextField(blank=True)
-    short_description = models.TextField(blank=True)
-    region = models.CharField(max_length=256, blank=True)
+    description = models.TextField(null=True, blank=True)
+    short_description = models.TextField(null=True, blank=True)
+    region = models.CharField(max_length=256, null=True, blank=True)
     step = models.PositiveSmallIntegerField(choices=VERBOSE_STEPS, null=True, blank=True)
 
     industry = models.ForeignKey(
         Industry,
         on_delete=models.SET_NULL,
         null=True,
+        blank=True,
         related_name="projects",
     )
-    presentation_address = models.URLField(blank=True)
-    image_address = models.URLField(blank=True)
+    presentation_address = models.URLField(null=True, blank=True)
+    image_address = models.URLField(null=True, blank=True)
 
     leader = models.ForeignKey(
         User,
-        on_delete=models.SET_NULL,
-        null=True,
+        on_delete=models.CASCADE,
         related_name="leaders_projects",  # projects in which this user is the leader
     )
 
@@ -85,10 +85,7 @@ class Achievement(models.Model):
     status = models.CharField(max_length=256, null=False)
 
     project = models.ForeignKey(
-        Project,
-        on_delete=models.SET_NULL,
-        null=True,
-        related_name="achievements",
+        Project, on_delete=models.SET_NULL, null=True, related_name="achievements"
     )
 
     objects = AchievementManager()

projects/serializers.py

@@ -3,7 +3,6 @@
 from industries.models import Industry
 from projects.models import Project, Achievement, Collaborator
 from projects.validators import validate_project
-from users.models import CustomUser
 from vacancy.serializers import ProjectVacancyListSerializer
 
 
@@ -95,10 +94,6 @@ class Meta:
 
 
 class ProjectListSerializer(serializers.ModelSerializer):
-    def __init__(self, *args, **kwargs):
-        # might be unnecessary
-        self.max_collaborator_count = kwargs.pop("max_collaborator_count", 4)
-        super().__init__(*args, **kwargs)
 
     collaborators = serializers.SerializerMethodField(method_name="get_collaborators")
     collaborator_count = serializers.SerializerMethodField(
@@ -111,8 +106,9 @@ def get_collaborator_count(cls, obj):
         return len(obj.collaborator_set.all())
 
     def get_collaborators(self, obj):
+        max_collaborator_count = 4
         return CollaboratorSerializer(
-            instance=obj.collaborator_set.all()[: self.max_collaborator_count], many=True
+            instance=obj.collaborator_set.all()[:max_collaborator_count], many=True
         ).data
 
     class Meta:
@@ -144,15 +140,13 @@ def validate(self, data):
         super().validate(data)
         return validate_project(data)
 
-    def create(self, validated_data):
-        industry = Industry.objects.get(id=validated_data.pop("industry"))
-        leader = CustomUser.objects.get(id=validated_data.pop("leader"))
-        project = Project.objects.create(
-            **validated_data,
-            industry=industry,
-            leader=leader,
-        )
-        return project
+    # def create(self, validated_data):
+    #     project = Project.objects.create(
+    #         **validated_data,
+    #         industry_id=validated_data.pop("industry"),
+    #         leader_id=validated_data.pop("leader"),
+    #     )
+    #     return project
 
 
 class ProjectIndustrySerializer(serializers.ModelSerializer):

projects/views.py

@@ -30,7 +30,7 @@ def create(self, request, *args, **kwargs):
         serializer = self.get_serializer(data=request.data)
         serializer.is_valid(raise_exception=True)
         # Doesn't work if not explicitly set like this
-        serializer.validated_data["leader"] = request.user.id
+        serializer.validated_data["leader"] = request.user
 
         self.perform_create(serializer)
         headers = self.get_success_headers(serializer.data)