mirror of https://github.com/jumpserver/jumpserver
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.
85 lines
2.7 KiB
85 lines
2.7 KiB
4 years ago
|
from django.db.models import F
|
||
|
|
|
||
|
|
from common.utils.timezone import now
|
||
|
|
from users.models import User
|
||
|
|
from .models import SiteMessage as SiteMessageModel, SiteMessageUsers
|
||
|
|
|
||
|
|
|
||
|
|
class SiteMessage:
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def send_msg(cls, subject, message, user_ids=(), group_ids=(), sender=None, is_broadcast=False):
|
||
|
|
if not any((user_ids, group_ids, is_broadcast)):
|
||
|
|
raise ValueError('No recipient is specified')
|
||
|
|
|
||
|
|
site_msg = SiteMessageModel.objects.create(
|
||
|
|
subject=subject, message=message,
|
||
|
|
is_broadcast=is_broadcast, sender=sender
|
||
|
|
)
|
||
|
|
|
||
|
|
if is_broadcast:
|
||
|
|
user_ids = User.objects.all().values_list('id', flat=True)
|
||
|
|
else:
|
||
|
|
if group_ids:
|
||
|
|
site_msg.groups.add(*group_ids)
|
||
|
|
|
||
|
|
user_ids_from_group = User.groups.through.objects.filter(
|
||
|
|
usergroup_id__in=group_ids
|
||
|
|
).values_list('user_id', flat=True)
|
||
|
|
|
||
|
|
user_ids = [*user_ids, *user_ids_from_group]
|
||
|
|
|
||
|
|
site_msg.users.add(*user_ids)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_user_all_msgs(cls, user_id):
|
||
|
|
site_msgs = SiteMessageModel.objects.filter(
|
||
|
|
m2m_sitemessageusers__user_id=user_id
|
||
|
|
).distinct().annotate(
|
||
|
|
has_read=F('m2m_sitemessageusers__has_read'),
|
||
|
|
read_at=F('m2m_sitemessageusers__read_at')
|
||
|
|
).order_by('-date_created')
|
||
|
|
|
||
|
|
return site_msgs
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_user_all_msgs_count(cls, user_id):
|
||
|
|
site_msgs_count = SiteMessageModel.objects.filter(
|
||
|
|
m2m_sitemessageusers__user_id=user_id
|
||
|
|
).distinct().count()
|
||
|
|
return site_msgs_count
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_user_unread_msgs(cls, user_id):
|
||
|
|
site_msgs = SiteMessageModel.objects.filter(
|
||
|
|
m2m_sitemessageusers__user_id=user_id,
|
||
|
|
m2m_sitemessageusers__has_read=False
|
||
|
|
).distinct().annotate(
|
||
|
|
has_read=F('m2m_sitemessageusers__has_read'),
|
||
|
|
read_at=F('m2m_sitemessageusers__read_at')
|
||
|
|
).order_by('-date_created')
|
||
|
|
|
||
|
|
return site_msgs
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_user_unread_msgs_count(cls, user_id):
|
||
|
|
site_msgs_count = SiteMessageModel.objects.filter(
|
||
|
|
m2m_sitemessageusers__user_id=user_id,
|
||
|
|
m2m_sitemessageusers__has_read=False
|
||
|
|
).distinct().count()
|
||
|
|
return site_msgs_count
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def mark_msgs_as_read(cls, user_id, msg_ids):
|
||
|
|
sitemsg_users = SiteMessageUsers.objects.filter(
|
||
|
|
user_id=user_id, sitemessage_id__in=msg_ids,
|
||
|
|
has_read=False
|
||
|
|
)
|
||
|
|
|
||
|
|
for sitemsg_user in sitemsg_users:
|
||
|
|
sitemsg_user.has_read = True
|
||
|
|
sitemsg_user.read_at = now()
|
||
|
|
|
||
|
|
SiteMessageUsers.objects.bulk_update(
|
||
|
|
sitemsg_users, fields=('has_read', 'read_at'))
|