Init

music_storage/core/models.py

@@ -0,0 +1,42 @@
+from django.db import models
+from django.contrib.auth import get_user_model
+from core.middleware.current_request import get_current_user
+
+
+UserModel = get_user_model()
+
+
+class BaseModel(models.Model):
+    """Abstract base model that includes created_by and updated_by fields.
+
+    The fields are automatically populated with the current user from the request.
+    """
+
+    created_by = models.ForeignKey(
+        UserModel,
+        on_delete=models.SET_NULL,
+        null=True,
+        blank=True,
+        related_name="%(class)s_created",
+        editable=False,
+    )
+    updated_by = models.ForeignKey(
+        UserModel,
+        on_delete=models.SET_NULL,
+        null=True,
+        blank=True,
+        related_name="%(class)s_updated",
+        editable=False,
+    )
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+
+    class Meta:
+        abstract = True
+
+    def save(self, *args, **kwargs):
+        current_user = get_current_user()
+        if not self.pk and not self.created_by:
+            self.created_by = current_user
+        self.updated_by = current_user
+        super().save(*args, **kwargs)