robosats/chat/models.py

89 lines
2.5 KiB
Python
Raw Normal View History

2022-01-13 11:07:20 +00:00
from django.db import models
2022-03-10 21:35:16 +00:00
from api.models import User, Order
2022-05-28 13:01:50 +00:00
from django.utils import timezone
import uuid
2022-03-10 21:35:16 +00:00
class ChatRoom(models.Model):
'''
Simple ChatRoom model. Needed to facilitate communication: Is my counterpart in the room?
'''
id = models.PositiveBigIntegerField(primary_key=True, null=False,default=None, blank=True)
order = models.ForeignKey(
Order,
2022-05-28 13:01:50 +00:00
related_name="chatroom",
2022-03-10 21:35:16 +00:00
on_delete=models.SET_NULL,
null=True,
default=None)
maker = models.ForeignKey(
User,
related_name="chat_maker",
on_delete=models.SET_NULL,
null=True,
default=None)
taker = models.ForeignKey(
User,
related_name="chat_taker",
on_delete=models.SET_NULL,
null=True,
default=None,
blank=True,
)
maker_connected = models.BooleanField(default=False, null=False)
taker_connected = models.BooleanField(default=False, null=False)
maker_connect_date = models.DateTimeField(auto_now_add=True)
taker_connect_date = models.DateTimeField(auto_now_add=True)
room_group_name = models.CharField(
max_length=50,
null=True,
default=None,
blank=True,
2022-05-28 13:01:50 +00:00
)
def __str__(self):
return f"Chat:{str(self.id)}"
2022-05-28 13:01:50 +00:00
class Message(models.Model):
2022-05-28 14:59:32 +00:00
class Meta:
get_latest_by = 'index'
# id = models.PositiveBigIntegerField(primary_key=True, default=uuid.uuid4, editable=False)
2022-05-28 13:01:50 +00:00
order = models.ForeignKey(
Order,
related_name="message",
on_delete=models.CASCADE,
null=True,
default=None)
chatroom = models.ForeignKey(
ChatRoom,
related_name="chatroom",
on_delete=models.CASCADE,
null=True,
default=None)
index = models.PositiveIntegerField(null=False,default=None, blank=True)
sender = models.ForeignKey(
User,
related_name="message_sender",
on_delete=models.SET_NULL,
null=True,
default=None)
2022-05-28 14:59:32 +00:00
receiver = models.ForeignKey(
User,
related_name="message_receiver",
on_delete=models.SET_NULL,
null=True,
default=None)
2022-05-28 13:01:50 +00:00
PGP_message = models.TextField(max_length=5000,
null=True,
default=None,
blank=True)
created_at = models.DateTimeField(default=timezone.now)
def __str__(self):
return f"Chat:{str(self.chatroom.id)} - Idx:{self.index}"