mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.7 KiB
81 lines
2.7 KiB
from django.template.loader import render_to_string |
|
from django.utils.translation import gettext as _ |
|
|
|
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 = _('today') if day_count == 0 else day_count + _('day') |
|
|
|
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 = _('today') if day_count == 0 else day_count + _('day') |
|
|
|
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, is_console=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': str(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)
|
|
|