Merge pull request #19 from farooq13/feature

add DM

Author: farooq13

base/__pycache__/models.cpython-312.pyc

@@ -0,0 +1,37 @@
+# Generated by Django 5.1.1 on 2024-11-24 19:57
+
+import django.db.models.deletion
+import django.utils.timezone
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('base', '0004_post_image'),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='ThreadModel',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('receiver', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='+', to=settings.AUTH_USER_MODEL)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='+', to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='Message',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('body', models.CharField(max_length=1000)),
+                ('date', models.DateTimeField(default=django.utils.timezone.now)),
+                ('is_read', models.BooleanField(default=False)),
+                ('receiver_user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='+', to=settings.AUTH_USER_MODEL)),
+                ('sender_user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='+', to=settings.AUTH_USER_MODEL)),
+                ('thread', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='+', to='base.threadmodel')),
+            ],
+        ),
+    ]

base/migrations/0005_threadmodel_message.py

@@ -72,3 +72,17 @@ class Notification(models.Model):
   comment = models.ForeignKey('Comment', on_delete=models.CASCADE, related_name='+', null=True, blank=True)
   date = models.DateTimeField(default=timezone.now)
   user_has_seen = models.BooleanField(default=False)
+
+
+class ThreadModel(models.Model):
+  user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='+')
+  receiver = models.ForeignKey(User, on_delete=models.CASCADE, related_name='+')
+
+
+class Message(models.Model):
+  thread = models.ForeignKey(ThreadModel, on_delete=models.CASCADE, related_name='+', blank=True, null=True)
+  sender_user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='+')
+  receiver_user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='+')
+  body = models.CharField(max_length=1000)
+  date = models.DateTimeField(default=timezone.now)
+  is_read = models.BooleanField(default=False)

base/migrations/__pycache__/0005_threadmodel_message.cpython-312.pyc

@@ -5,7 +5,7 @@
 from django.shortcuts import get_object_or_404
 from django.http import HttpResponseRedirect, HttpResponse
 from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin
-from .models import Post, Comment, UserProfile, Notification
+from .models import Post, Comment, UserProfile, Notification, ThreadModel, Message
 from django.db.models import Q
 from .forms import PostForm, CommentForm, ProfileForm
 
@@ -319,4 +319,18 @@ def delete(self, request, notification_pk, *args, **kwargs):
     notification = Notification.objects.get(pk=notification_pk)
     notification.user_has_seen = True
     notification.save()
-    return HttpResponse('success', content_type='text/plain')
+    return HttpResponse('success', content_type='text/plain')
+  
+
+
+class ListThread(View):
+  def get(self, request, *args, **kwargs):
+    threads = ThreadModel.objects.filter(
+      Q(user=request.user) |
+      Q(receiver=request.user)
+    )
+
+    context = {
+      'threads': threads
+    }
+    return render(request, 'base/inbox.html', context)