Add Sublet Drafting

backend/sublet/migrations/0005_sublet_is_draft_alter_sublet_end_date_and_more.py

@@ -0,0 +1,30 @@
+# Generated by Django 5.0.2 on 2024-03-15 22:34
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("sublet", "0004_alter_sublet_external_link"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="sublet", name="is_draft", field=models.BooleanField(default=True),
+        ),
+        migrations.AlterField(
+            model_name="sublet", name="end_date", field=models.DateField(blank=True, null=True),
+        ),
+        migrations.AlterField(
+            model_name="sublet",
+            name="expires_at",
+            field=models.DateTimeField(blank=True, null=True),
+        ),
+        migrations.AlterField(
+            model_name="sublet", name="price", field=models.IntegerField(blank=True, null=True),
+        ),
+        migrations.AlterField(
+            model_name="sublet", name="start_date", field=models.DateField(blank=True, null=True),
+        ),
+    ]

backend/sublet/models.py

@@ -33,6 +33,7 @@ class Sublet(models.Model):
     sublettees = models.ManyToManyField(
         User, through=Offer, related_name="sublets_offered", blank=True
     )
+    is_draft = models.BooleanField(default=True)
     favorites = models.ManyToManyField(User, related_name="sublets_favorited", blank=True)
     amenities = models.ManyToManyField(Amenity, blank=True)
 
@@ -42,12 +43,12 @@ class Sublet(models.Model):
     baths = models.DecimalField(max_digits=3, decimal_places=1, null=True, blank=True)
     description = models.TextField(null=True, blank=True)
     external_link = models.URLField(max_length=255, null=True, blank=True)
-    price = models.IntegerField()
+    price = models.IntegerField(null=True, blank=True)
     negotiable = models.BooleanField(default=True)
     created_at = models.DateTimeField(auto_now_add=True)
-    expires_at = models.DateTimeField()
-    start_date = models.DateField()
-    end_date = models.DateField()
+    expires_at = models.DateTimeField(null=True, blank=True)
+    start_date = models.DateField(null=True, blank=True)
+    end_date = models.DateField(null=True, blank=True)
 
     def __str__(self):
         return f"{self.title} by {self.subletter}"

backend/sublet/serializers.py

@@ -61,6 +61,35 @@
         many=True, queryset=Amenity.objects.all(), required=False
     )
 
+    def validate_not_draft(self, validated_data, instance=None):
+        if "is_draft" in validated_data and not validated_data["is_draft"]:
+            # check that certain fields are there
+            fields = [
+                "title",
+                "address",
+                "price",
+                "negotiable",
+                "start_date",
+                "end_date",
+                "expires_at",
+            ]
+            if bad_fields := [
+                field
+                for field in fields
+                if (field in validated_data and not validated_data[field])
+                or (
+                    field not in validated_data
+                    and (
+                        instance is None
+                        or not hasattr(instance, field)
+                        or not getattr(instance, field)
+                    )
+                )
+            ]:
+                raise serializers.ValidationError(
+                    f"{', '.join(bad_fields)} are required to publish sublet."
+                )
+
     class Meta:
         model = Sublet
         read_only_fields = [
@@ -72,7 +101,7 @@
         ]
         fields = [
             "id",
-            "subletter",
+            "is_draft",
             "amenities",
             "title",
             "address",
@@ -93,6 +122,7 @@
 
     def create(self, validated_data):
         validated_data["subletter"] = self.context["request"].user
+        self.validate_not_draft(validated_data)
         instance = super().create(validated_data)
         instance.save()
         return instance
@@ -104,6 +134,8 @@
             self.context["request"].user == instance.subletter
             or self.context["request"].user.is_superuser
         ):
+            if instance.is_draft:
+                self.validate_not_draft(validated_data, instance)
             instance = super().update(instance, validated_data)
             instance.save()
             return instance