mirror of https://github.com/jumpserver/jumpserver
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
from django.template.loader import render_to_string
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from accounts.models import Account
|
|
from assets.models import Asset
|
|
from audits.models import UserLoginLog
|
|
from notifications.notifications import UserMessage
|
|
from users.models import User
|
|
|
|
|
|
class UserLoginReminderMsg(UserMessage):
|
|
subject = _('User login reminder')
|
|
|
|
def __init__(self, user, user_log: UserLoginLog):
|
|
self.user_log = user_log
|
|
super().__init__(user)
|
|
|
|
def get_html_msg(self) -> dict:
|
|
user_log = self.user_log
|
|
context = {
|
|
'ip': user_log.ip,
|
|
'city': user_log.city,
|
|
'username': user_log.username,
|
|
'recipient': self.user,
|
|
'user_agent': user_log.user_agent,
|
|
}
|
|
message = render_to_string('acls/user_login_reminder.html', context)
|
|
|
|
return {
|
|
'subject': str(self.subject),
|
|
'message': message
|
|
}
|
|
|
|
@classmethod
|
|
def gen_test_msg(cls):
|
|
user = User.objects.first()
|
|
user_log = UserLoginLog.objects.first()
|
|
return cls(user, user_log)
|
|
|
|
|
|
class AssetLoginReminderMsg(UserMessage):
|
|
subject = _('Asset login reminder')
|
|
|
|
def __init__(self, user, asset: Asset, login_user: User, account_username):
|
|
self.asset = asset
|
|
self.login_user = login_user
|
|
self.account_username = account_username
|
|
super().__init__(user)
|
|
|
|
def get_html_msg(self) -> dict:
|
|
account = Account.objects.get(asset=self.asset, username=self.account_username)
|
|
context = {
|
|
'recipient': self.user,
|
|
'username': self.login_user.username,
|
|
'name': self.login_user.name,
|
|
'asset': str(self.asset),
|
|
'account': self.account_username,
|
|
'account_name': account.name,
|
|
}
|
|
message = render_to_string('acls/asset_login_reminder.html', context)
|
|
|
|
return {
|
|
'subject': str(self.subject),
|
|
'message': message
|
|
}
|
|
|
|
@classmethod
|
|
def gen_test_msg(cls):
|
|
user = User.objects.first()
|
|
asset = Asset.objects.first()
|
|
return cls(user, asset, user)
|