Fix issue #197: Add timestamps to models demo

photo/migrations/0005_collection_created_at_collection_updated_at_and_more.py

@@ -0,0 +1,67 @@
+# Generated by Django 4.2.8 on 2025-02-11 11:40
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("photo", "0004_alter_contest_prize"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="collection",
+            name="created_at",
+            field=models.DateTimeField(auto_now_add=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="collection",
+            name="updated_at",
+            field=models.DateTimeField(auto_now=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="contest",
+            name="created_at",
+            field=models.DateTimeField(auto_now_add=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="contest",
+            name="updated_at",
+            field=models.DateTimeField(auto_now=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="contestsubmission",
+            name="created_at",
+            field=models.DateTimeField(auto_now_add=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="contestsubmission",
+            name="updated_at",
+            field=models.DateTimeField(auto_now=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="picture",
+            name="created_at",
+            field=models.DateTimeField(auto_now_add=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="picture",
+            name="updated_at",
+            field=models.DateTimeField(auto_now=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="picturecomment",
+            name="updated_at",
+            field=models.DateTimeField(auto_now=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="user",
+            name="created_at",
+            field=models.DateTimeField(auto_now_add=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="user",
+            name="updated_at",
+            field=models.DateTimeField(auto_now=True, null=True),
+        ),
+    ]

photo/models.py

@@ -48,6 +48,8 @@ def create_superuser(self, email, password=None, **kwargs):
 
 class SoftDeleteModel(models.Model):
     is_deleted = models.BooleanField(default=False)
+    created_at = models.DateTimeField(null=True, blank=True, auto_now_add=True)
+    updated_at = models.DateTimeField(null=True, blank=True, auto_now=True)
     objects = SoftDeleteManager()
     all_objects = models.Manager()
 

photo/tests/test_timestamps.py

@@ -0,0 +1,57 @@
+from django.test import TestCase
+from django.utils import timezone
+from photo.models import User, Picture, Collection, Contest, ContestSubmission
+from django.core.files.uploadedfile import SimpleUploadedFile
+
+
+class TimestampFieldsTest(TestCase):
+    def setUp(self):
+        self.user = User.objects.create_user(
+            email="test@example.com",
+            password="testpass123",
+            name_first="Test",
+            name_last="User"
+        )
+
+        self.picture = Picture.objects.create(
+            user=self.user,
+            name="Test Picture",
+            file=SimpleUploadedFile(
+                "test.jpg",
+                b"file_content",
+                content_type="image/jpeg"
+            )
+        )
+
+    def test_timestamp_fields_on_create(self):
+        """Test that created_at and updated_at are set on model creation"""
+        collection = Collection.objects.create(
+            name="Test Collection",
+            user=self.user
+        )
+
+        self.assertIsNotNone(collection.created_at)
+        self.assertIsNotNone(collection.updated_at)
+        self.assertIsInstance(collection.created_at, timezone.datetime)
+        self.assertIsInstance(collection.updated_at, timezone.datetime)
+        self.assertEqual(collection.created_at.date(), timezone.now().date())
+        self.assertEqual(collection.updated_at.date(), timezone.now().date())
+
+    def test_updated_at_changes_on_update(self):
+        """Test that updated_at changes when model is updated"""
+        collection = Collection.objects.create(
+            name="Test Collection",
+            user=self.user
+        )
+        original_updated_at = collection.updated_at
+
+        # Wait a moment to ensure time difference
+        import time
+        time.sleep(0.1)
+
+        collection.name = "Updated Collection"
+        collection.save()
+
+        self.assertNotEqual(collection.updated_at, original_updated_at)
+        self.assertEqual(collection.created_at.date(), timezone.now().date())
+        self.assertEqual(collection.updated_at.date(), timezone.now().date())

photo/types.py

@@ -23,6 +23,8 @@ class UserType:
     profile_picture: "PictureType"
     profile_picture_updated_at: strawberry.auto
     user_handle: str
+    created_at: strawberry.auto | None
+    updated_at: strawberry.auto | None
 
 
 @strawberry.django.type(Picture)
@@ -32,6 +34,8 @@ class PictureType:
     name: str
     file: str
     likes: List[UserType]
+    created_at: strawberry.auto | None
+    updated_at: strawberry.auto | None
 
 
 @strawberry.django.type(PictureComment)
@@ -41,6 +45,7 @@ class PictureCommentType:
     picture: "PictureType"
     text: str
     created_at: strawberry.auto
+    updated_at: strawberry.auto | None
 
 
 @strawberry.django.type(Collection)
@@ -49,6 +54,8 @@ class CollectionType:
     name: str
     user: "UserType"
     pictures: List[PictureType]
+    created_at: strawberry.auto | None
+    updated_at: strawberry.auto | None
 
 
 @strawberry.django.type(Contest)
@@ -67,6 +74,8 @@ class ContestType:
     winners: List[UserType]
     created_by: "UserType"
     status: str
+    created_at: strawberry.auto | None
+    updated_at: strawberry.auto | None
 
     @strawberry.field
     def status(self) -> str:
@@ -92,6 +101,8 @@ class ContestSubmissionType:
     picture: PictureType
     submission_date: strawberry.auto
     votes: List[UserType]
+    created_at: strawberry.auto | None
+    updated_at: strawberry.auto | None
 
 
 @strawberry.type