def delete_data_senders(request):
manager = get_database_manager(request.user)
organization = get_organization(request)
entity_type = request.POST['entity_type']
all_ids = data_sender_short_codes(request, manager)
superusers = rep_id_name_dict_of_users(manager)
non_superuser_rep_ids = [
id for id in all_ids if id not in superusers.keys()
]
transport_info = TransportInfo("web", request.user.username, "")
delete_datasenders_from_project(manager, non_superuser_rep_ids)
delete_entity_instance(manager, non_superuser_rep_ids, entity_type,
transport_info)
delete_datasender_users_if_any(non_superuser_rep_ids, organization)
if organization.in_trial_mode:
delete_datasender_for_trial_mode(manager, non_superuser_rep_ids,
entity_type)
log_activity(
request,
DELETED_DATA_SENDERS,
"%s: [%s]" %
(entity_type.capitalize(), ", ".join(non_superuser_rep_ids)),
)
messages = get_success_message(entity_type)
return HttpResponse(json.dumps({'success': True, 'message': messages}))
def delete_users(request):
if request.method == 'GET':
raise Http404
django_ids = request.POST.get("all_ids").split(";")
all_ids = NGOUserProfile.objects.filter(user__in=django_ids).values_list(
'reporter_id', flat=True)
manager = get_database_manager(request.user)
organization = get_organization(request)
transport_info = TransportInfo("web", request.user.username, "")
ngo_admin_user_profile = get_ngo_admin_user_profiles_for(organization)[0]
if ngo_admin_user_profile.reporter_id in all_ids:
messages.error(
request,
_("Your organization's account Administrator %s cannot be deleted")
% (ngo_admin_user_profile.user.first_name), "error_message")
else:
detail = user_activity_log_details(
User.objects.filter(id__in=django_ids))
delete_datasenders_from_project(manager, all_ids)
for user_id in django_ids:
handle_feature_toggle_impact_for_deleted_user(user_id)
delete_entity_instance(manager, all_ids, REPORTER, transport_info)
delete_datasender_users_if_any(all_ids, organization)
if organization.in_trial_mode:
delete_datasender_for_trial_mode(manager, all_ids, REPORTER)
action = DELETED_USERS
UserActivityLog().log(request, action=action, detail=detail)
messages.success(request, _("User(s) successfully deleted."))
return HttpResponse(json.dumps({'success': True}))
def test_should_delete_a_datasender_from_associated_projects(self):
entity_ids = ['rep1', 'rep2']
with patch("datawinners.project.models.Project") as mock_project_class:
mock_project = Mock(spec=Project)
mock_project_class.get.return_value = mock_project
with patch("datawinners.project.models.get_all_projects"
) as all_projects:
all_projects.return_value = [{'value': {'_id': 1}}]
dbm = Mock()
dbm.database.return_value = ''
delete_datasenders_from_project(dbm, entity_ids)
self.assertEqual(mock_project.delete_datasender.call_count, 2)
