mirror of
https://github.com/ansible/awx.git
synced 2026-01-11 18:09:57 -03:30
add schedule_deletion method and signal
This commit is contained in:
parent
7d283f1dfd
commit
e84ecabe72
@ -59,7 +59,7 @@ import ansiconv
|
||||
from social.backends.utils import load_backends
|
||||
|
||||
# AWX
|
||||
from awx.main.tasks import send_notifications, update_host_smart_inventory_memberships, delete_inventory
|
||||
from awx.main.tasks import send_notifications, update_host_smart_inventory_memberships
|
||||
from awx.main.access import get_user_queryset
|
||||
from awx.main.ha import is_ha_environment
|
||||
from awx.api.authentication import TaskAuthentication, TokenGetAuthentication
|
||||
@ -1839,15 +1839,13 @@ class InventoryDetail(ControlledByScmMixin, RetrieveUpdateDestroyAPIView):
|
||||
|
||||
def destroy(self, request, *args, **kwargs):
|
||||
obj = self.get_object()
|
||||
if obj.pending_deletion is True:
|
||||
return Response(dict(error=_("Inventory is already being deleted.")), status=status.HTTP_400_BAD_REQUEST)
|
||||
if not request.user.can_access(self.model, 'delete', obj):
|
||||
raise PermissionDenied()
|
||||
obj.websocket_emit_status('pending_deletion')
|
||||
delete_inventory.delay(obj.id)
|
||||
obj.pending_deletion = True
|
||||
obj.save(update_fields=['pending_deletion'])
|
||||
return Response(status=status.HTTP_202_ACCEPTED)
|
||||
try:
|
||||
obj.schedule_deletion()
|
||||
return Response(status=status.HTTP_202_ACCEPTED)
|
||||
except RuntimeError, e:
|
||||
return Response(dict(error=_("{0}".format(e))), status=status.HTTP_400_BAD_REQUEST)
|
||||
|
||||
|
||||
class InventoryActivityStreamList(ActivityStreamEnforcementMixin, SubListAPIView):
|
||||
|
||||
@ -373,6 +373,16 @@ class Inventory(CommonModelNameNotUnique, ResourceMixin):
|
||||
raise ValidationError(_("Credential kind must be 'insights'."))
|
||||
return self.insights_credential
|
||||
|
||||
@transaction.atomic
|
||||
def schedule_deletion(self):
|
||||
from awx.main.tasks import delete_inventory
|
||||
if self.pending_deletion is True:
|
||||
raise RuntimeError("Inventory is already pending deletion.")
|
||||
self.websocket_emit_status('pending_deletion')
|
||||
delete_inventory.delay(self.pk)
|
||||
self.pending_deletion = True
|
||||
self.save(update_fields=['pending_deletion'])
|
||||
|
||||
|
||||
class SmartInventoryMembership(BaseModel):
|
||||
'''
|
||||
|
||||
@ -510,3 +510,13 @@ def get_current_user_from_drf_request(sender, **kwargs):
|
||||
request = get_current_request()
|
||||
drf_request = getattr(request, 'drf_request', None)
|
||||
return (getattr(drf_request, 'user', False), 0)
|
||||
|
||||
|
||||
@receiver(pre_delete, sender=Organization)
|
||||
def delete_inventory_for_org(sender, instance, **kwargs):
|
||||
inventories = Inventory.objects.filter(organization__pk=instance.pk)
|
||||
for inventory in inventories:
|
||||
try:
|
||||
inventory.schedule_deletion()
|
||||
except RuntimeError, e:
|
||||
logger.debug(e)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user