You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
956 B
33 lines
956 B
from django.db import models
|
|
from django.conf import settings
|
|
|
|
|
|
class MessagesManager(models.Manager):
|
|
def get_my_messages(self, request):
|
|
if request.user.is_authenticated():
|
|
num = self.filter(recepient=request.user, is_viewed=False).count()
|
|
else:
|
|
num = 0
|
|
return int(num)
|
|
|
|
|
|
class Dialog(models.Model):
|
|
title = models.CharField(max_length=127)
|
|
owner = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='+')
|
|
recepient = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='+')
|
|
date_create = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
|
|
class PrivateMessages(models.Model):
|
|
dialog = models.ForeignKey(Dialog)
|
|
date_send = models.DateTimeField(auto_now_add=True)
|
|
text = models.TextField()
|
|
is_viewed = models.BooleanField(default=False)
|
|
|
|
objects = MessagesManager()
|
|
|
|
def __str__(self):
|
|
return self.text
|