mirror of https://github.com/jumpserver/jumpserver
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from django.utils import timezone
|
|
from django.utils.translation import ugettext as _
|
|
from django.template.loader import render_to_string
|
|
|
|
from notifications.notifications import UserMessage
|
|
from common.utils import get_logger
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
class DifferentCityLoginMessage(UserMessage):
|
|
def __init__(self, user, ip, city):
|
|
self.ip = ip
|
|
self.city = city
|
|
super().__init__(user)
|
|
|
|
def get_html_msg(self) -> dict:
|
|
now_local = timezone.localtime(timezone.now())
|
|
now = now_local.strftime("%Y-%m-%d %H:%M:%S")
|
|
subject = _('Different city login reminder')
|
|
context = dict(
|
|
subject=subject,
|
|
name=self.user.name,
|
|
username=self.user.username,
|
|
ip=self.ip,
|
|
time=now,
|
|
city=self.city,
|
|
)
|
|
message = render_to_string('authentication/_msg_different_city.html', context)
|
|
return {
|
|
'subject': subject,
|
|
'message': message
|
|
}
|
|
|
|
@classmethod
|
|
def gen_test_msg(cls):
|
|
from users.models import User
|
|
user = User.objects.first()
|
|
ip = '8.8.8.8'
|
|
city = '洛杉矶'
|
|
return cls(user, ip, city)
|