mirror of https://github.com/jumpserver/jumpserver
124 lines
4.7 KiB
Python
124 lines
4.7 KiB
Python
from django.utils import timezone
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from accounts.models import Account
|
|
from acls.models import LoginACL, LoginAssetACL
|
|
from assets.models import Asset
|
|
from audits.models import UserLoginLog
|
|
from common.views.template import custom_render_to_string
|
|
from notifications.notifications import UserMessage
|
|
from users.models import User
|
|
|
|
|
|
class UserLoginReminderMsg(UserMessage):
|
|
subject = _('User login reminder')
|
|
template_name = 'acls/user_login_reminder.html'
|
|
contexts = [
|
|
{"name": "city", "label": _('Login city'), "default": "Shanghai"},
|
|
{"name": "username", "label": _('User'), "default": "John"},
|
|
{"name": "ip", "label": "IP", "default": "192.168.1.1"},
|
|
{"name": "recipient_name", "label": _("Recipient name"), "default": "John"},
|
|
{"name": "recipient_username", "label": _("Recipient username"), "default": "john"},
|
|
{"name": "user_agent", "label": _('User agent'), "default": "Mozilla/5.0"},
|
|
{"name": "acl_name", "label": _('ACL name'), "default": "login acl"},
|
|
{"name": "login_from", "label": _('Login from'), "default": "web"},
|
|
{"name": "time", "label": _('Login time'), "default": "2025-01-01 12:00:00"},
|
|
]
|
|
|
|
def __init__(self, user, user_log: UserLoginLog, acl: LoginACL):
|
|
self.user_log = user_log
|
|
self.acl_name = str(acl)
|
|
self.login_from = user_log.get_type_display()
|
|
now = timezone.localtime(user_log.datetime)
|
|
self.time = now.strftime('%Y-%m-%d %H:%M:%S')
|
|
super().__init__(user)
|
|
|
|
def get_html_msg(self) -> dict:
|
|
user_log = self.user_log
|
|
context = {
|
|
'ip': user_log.ip,
|
|
'time': self.time,
|
|
'city': user_log.city,
|
|
'acl_name': self.acl_name,
|
|
'login_from': self.login_from,
|
|
'username': user_log.username,
|
|
'recipient_name': self.user.name,
|
|
'recipient_username': self.user.username,
|
|
'user_agent': user_log.user_agent,
|
|
}
|
|
message = custom_render_to_string(self.template_name, 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 = _('User login alert for asset')
|
|
template_name = 'acls/asset_login_reminder.html'
|
|
contexts = [
|
|
{"name": "city", "label": _('Login city'), "default": "Shanghai"},
|
|
{"name": "username", "label": _('User'), "default": "John"},
|
|
{"name": "name", "label": _('Name'), "default": "John"},
|
|
{"name": "asset", "label": _('Asset'), "default": "dev server"},
|
|
{"name": "recipient_name", "label": _('Recipient name'), "default": "John"},
|
|
{"name": "recipient_username", "label": _('Recipient username'), "default": "john"},
|
|
{"name": "account", "label": _('Account Input username'), "default": "root"},
|
|
{"name": "account_name", "label": _('Account name'), "default": "root"},
|
|
{"name": "acl_name", "label": _('ACL name'), "default": "login acl"},
|
|
{"name": "ip", "label": "IP", "default": "192.168.1.1"},
|
|
{"name": "login_from", "label": _('Login from'), "default": "web"},
|
|
{"name": "time", "label": _('Login time'), "default": "2025-01-01 12:00:00"}
|
|
]
|
|
|
|
def __init__(
|
|
self, user, asset: Asset, login_user: User,
|
|
account: Account, acl: LoginAssetACL,
|
|
ip, input_username, login_from
|
|
):
|
|
self.ip = ip
|
|
self.asset = asset
|
|
self.login_user = login_user
|
|
self.account = account
|
|
self.acl_name = str(acl)
|
|
self.login_from = login_from
|
|
self.login_user = login_user
|
|
self.input_username = input_username
|
|
|
|
now = timezone.localtime(timezone.now())
|
|
self.time = now.strftime('%Y-%m-%d %H:%M:%S')
|
|
super().__init__(user)
|
|
|
|
def get_html_msg(self) -> dict:
|
|
context = {
|
|
'ip': self.ip,
|
|
'time': self.time,
|
|
'login_from': self.login_from,
|
|
'recipient': self.user,
|
|
'username': self.login_user.username,
|
|
'name': self.login_user.name,
|
|
'asset': str(self.asset),
|
|
'account': self.input_username,
|
|
'account_name': self.account.name,
|
|
'acl_name': self.acl_name,
|
|
}
|
|
message = custom_render_to_string(self.template_name, 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)
|