from django.db import models from osinacore.models import * import mimetypes import os from channels.layers import get_channel_layer from asgiref.sync import async_to_sync from fcm_django.models import FCMDevice from firebase_admin.messaging import Message,AndroidConfig,APNSConfig,APNSPayload,ApsAlert,Aps,AndroidNotification, Notification as NotificationFB from django.utils.safestring import mark_safe def send_notification(notification): notification_data = { 'title': notification.title, 'body': mark_safe(notification.message), } android_config = AndroidConfig( notification=AndroidNotification( title=notification_data['title'], body=notification_data['body'], sound='default' ) ) apns_config = APNSConfig( payload= APNSPayload( aps=Aps( alert=ApsAlert( title=notification_data['title'], body=notification_data['body'], ), sound='default', ) ) ) if notification.image: FCMDevice.objects.send_message( Message(notification=NotificationFB( title=notification_data['title'], body=notification_data['body'], image= notification.image, ), data={"image": notification.image},android=android_config,apns=apns_config) ) else: FCMDevice.objects.send_message( Message(notification=NotificationFB( title=notification_data['title'], body=notification_data['body'], ), android=android_config, apns=apns_config) ) # Create your models here. class ChatNotification(models.Model): title = models.CharField(max_length=255) message = models.TextField() image = models.TextField(blank=True,null=True) created_at = models.DateTimeField(auto_now_add=True) def save(self, *args, **kwargs): is_new = not self.pk super().save(*args, **kwargs) if is_new: send_notification(self) class Visitor(models.Model): session_id = models.CharField(max_length=300) ip_address = models.CharField(max_length=300) country = models.CharField(max_length=15, null=True) region = models.CharField(max_length=25, null=True) name = models.CharField(max_length=200, blank=True, null=True) mobile_number = models.CharField(max_length=10, null=True, blank=True) email = models.CharField(max_length=100, null=True, blank=True) browser_name = models.CharField(max_length=100, null=True, blank=True) os_name = models.CharField(max_length=100, null=True, blank=True) @property def flag_image_url(self): flag_url = f"https://osina.ositcom.com/static/images/flags/{self.country.lower()}.svg" return flag_url @property def notification_flag_image_url(self): flag_url = f"https://flagcdn.com/w640/{self.country.lower()}.png" return flag_url @property def is_online(self): latest_log = self.visitorlog_set.order_by('-visit_date').first() if latest_log and latest_log.left_date is None: return True return False @property def total_duration(self): total_seconds = 0 for log in self.visitorlog_set.all(): end_time = log.left_date if log.left_date else timezone.now() total_seconds += (end_time - log.visit_date).total_seconds() duration = timedelta(seconds=total_seconds) hours, remainder = divmod(duration.total_seconds(), 3600) minutes, seconds = divmod(remainder, 60) if hours >= 1: return f"{int(hours):02}:{int(minutes):02}:{int(seconds):02}" else: return f"{int(minutes):02}:{int(seconds):02}" def save(self, *args, **kwargs): super().save(*args, **kwargs) channel_layer = get_channel_layer() event = { 'type': 'new_visitor_update_handler', 'visitor_id': self.id, 'action': 'new_visitor' } async_to_sync(channel_layer.group_send)("osichat", event) self.send_visitor_notification() def send_visitor_notification(self): notification = ChatNotification.objects.create( title="New visitor on Ositcom!", image = self.notification_flag_image_url ) class VisitorLog(models.Model): visitor = models.ForeignKey(Visitor, on_delete=models.CASCADE, null=True) url = models.URLField() title = models.CharField(null=True, blank=True, max_length=500) referrer = models.URLField(null=True, blank=True) visit_date = models.DateTimeField(null=True) left_date = models.DateTimeField(null=True) @property def log_duration(self): end_time = self.left_date if self.left_date else timezone.now() total_seconds = (end_time - self.visit_date).total_seconds() duration = timedelta(seconds=total_seconds) hours, remainder = divmod(duration.total_seconds(), 3600) minutes, seconds = divmod(remainder, 60) if hours >= 1: return f"{int(hours):02}:{int(minutes):02}:{int(seconds):02}" else: return f"{int(minutes):02}:{int(seconds):02}" def save(self, *args, **kwargs): is_new = not self.pk if self.left_date: action = 'end_log' else: action = 'new_log' super().save(*args, **kwargs) channel_layer = get_channel_layer() event = { 'type': 'new_visitor_update_handler', 'visitor_id': self.visitor.id, 'action': action } async_to_sync(channel_layer.group_send)("osichat", event) if is_new: self.send_visitorlog_notification() def send_visitorlog_notification(self): notification = ChatNotification.objects.create( title=f"Visitor navigated to: {self.title}.", image=self.visitor.notification_flag_image_url ) class ChatRoom(models.Model): name = models.CharField(max_length=300) created_by = models.ForeignKey(User, null=True, on_delete=models.SET_NULL, blank=True) date_created = models.DateTimeField() terminated_by = models.ForeignKey(User, null=True, on_delete=models.SET_NULL, blank=True, related_name='terminated_chatrooms') date_terminated = models.DateTimeField(null=True, blank=True) @property def last_updated(self): last_message = ChatMessage.objects.filter(room=self).order_by('-date_sent').first() if last_message: last_updated_time = last_message.date_sent else: last_updated_time = self.date_created now = datetime.now() if last_updated_time.date() == now.date(): return last_updated_time.strftime('%I:%M %p') else: return last_updated_time.strftime('%d-%m-%Y') def unread_messages(self, user): return ChatMessage.objects.filter(room=self).exclude(member=user).exclude(chatmessageseen__member=user).count() def save(self, *args, **kwargs): super().save(*args, **kwargs) channel_layer = get_channel_layer() event = { 'type': 'new_chat_update_handler', 'chatroom_id': self.id, } async_to_sync(channel_layer.group_send)("osichat", event) class ChatRoomGuest(models.Model): room = models.OneToOneField(ChatRoom, on_delete=models.CASCADE, null=True) visitor = models.ForeignKey(Visitor, null=True, on_delete=models.CASCADE) class ChatRoomReview(models.Model): REACTION_CHOICES = ( ('Happy', 'Happy'), ('Indifferent', 'Indifferent'), ('Sad', 'Sad'), ) room = models.OneToOneField(ChatRoom, on_delete=models.CASCADE, null=True) reaction = models.CharField(max_length=50, choices=REACTION_CHOICES, null=True) details = models.TextField(null=True, blank=True) class ChatMember(models.Model): member = models.ForeignKey(User, on_delete=models.CASCADE) room = models.ForeignKey(ChatRoom, on_delete=models.CASCADE) date_joined = models.DateTimeField() class ChatProject(models.Model): room = models.OneToOneField(ChatRoom, on_delete=models.CASCADE) project = models.ForeignKey(Project, on_delete=models.CASCADE) public = models.BooleanField() class ChatMessage(models.Model): room = models.ForeignKey(ChatRoom, on_delete=models.CASCADE, null=True) member = models.ForeignKey(User, null=True, on_delete=models.SET_NULL, blank=True) content = models.TextField(null=True, blank=True) date_sent = models.DateTimeField() def save(self, *args, **kwargs): super().save(*args, **kwargs) channel_layer = get_channel_layer() event = { 'type': 'new_chat_update_handler', 'chatroom_id': self.room.id, } async_to_sync(channel_layer.group_send)("osichat", event) class ChatMessageAttachment(models.Model): message = models.OneToOneField(ChatMessage, on_delete=models.CASCADE) attachment = models.TextField() def is_image(self): mime_type, _ = mimetypes.guess_type(self.attachment) return mime_type and mime_type.startswith('image') @property def file_name(self): return os.path.basename(self.attachment) class ChatMessageReaction(models.Model): message = models.ForeignKey(ChatMessage, on_delete=models.CASCADE) member = models.ForeignKey(User, on_delete=models.CASCADE) reaction = models.CharField(max_length=200) class ChatMessageSeen(models.Model): message = models.ForeignKey(ChatMessage, on_delete=models.CASCADE) member = models.ForeignKey(User, on_delete=models.CASCADE, null=True, blank=True) guest = models.ForeignKey(ChatRoomGuest, on_delete=models.CASCADE, null=True, blank=True) seen_date = models.DateTimeField(null=True)