Commit 7eed1c076936a8d9bcb7e4d409eafd6f0d800975

Authored by Gustavo Bernardo
1 parent ec6c233a

Now when the user is deleted, his profile picture is also deleted

Showing 1 changed file with 6 additions and 3 deletions   Show diff stats
users/views.py
... ... @@ -30,7 +30,7 @@ from django.core.mail import EmailMessage
30 30 from django.core.mail.backends.smtp import EmailBackend
31 31  
32 32 from mailsender.models import MailSender
33   -
  33 +import os
34 34 #API IMPORTS
35 35 from rest_framework import viewsets
36 36 from .serializers import UserSerializer
... ... @@ -209,6 +209,8 @@ class DeleteView(braces_mixins.LoginRequiredMixin, LogMixin, generic.DeleteView)
209 209 email = self.kwargs.get('email', None)
210 210 user = self.get_object()
211 211  
  212 +
  213 +
212 214 if email is None:
213 215 self.log_action = 'remove_account'
214 216  
... ... @@ -233,11 +235,12 @@ class DeleteView(braces_mixins.LoginRequiredMixin, LogMixin, generic.DeleteView)
233 235 redirect_url = redirect(error_url)
234 236 else:
235 237 self.log_context['dependencies'] = False
236   -
  238 + image_path_to_delete = user.image.path
237 239 user.delete()
238 240  
239 241 messages.success(self.request, success_msg)
240   -
  242 + #deleting the user image
  243 + os.remove(image_path_to_delete)
241 244 redirect_url = redirect(success_url)
242 245  
243 246 super(DeleteView, self).createLog(self.request.user, self.log_component, self.log_action, self.log_resource, self.log_context)
... ...