mirror of
https://github.com/ansible/awx.git
synced 2026-02-17 11:10:03 -03:30
add schedule_deletion method and signal
This commit is contained in:
@@ -59,7 +59,7 @@ import ansiconv
|
|||||||
from social.backends.utils import load_backends
|
from social.backends.utils import load_backends
|
||||||
|
|
||||||
# AWX
|
# 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.access import get_user_queryset
|
||||||
from awx.main.ha import is_ha_environment
|
from awx.main.ha import is_ha_environment
|
||||||
from awx.api.authentication import TaskAuthentication, TokenGetAuthentication
|
from awx.api.authentication import TaskAuthentication, TokenGetAuthentication
|
||||||
@@ -1839,15 +1839,13 @@ class InventoryDetail(ControlledByScmMixin, RetrieveUpdateDestroyAPIView):
|
|||||||
|
|
||||||
def destroy(self, request, *args, **kwargs):
|
def destroy(self, request, *args, **kwargs):
|
||||||
obj = self.get_object()
|
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):
|
if not request.user.can_access(self.model, 'delete', obj):
|
||||||
raise PermissionDenied()
|
raise PermissionDenied()
|
||||||
obj.websocket_emit_status('pending_deletion')
|
try:
|
||||||
delete_inventory.delay(obj.id)
|
obj.schedule_deletion()
|
||||||
obj.pending_deletion = True
|
return Response(status=status.HTTP_202_ACCEPTED)
|
||||||
obj.save(update_fields=['pending_deletion'])
|
except RuntimeError, e:
|
||||||
return Response(status=status.HTTP_202_ACCEPTED)
|
return Response(dict(error=_("{0}".format(e))), status=status.HTTP_400_BAD_REQUEST)
|
||||||
|
|
||||||
|
|
||||||
class InventoryActivityStreamList(ActivityStreamEnforcementMixin, SubListAPIView):
|
class InventoryActivityStreamList(ActivityStreamEnforcementMixin, SubListAPIView):
|
||||||
|
|||||||
@@ -373,6 +373,16 @@ class Inventory(CommonModelNameNotUnique, ResourceMixin):
|
|||||||
raise ValidationError(_("Credential kind must be 'insights'."))
|
raise ValidationError(_("Credential kind must be 'insights'."))
|
||||||
return self.insights_credential
|
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):
|
class SmartInventoryMembership(BaseModel):
|
||||||
'''
|
'''
|
||||||
|
|||||||
@@ -510,3 +510,13 @@ def get_current_user_from_drf_request(sender, **kwargs):
|
|||||||
request = get_current_request()
|
request = get_current_request()
|
||||||
drf_request = getattr(request, 'drf_request', None)
|
drf_request = getattr(request, 'drf_request', None)
|
||||||
return (getattr(drf_request, 'user', False), 0)
|
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)
|
||||||
|
|||||||
Reference in New Issue
Block a user