mirror of https://github.com/jumpserver/jumpserver
83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
|
|
from django.utils.translation import ugettext as _
|
|
from django.template.loader import render_to_string
|
|
|
|
from common.utils import reverse as js_reverse
|
|
from notifications.notifications import UserMessage
|
|
|
|
|
|
class PermedAssetsWillExpireUserMsg(UserMessage):
|
|
def __init__(self, user, assets, day_count=0):
|
|
super().__init__(user)
|
|
self.assets = assets
|
|
self.day_count = day_count
|
|
|
|
def get_html_msg(self) -> dict:
|
|
subject = _("You permed assets is about to expire")
|
|
context = {
|
|
'name': self.user.name,
|
|
'count': self.day_count,
|
|
'items': [str(asset) for asset in self.assets],
|
|
'item_type': _("permed assets"),
|
|
'show_help': True
|
|
}
|
|
message = render_to_string('perms/_msg_permed_items_expire.html', context)
|
|
return {
|
|
'subject': subject,
|
|
'message': message
|
|
}
|
|
|
|
@classmethod
|
|
def gen_test_msg(cls):
|
|
from users.models import User
|
|
from assets.models import Asset
|
|
user = User.objects.first()
|
|
assets = Asset.objects.all()[:10]
|
|
return cls(user, assets)
|
|
|
|
|
|
class AssetPermsWillExpireForOrgAdminMsg(UserMessage):
|
|
|
|
def __init__(self, user, perms, org, day_count=0):
|
|
super().__init__(user)
|
|
self.perms = perms
|
|
self.org = org
|
|
self.day_count = day_count
|
|
|
|
def get_items_with_url(self):
|
|
items_with_url = []
|
|
for perm in self.perms:
|
|
url = js_reverse(
|
|
'perms:asset-permission-detail',
|
|
kwargs={'pk': perm.id}, external=True,
|
|
api_to_ui=True
|
|
) + f'?oid={perm.org_id}'
|
|
items_with_url.append([perm.name, url])
|
|
return items_with_url
|
|
|
|
def get_html_msg(self):
|
|
items_with_url = self.get_items_with_url()
|
|
subject = _("Asset permissions is about to expire")
|
|
context = {
|
|
'name': self.user.name,
|
|
'count': self.day_count,
|
|
'items_with_url': items_with_url,
|
|
'item_type': _('asset permissions of organization {}').format(self.org)
|
|
}
|
|
message = render_to_string('perms/_msg_item_permissions_expire.html', context)
|
|
return {
|
|
'subject': subject,
|
|
'message': message
|
|
}
|
|
|
|
@classmethod
|
|
def gen_test_msg(cls):
|
|
from users.models import User
|
|
from perms.models import AssetPermission
|
|
from orgs.models import Organization
|
|
|
|
user = User.objects.first()
|
|
perms = AssetPermission.objects.all()[:10]
|
|
org = Organization.objects.first()
|
|
return cls(user, perms, org)
|