mirror of https://github.com/jumpserver/jumpserver
98 lines
3.3 KiB
Python
98 lines
3.3 KiB
Python
from django.template.loader import render_to_string
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from common.tasks import send_mail_attachment_async, upload_backup_to_obj_storage
|
|
from notifications.notifications import UserMessage
|
|
from users.models import User
|
|
from terminal.models.component.storage import ReplayStorage
|
|
|
|
|
|
class AccountBackupExecutionTaskMsg(object):
|
|
subject = _('Notification of account backup route task results')
|
|
|
|
def __init__(self, name: str, user: User):
|
|
self.name = name
|
|
self.user = user
|
|
|
|
@property
|
|
def message(self):
|
|
name = self.name
|
|
if self.user.secret_key:
|
|
return _('{} - The account backup passage task has been completed.'
|
|
' See the attachment for details').format(name)
|
|
else:
|
|
return _("{} - The account backup passage task has been completed: "
|
|
"the encryption password has not been set - "
|
|
"please go to personal information -> file encryption password "
|
|
"to set the encryption password").format(name)
|
|
|
|
def publish(self, attachment_list=None):
|
|
send_mail_attachment_async(
|
|
self.subject, self.message, [self.user.email], attachment_list
|
|
)
|
|
|
|
|
|
class AccountBackupByObjStorageExecutionTaskMsg(object):
|
|
subject = _('Notification of account backup route task results')
|
|
|
|
def __init__(self, name: str, obj_storage: ReplayStorage):
|
|
self.name = name
|
|
self.obj_storage = obj_storage
|
|
|
|
@property
|
|
def message(self):
|
|
name = self.name
|
|
return _('{} - The account backup passage task has been completed.'
|
|
' See the attachment for details').format(name)
|
|
|
|
def publish(self, attachment_list=None):
|
|
upload_backup_to_obj_storage(
|
|
self.obj_storage, attachment_list
|
|
)
|
|
|
|
|
|
class ChangeSecretExecutionTaskMsg(object):
|
|
subject = _('Notification of implementation result of encryption change plan')
|
|
|
|
def __init__(self, name: str, user: User):
|
|
self.name = name
|
|
self.user = user
|
|
|
|
@property
|
|
def message(self):
|
|
name = self.name
|
|
if self.user.secret_key:
|
|
return _('{} - The encryption change task has been completed. '
|
|
'See the attachment for details').format(name)
|
|
else:
|
|
return _("{} - The encryption change task has been completed: the encryption "
|
|
"password has not been set - please go to personal information -> "
|
|
"file encryption password to set the encryption password").format(name)
|
|
|
|
def publish(self, attachments=None):
|
|
send_mail_attachment_async(
|
|
self.subject, self.message, [self.user.email], attachments
|
|
)
|
|
|
|
|
|
class GatherAccountChangeMsg(UserMessage):
|
|
subject = _('Gather account change information')
|
|
|
|
def __init__(self, user, change_info: dict):
|
|
self.change_info = change_info
|
|
super().__init__(user)
|
|
|
|
def get_html_msg(self) -> dict:
|
|
context = {'change_info': self.change_info}
|
|
message = render_to_string('accounts/asset_account_change_info.html', context)
|
|
|
|
return {
|
|
'subject': str(self.subject),
|
|
'message': message
|
|
}
|
|
|
|
@classmethod
|
|
def gen_test_msg(cls):
|
|
user = User.objects.first()
|
|
return cls(user, {})
|