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.
73 lines
2.5 KiB
73 lines
2.5 KiB
from django.utils.translation import gettext as _
|
|
|
|
from common.utils import get_logger
|
|
from orgs.utils import tmp_to_org
|
|
from perms.models import AssetPermission
|
|
from tickets.models import ApplyAssetTicket
|
|
from .base import BaseHandler
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
class Handler(BaseHandler):
|
|
ticket: ApplyAssetTicket
|
|
|
|
def _on_step_approved(self, step):
|
|
is_finished = super()._on_step_approved(step)
|
|
if is_finished:
|
|
self._create_asset_permission()
|
|
|
|
def _create_asset_permission(self):
|
|
org_id = self.ticket.org_id
|
|
with tmp_to_org(org_id):
|
|
asset_permission = AssetPermission.objects.filter(id=self.ticket.id).first()
|
|
if asset_permission:
|
|
return asset_permission
|
|
|
|
apply_nodes = self.ticket.apply_nodes.all()
|
|
apply_assets = self.ticket.apply_assets.all()
|
|
|
|
apply_permission_name = self.ticket.apply_permission_name
|
|
apply_actions = self.ticket.apply_actions
|
|
apply_accounts = self.ticket.apply_accounts
|
|
apply_date_start = self.ticket.apply_date_start
|
|
apply_date_expired = self.ticket.apply_date_expired
|
|
permission_created_by = '{}:{}'.format(
|
|
str(self.ticket.__class__.__name__), str(self.ticket.id)
|
|
)
|
|
permission_comment = _(
|
|
'Created by the ticket '
|
|
'ticket title: {} '
|
|
'ticket applicant: {} '
|
|
'ticket processor: {} '
|
|
'ticket ID: {}'
|
|
).format(
|
|
self.ticket.title,
|
|
self.ticket.applicant,
|
|
','.join([i['processor_display'] for i in self.ticket.process_map]),
|
|
str(self.ticket.id)
|
|
)
|
|
|
|
permission_data = {
|
|
'from_ticket': True,
|
|
'id': self.ticket.id,
|
|
'actions': apply_actions,
|
|
'accounts': apply_accounts,
|
|
'name': apply_permission_name,
|
|
'date_start': apply_date_start,
|
|
'date_expired': apply_date_expired,
|
|
'comment': str(permission_comment),
|
|
'created_by': permission_created_by,
|
|
}
|
|
with tmp_to_org(self.ticket.org_id):
|
|
asset_permission = AssetPermission.objects.create(**permission_data)
|
|
asset_permission.nodes.set(apply_nodes)
|
|
asset_permission.assets.set(apply_assets)
|
|
|
|
try:
|
|
asset_permission.users.add(self.ticket.applicant)
|
|
except Exception as e:
|
|
logger.error('Add user to asset permission failed: %s', e)
|
|
|
|
return asset_permission
|